krb-name-mapping-test.cc

Go to the documentation of this file.

#include "krb-name-mapper.h"

#include "common/logging.h"

#include <boost/foreach.hpp>

using namespace std;
using namespace impala;



int main(int argc, char** argv) {
  vector<string> rules;
  rules.push_back("RULE:[1:$1@$0](.*@YAHOO\\.COM)s/@.*//");
  rules.push_back("RULE:[2:$1](johndoe)s/^.*$/guest/");
  rules.push_back("RULE:[2:$1;$2](^.*;admin$)s/;admin$//");
  rules.push_back("RULE:[2:$2](root)");
  rules.push_back("DEFAULT");

  FLAGS_kerberos_default_realm = "CLOUDERA.COM";
  KrbNameMapper k(rules);
  vector<string> principals;
  principals.push_back("omalley@CLOUDERA.COM");
  principals.push_back("hdfs/10.0.0.1@CLOUDERA.COM");
  principals.push_back("johndoe/zoo@FOO.COM");
  principals.push_back("joe/admin@FOO.COM");
  principals.push_back("joe/root@FOO.COM");

  BOOST_FOREACH(const string& principal, principals) {
    string out;
    if (k.PrincipalToLocal(principal, &out)) {
      LOG(INFO) << "Mapped: " << principal << " to: " << out;
    } else {
      LOG(INFO) << "Mapping for: " << principal << " failed!";
    }
  }
}