Beispiel #1
0
  @ScriptUsage(
      description = "join security group",
      arguments = {
        @ScriptArgument(
            name = "group name",
            type = "string",
            description = "name of security group"),
        @ScriptArgument(name = "login name", type = "string", description = "account login name")
      })
  public void joinSecurityGroup(String[] args) {
    SecurityGroup group = findSecurityGroupByName(args[0]);
    if (group == null) {
      context.println("security group not found");
      return;
    }

    for (int i = 1; i < args.length; i++) {
      String loginName = args[i];
      Account account = accountService.getAccount(loginName);
      if (account == null) {
        context.println("account [" + loginName + "] not found. skipping.");
        continue;
      }

      group.getAccounts().add(account.getLoginName());
    }

    accountService.updateSecurityGroup(null, group);
    context.println("updated");
  }
Beispiel #2
0
  /** @since 2.0.3 */
  @ScriptUsage(
      description = "delete saved query result",
      arguments = {
        @ScriptArgument(
            name = "guid",
            type = "string",
            description = "the guid of saved query result")
      })
  public void deleteSavedResult(String[] args) throws IOException {
    String guid = args[0];
    SavedResult sr = savedResultManager.getResult(guid);
    if (sr == null) {
      context.println("query result not found");
      return;
    }

    savedResultManager.deleteResult(guid);
    context.println(
        "deleted '"
            + sr.getTitle()
            + "', "
            + sr.getFileSize()
            + " bytes, "
            + sr.getRowCount()
            + " rows");
  }
Beispiel #3
0
 public void csvLookups(String[] args) {
   context.println("CSV Mapping Files");
   context.println("-------------------");
   for (File f : csvRegistry.getCsvFiles()) {
     context.println(f.getAbsolutePath());
   }
 }
Beispiel #4
0
  @ScriptUsage(
      description = "revoke procedure",
      arguments = {
        @ScriptArgument(name = "procedure name", type = "string", description = "procedure name"),
        @ScriptArgument(name = "type", type = "string", description = "user or group"),
        @ScriptArgument(name = "login name", type = "string", description = "login name")
      })
  public void revokeProcedure(String[] args) {
    String procName = args[0];
    String type = args[1];
    String target = args[2];

    Procedure p = procedureRegistry.getProcedure(procName);
    if (p == null) {
      context.println("procedure not found");
      return;
    }

    if (type.equals("user")) {
      p.getGrants().remove(target);
      procedureRegistry.updateProcedure(p);
    } else if (type.equals("group")) {
      Map<String, SecurityGroup> groupMap = getSecurityGroupMap();
      SecurityGroup old = groupMap.get(target);
      if (old != null) {
        p.getGrantGroups().remove(old.getGuid());
        procedureRegistry.updateProcedure(p);
      }
    } else {
      context.println("invalid type. use 'user' or 'group'");
      return;
    }

    context.println("revoked");
  }
Beispiel #5
0
 public void sessions(String[] args) {
   context.println("Current Sessions");
   context.println("------------------");
   for (Session session : accountService.getSessions()) {
     context.println(session);
   }
 }
Beispiel #6
0
  public void securityGroups(String[] args) {
    context.println("Security Groups");
    context.println("-----------------");

    for (SecurityGroup group : accountService.getSecurityGroups()) {
      context.println(group);
    }
  }
Beispiel #7
0
 public void accounts(String[] args) {
   context.println("Accounts");
   context.println("----------");
   for (String loginName : accountService.getAccountNames()) {
     Account account = accountService.getAccount(loginName);
     String admin = "";
     if (account.isAdmin()) admin = " (admin)";
     context.println(loginName + admin);
   }
 }
Beispiel #8
0
  public void authServices(String[] args) {
    context.println("External Auth Services");
    context.println("------------------------");

    ExternalAuthService using = accountService.getUsingAuthService();
    for (ExternalAuthService s : accountService.getAuthServices()) {
      context.print(using == s ? "[*] " : "[ ] ");
      context.println(s.getName() + " - " + s);
    }
  }
Beispiel #9
0
  public void createProcedure(String[] args) {
    try {
      Procedure proc = inputProcedure(null);

      procedureRegistry.createProcedure(proc);
      context.println("created");
    } catch (InterruptedException e) {
      context.println("");
      context.println("interrupted");
    }
  }
Beispiel #10
0
  @ScriptUsage(
      description = "print procedure details",
      arguments = {@ScriptArgument(name = "name", type = "string", description = "procedure name")})
  public void procedure(String[] args) {
    Procedure p = procedureRegistry.getProcedure(args[0]);
    if (p == null) {
      context.println("procedure not found");
      return;
    }

    context.println(p);
  }
Beispiel #11
0
  public void scripts(String[] args) {
    context.println("Log Scripts");
    context.println("--------------");

    for (String workspace : scriptRegistry.getWorkspaceNames()) {
      context.println("Workspace: " + workspace);
      for (String name : scriptRegistry.getScriptFactoryNames(workspace)) {
        QueryScriptFactory factory = scriptRegistry.getScriptFactory(workspace, name);
        context.println("  " + name + " - " + factory);
      }
    }
  }
Beispiel #12
0
  @ScriptUsage(
      description = "revoke table access from group",
      arguments = {
        @ScriptArgument(name = "table name", type = "string", description = "table name"),
        @ScriptArgument(name = "type", type = "string", description = "user or group"),
        @ScriptArgument(
            name = "login name or group name",
            type = "string",
            description = "name of user or security group")
      })
  public void revokeTable(String[] args) {
    String tableName = args[0];
    String type = args[1];

    if (!tableRegistry.exists(tableName)) {
      context.println("table not found: " + tableName);
      return;
    }

    if (!type.equals("user") && !type.equals("group")) {
      context.println("invalid type, use 'user' or 'group'");
      return;
    }

    if (type.equals("user")) {
      for (int i = 2; i < args.length; i++) {
        String loginName = args[i];
        if (accountService.getAccount(loginName) == null) {
          context.println("account [" + loginName + "] not found, skipping");
          continue;
        }

        accountService.revokePrivilege(null, loginName, tableName, Permission.READ);
      }
    } else if (type.equals("group")) {
      Map<String, SecurityGroup> groupMap = getSecurityGroupMap();
      for (int i = 2; i < args.length; i++) {
        String groupName = args[i];
        SecurityGroup group = groupMap.get(groupName);
        if (group == null) {
          context.println("security group [" + groupName + "] not found, skipping");
          continue;
        }

        group.getReadableTables().remove(tableName);
        accountService.updateSecurityGroup(null, group);
      }
    }

    context.println("revoked");
  }
Beispiel #13
0
 /** @since 0.14.0 */
 @ScriptUsage(
     description = "remove query",
     arguments = {@ScriptArgument(name = "query id", type = "int", description = "query id")})
 public void removeQuery(String[] args) {
   for (String arg : args) {
     Integer id = Integer.valueOf(arg);
     if (qs.getQuery(id) != null) {
       qs.removeQuery(id);
       context.println("removed query " + arg);
     } else {
       context.println("query " + id + " not found");
     }
   }
 }
Beispiel #14
0
  /** @since 0.16.2 */
  @ScriptUsage(
      description = "print specific query status",
      arguments = {@ScriptArgument(name = "query id", type = "int", description = "query id")})
  public void queryStatus(String[] args) {
    Integer id = Integer.valueOf(args[0]);
    Query q = qs.getQuery(id);
    if (q == null) {
      context.println("query " + id + " not found");
      return;
    }

    context.println("Query Status");
    context.println("------------------");
    context.println(QueryPrintHelper.getQueryStatus(q));
  }
Beispiel #15
0
  public void procedures(String[] args) {
    context.println("Procedures");
    context.println("------------");

    for (Procedure p : procedureRegistry.getProcedures()) {
      int i = 0;
      String signature = p.getName() + "(";
      for (ProcedureParameter param : p.getParameters()) {
        if (i++ != 0) signature += ", ";
        signature += param;
      }
      signature += ")";

      context.println(signature + ", owner=" + p.getOwner());
    }
  }
Beispiel #16
0
  @ScriptUsage(
      description = "remove procedure",
      arguments = {
        @ScriptArgument(name = "procedure name", type = "string", description = "procedure name")
      })
  public void removeProcedure(String[] args) {
    String name = args[0];
    Procedure p = procedureRegistry.getProcedure(name);
    if (p == null) {
      context.println("procedure not found: " + name);
      return;
    }

    procedureRegistry.removeProcedure(name);
    context.println("removed");
  }
Beispiel #17
0
  public void updateProcedure(String[] args) {
    try {
      String name = readLine("name", null);
      Procedure old = procedureRegistry.getProcedure(name);
      if (old == null) {
        context.println("procedure not found: " + name);
        return;
      }

      Procedure proc = inputProcedure(old);
      proc.setCreated(old.getCreated());
      procedureRegistry.updateProcedure(proc);
      context.println("updated");
    } catch (InterruptedException e) {
      context.println("");
      context.println("interrupted");
    }
  }
Beispiel #18
0
  @ScriptUsage(
      description = "remove security group",
      arguments = {
        @ScriptArgument(
            name = "group name",
            type = "string",
            description = "name of security group")
      })
  public void removeSecurityGroup(String[] args) {
    SecurityGroup found = findSecurityGroupByName(args[0]);
    if (found == null) {
      context.println("security group not found");
      return;
    }

    accountService.removeSecurityGroup(null, found.getGuid());
    context.println("removed");
  }
Beispiel #19
0
 @ScriptUsage(
     description = "set hash join threshold",
     arguments = {
       @ScriptArgument(name = "threshold", type = "int", description = "hash join threshold")
     })
 public void setHashJoinThreshold(String[] args) {
   System.setProperty("araqne.hashjointhreshold", args[0]);
   context.println("set hash join threshold");
 }
Beispiel #20
0
 @ScriptUsage(
     description = "load csv lookup mapping file",
     arguments = {
       @ScriptArgument(
           name = "path",
           type = "string",
           description =
               "csv (comma separated value) file path. first line should be column headers.",
           autocompletion = PathAutoCompleter.class)
     })
 public void loadCsvLookup(String[] args) throws IOException {
   try {
     File f = new File(args[0]);
     csvRegistry.loadCsvFile(f);
     context.println("loaded " + f.getAbsolutePath());
   } catch (IllegalStateException e) {
     context.println(e);
   }
 }
Beispiel #21
0
 @ScriptUsage(
     description = "create new query script workspace",
     arguments = {
       @ScriptArgument(
           name = "workspace name",
           type = "string",
           description = "query script workspace name")
     })
 public void createScriptWorkspace(String[] args) {
   scriptRegistry.createWorkspace(args[0]);
   context.println("created");
 }
Beispiel #22
0
 @ScriptUsage(
     description = "remove query script workspace",
     arguments = {
       @ScriptArgument(
           name = "workspace name",
           type = "string",
           description = "query script workspace name")
     })
 public void dropScriptWorkspace(String[] args) {
   scriptRegistry.dropWorkspace(args[0]);
   context.println("dropped");
 }
Beispiel #23
0
  @ScriptUsage(
      description = "leave security group",
      arguments = {
        @ScriptArgument(
            name = "group name",
            type = "string",
            description = "name of security group"),
        @ScriptArgument(name = "login name", type = "string", description = "account login name")
      })
  public void leaveSecurityGroup(String[] args) {
    SecurityGroup group = findSecurityGroupByName(args[0]);
    if (group == null) {
      context.println("security group not found");
      return;
    }

    for (int i = 1; i < args.length; i++) group.getAccounts().remove(args[i]);

    accountService.updateSecurityGroup(null, group);
    context.println("updated");
  }
Beispiel #24
0
  private Procedure inputProcedure(Procedure old) throws InterruptedException {
    Procedure proc = new Procedure();

    if (old == null) proc.setName(readLine("name", null));
    else proc.setName(old.getName());

    proc.setOwner(readLine("owner", old != null ? old.getOwner() : null));
    proc.setGrants(
        new HashSet<String>(
            Strings.tokenize(
                readLine("grants", old != null ? Strings.join(old.getGrants(), ", ") : null),
                ",")));
    proc.setQueryString(readLine("query", old != null ? old.getQueryString() : null));

    List<ProcedureParameter> parameters = new ArrayList<ProcedureParameter>();
    context.println(
        "Type parameter definitions in \"type name\" format. e.g. \"string opt\", press enter to end.");
    int idx = 0;
    while (true) {
      ProcedureParameter oldParam = null;
      if (old != null && idx < old.getParameters().size()) {
        oldParam = old.getParameters().get(idx);
      }

      context.print("parameter? ");
      String line = context.readLine(oldParam != null ? oldParam.toString() : null);
      if (line.isEmpty()) break;

      int p = line.indexOf(" ");
      String type = line.substring(0, p).trim();
      String key = line.substring(p + 1).trim();

      parameters.add(new ProcedureParameter(key, type));
      idx++;
    }

    proc.setParameters(parameters);

    return proc;
  }
Beispiel #25
0
  /** @since 2.0.3 */
  @ScriptUsage(
      description = "list all saved query results",
      arguments = {
        @ScriptArgument(
            name = "owner",
            type = "string",
            description = "login name",
            optional = true)
      })
  public void savedResults(String[] args) {
    String owner = args.length > 0 ? args[0] : null;
    List<SavedResult> savedResults = savedResultManager.getResultList(owner);

    context.println("Saved Query Results");
    context.println("---------------------");

    for (SavedResult sr : savedResults) {
      context.println(sr);
    }

    context.println("total " + savedResults.size() + " results");
  }
Beispiel #26
0
 @ScriptUsage(
     description = "unload csv lookup mapping file",
     arguments = {
       @ScriptArgument(
           name = "path",
           type = "string",
           description = "registered csv file path",
           autocompletion = PathAutoCompleter.class)
     })
 public void unloadCsvLookup(String[] args) {
   File f = new File(args[0]);
   csvRegistry.unloadCsvFile(f);
   context.println("unloaded" + f.getAbsolutePath());
 }
Beispiel #27
0
  @ScriptUsage(
      description = "create security group",
      arguments = {
        @ScriptArgument(
            name = "group name",
            type = "string",
            description = "name of security group"),
        @ScriptArgument(
            name = "description",
            type = "string",
            description = "description",
            optional = true)
      })
  public void createSecurityGroup(String[] args) {
    SecurityGroup group = new SecurityGroup();
    group.setName(args[0]);
    if (args.length > 1) group.setDescription(args[1]);

    accountService.createSecurityGroup(null, group);
    context.println("created");
  }
Beispiel #28
0
 public void useAuthService(String[] args) {
   accountService.useAuthService(args.length > 0 ? args[0] : null);
   context.println(args.length > 0 ? "set" : "unset");
 }
Beispiel #29
0
 @ScriptUsage(description = "get hash join threshold")
 public void getHashJoinThreshold(String[] args) {
   context.println(
       "hash join threshold :" + System.getProperty("araqne.hashjointhreshold", "100000"));
 }
Beispiel #30
0
 private String readLine(String prompt, String old) throws InterruptedException {
   context.print(prompt + "? ");
   return context.readLine(old);
 }