2 * ============LICENSE_START====================================================
4 * ===========================================================================
5 * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
7 * Modifications Copyright (C) 2019 IBM.
8 * ===========================================================================
9 * Licensed under the Apache License, Version 2.0 (the "License");
10 * you may not use this file except in compliance with the License.
11 * You may obtain a copy of the License at
13 * http://www.apache.org/licenses/LICENSE-2.0
15 * Unless required by applicable law or agreed to in writing, software
16 * distributed under the License is distributed on an "AS IS" BASIS,
17 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
18 * See the License for the specific language governing permissions and
19 * limitations under the License.
20 * ============LICENSE_END====================================================
24 package org.onap.aaf.auth.cmd;
26 import java.io.BufferedReader;
28 import java.io.FileReader;
29 import java.io.InputStreamReader;
30 import java.io.OutputStreamWriter;
31 import java.io.PrintWriter;
32 import java.io.Reader;
33 import java.io.Writer;
34 import java.net.HttpURLConnection;
35 import java.util.ArrayList;
36 import java.util.List;
38 import org.onap.aaf.auth.cmd.mgmt.Mgmt;
39 import org.onap.aaf.auth.cmd.ns.NS;
40 import org.onap.aaf.auth.cmd.perm.Perm;
41 import org.onap.aaf.auth.cmd.role.Role;
42 import org.onap.aaf.auth.cmd.user.User;
43 import org.onap.aaf.auth.common.Define;
44 import org.onap.aaf.auth.env.AuthzEnv;
45 import org.onap.aaf.cadi.Access;
46 import org.onap.aaf.cadi.Access.Level;
47 import org.onap.aaf.cadi.CadiException;
48 import org.onap.aaf.cadi.PropAccess;
49 import org.onap.aaf.cadi.SecuritySetter;
50 import org.onap.aaf.cadi.aaf.v2_0.AAFConHttp;
51 import org.onap.aaf.cadi.client.Retryable;
52 import org.onap.aaf.cadi.config.Config;
53 import org.onap.aaf.cadi.config.SecurityInfoC;
54 import org.onap.aaf.cadi.http.HBasicAuthSS;
55 import org.onap.aaf.cadi.http.HMangr;
56 import org.onap.aaf.cadi.sso.AAFSSO;
57 import org.onap.aaf.misc.env.APIException;
59 import jline.console.ConsoleReader;
62 protected static PrintWriter pw;
63 protected HMangr hman;
64 // Storage for last reused client. We can do this
65 // because we're technically "single" threaded calls.
66 public Retryable<?> prevCall;
68 protected SecuritySetter<HttpURLConnection> ss;
69 // protected AuthzEnv env;
70 private boolean close;
71 private List<Cmd> cmds;
74 private ArrayList<Integer> expect = new ArrayList<>();
75 private boolean verbose = true;
77 private SecurityInfoC<HttpURLConnection> si;
78 private boolean request = false;
79 private String force = null;
80 private boolean gui = false;
85 private static int TIMEOUT = Integer.parseInt(Config.AAF_CONN_TIMEOUT_DEF);
86 private static boolean isConsole = false;
87 private static boolean isTest = false;
88 private static boolean showDetails = false;
89 private static boolean ignoreDelay = false;
90 private static int globalDelay=0;
92 // Create when only have Access
93 public AAFcli(Access access, Writer wtr, HMangr hman, SecurityInfoC<HttpURLConnection> si, SecuritySetter<HttpURLConnection> ss) throws APIException {
94 this(access,new AuthzEnv(access.getProperties()),wtr,hman, si,ss);
97 public AAFcli(Access access, AuthzEnv env, Writer wtr, HMangr hman, SecurityInfoC<HttpURLConnection> si, SecuritySetter<HttpURLConnection> ss) throws APIException {
103 if (wtr instanceof PrintWriter) {
104 pw = (PrintWriter) wtr;
107 pw = new PrintWriter(wtr);
114 cmds = new ArrayList<>();
116 Role role = new Role(this);
117 cmds.add(new Help(this, cmds));
118 cmds.add(new Version(this));
119 cmds.add(new Perm(role));
121 cmds.add(new User(this));
122 cmds.add(new NS(this));
123 cmds.add(new Mgmt(this));
126 public AuthzEnv env() {
130 public static int timeout() {
134 public void verbose(boolean v) {
138 public void close() {
139 // if (hman != null) {
148 public boolean eval(String line) throws Exception {
149 if (line.length() == 0) {
151 } else if (line.startsWith("#")) {
156 String[] largs = argEval(line);
159 // Variable replacement
160 StringBuilder sb = null;
161 while (idx < largs.length) {
163 for (int v = largs[idx].indexOf("@["); v >= 0; v = largs[idx].indexOf("@[", v + 1)) {
165 sb = new StringBuilder();
167 sb.append(largs[idx], e, v);
168 if ((e = largs[idx].indexOf(']', v)) >= 0) {
169 String p = access.getProperty(largs[idx].substring(v + 2, e),null);
171 p = System.getProperty(largs[idx].substring(v+2,e));
179 if (sb != null && sb.length() > 0) {
180 sb.append(largs[idx], e, largs[idx].length());
181 largs[idx] = sb.toString();
189 while (rv && idx < largs.length) {
190 // Allow Script to change Credential
192 if ("as".equalsIgnoreCase(largs[idx])) {
193 if (largs.length > ++idx) {
194 // get Password from Props with ID as Key
195 String user = largs[idx++];
196 int colon = user.indexOf(':');
199 pass = user.substring(colon + 1);
200 user = user.substring(0, colon);
202 pass = access.getProperty(user, null);
205 pass = access.decrypt(pass, false);
206 access.getProperties().put(user, pass);
207 ss=new HBasicAuthSS(si, user, pass);
208 pw.println("as " + user);
209 } else { // get Pass from System Properties, under name of
211 pw.println("ERROR: No password set for " + user);
216 } else if ("expect".equalsIgnoreCase(largs[idx])) {
218 if ((largs.length > idx++)&&(!"nothing".equals(largs[idx]))) {
219 for (String str : largs[idx].split(",")) {
221 if ("Exception".equalsIgnoreCase(str)) {
224 expect.add(Integer.parseInt(str));
226 } catch (NumberFormatException e) {
227 throw new CadiException("\"expect\" should be followed by Number");
233 // Sleep, typically for reports, to allow DB to update
236 } else if ("sleep".equalsIgnoreCase(largs[idx])) {
237 Integer t = Integer.parseInt(largs[++idx]);
238 pw.println("sleep " + t);
242 } else if ("delay".equalsIgnoreCase(largs[idx])) {
243 delay = Integer.parseInt(largs[++idx]);
244 pw.println("delay " + delay);
247 } else if ("pause".equalsIgnoreCase(largs[idx])) {
248 pw.println("Press <Return> to continue...");
250 // Sonar insists we do something with the string, though it's only a pause. Not very helpful...
251 BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
254 sonar = br.readLine();
258 sonar=""; // this useless code brought to you by Sonar.
261 } else if ("exit".equalsIgnoreCase(largs[idx])) {
262 pw.println("Exiting...");
264 } else if ("set".equalsIgnoreCase(largs[idx])) {
265 while (largs.length > ++idx) {
266 int equals = largs[idx].indexOf('=');
270 value = access.getProperty(Config.AAF_APPPASS,null);
274 value = access.decrypt(value, false);
278 access.getProperties().put(tag, value);
279 pw.println("set " + tag + " <encrypted>");
282 tag = largs[idx].substring(0, equals);
283 value = largs[idx].substring(++equals);
284 pw.println("set " + tag + ' ' + value);
286 boolean isTrue = "TRUE".equalsIgnoreCase(value);
287 if ("FORCE".equalsIgnoreCase(tag)) {
289 } else if ("REQUEST".equalsIgnoreCase(tag)) {
291 } else if ("DETAILS".equalsIgnoreCase(tag)) {
292 showDetails = isTrue;
294 access.getProperties().put(tag, value);
298 // Allow Script to indicate if Failure is what is expected
303 if ("REQUEST".equalsIgnoreCase(largs[idx])) {
306 } else if ("FORCE".equalsIgnoreCase(largs[idx])) {
309 } else if ("DETAILS".equalsIgnoreCase(largs[idx])) {
316 if (largs[idx].equalsIgnoreCase(c.getName())) {
319 if (expect.size() > 0) {
320 pw.print("** Expect ");
321 boolean first = true;
322 for (Integer i : expect) {
334 ret = c.exec(++idx, largs);
335 if (delay+globalDelay > 0) {
336 Thread.sleep((long)(delay+globalDelay));
338 } catch (Exception e) {
339 if (expect.contains(-1)) {
340 pw.println(e.getMessage());
346 clearSingleLineProperties();
348 rv = expect.isEmpty() || expect.contains(ret);
353 pw.print("!!! Unexpected Return Code: ");
355 pw.println(", VALIDATE OUTPUT!!!");
361 pw.write("Unknown Instruction \"");
362 pw.write(largs[idx]);
364 idx = largs.length;// always end after one command
369 private String[] argEval(String line) {
370 StringBuilder sb = new StringBuilder();
371 ArrayList<String> arr = new ArrayList<>();
372 boolean start = true;
375 for (int i = 0; i < line.length(); ++i) {
378 if (Character.isWhitespace(ch)) {
379 if (start || last==',') {
381 } else if (quote != 0) {
384 arr.add(sb.toString());
388 } else if (ch == '\'' || ch == '"') { // toggle
394 } else if (ch=='|' && quote==0) {
395 arr.add(sb.toString());
404 if (sb.length() > 0) {
405 arr.add(sb.toString());
408 String[] rv = new String[arr.size()];
413 public static void keyboardHelp() {
414 System.out.println("'C-' means hold the ctrl key down while pressing the next key.");
415 System.out.println("'M-' means hold the alt key down while pressing the next key.");
416 System.out.println("For instance, C-b means hold ctrl key and press b, M-b means hold alt and press b\n");
418 System.out.println("Basic Keybindings:");
419 System.out.println("\tC-l - clear screen");
420 System.out.println("\tC-a - beginning of line");
421 System.out.println("\tC-e - end of line");
422 System.out.println("\tC-b - backward character (left arrow also works)");
423 System.out.println("\tM-b - backward word");
424 System.out.println("\tC-f - forward character (right arrow also works)");
425 System.out.println("\tM-f - forward word");
426 System.out.println("\tC-d - delete character under cursor");
427 System.out.println("\tM-d - delete word forward");
428 System.out.println("\tM-backspace - delete word backward");
429 System.out.println("\tC-k - delete from cursor to end of line");
430 System.out.println("\tC-u - delete entire line, regardless of cursor position\n");
432 System.out.println("Command History:");
433 System.out.println("\tC-r - search backward in history (repeating C-r continues the search)");
434 System.out.println("\tC-p - move backwards through history (up arrow also works)");
435 System.out.println("\tC-n - move forwards through history (down arrow also works)\n");
442 public static void main(String[] args) {
446 AAFSSO aafsso = new AAFSSO(args);
447 String noexit = aafsso.access().getProperty("no_exit");
449 PropAccess access = aafsso.access();
453 AuthzEnv env = new AuthzEnv(access);
456 boolean exitOnFailure = true;
458 * Check for "-" options anywhere in command line
460 StringBuilder sb = new StringBuilder();
461 for (int i = 0; i < args.length; ++i) {
462 if ("-i".equalsIgnoreCase(args[i])) {
463 rdr = new InputStreamReader(System.in);
464 // } else if ("-o".equalsIgnoreCase(args[i])) {
465 // // shall we do something different? Output stream is
467 } else if ("-f".equalsIgnoreCase(args[i])) {
468 if (args.length > i + 1) {
469 rdr = new FileReader(args[++i]);
471 } else if ("-a".equalsIgnoreCase(args[i])) {
472 exitOnFailure = false;
473 } else if ("-c".equalsIgnoreCase(args[i])) {
475 } else if ("-s".equalsIgnoreCase(args[i]) && args.length > i + 1) {
476 access.setProperty(Cmd.STARTDATE, args[++i]);
477 } else if ("-e".equalsIgnoreCase(args[i]) && args.length > i + 1) {
478 access.setProperty(Cmd.ENDDATE, args[++i]);
479 } else if ("-t".equalsIgnoreCase(args[i])) {
481 } else if ("-d".equalsIgnoreCase(args[i])) {
483 } else if ("-n".equalsIgnoreCase(args[i])) {
486 if (sb.length() > 0) {
493 AAFConHttp aafcon = new AAFConHttp(access);
495 // SecurityInfoC<?> si = aafcon.securityInfo();
498 aafsso.setLogDefault();
499 aafsso.setStdErrDefault();
501 // Note, with AAF Locator, this may not longer be necessary 3/2018 Jonathan
502 if (!aafsso.loginOnly()) {
504 // loc = new AAFLocator(si,new URI(access.getProperty(Config.AAF_URL)));
505 // } catch (Throwable t) {
506 // aafsso.setStdErrDefault();
509 // // Other Access is done writing to StdOut and StdErr, reset Std out
510 // aafsso.setLogDefault();
513 TIMEOUT = Integer.parseInt(access.getProperty(Config.AAF_CONN_TIMEOUT, Config.AAF_CONN_TIMEOUT_DEF));
514 // HMangr hman = new HMangr(access, loc).readTimeout(TIMEOUT).apiVersion(Config.AAF_DEFAULT_API_VERSION);
516 if (access.getProperty(Config.AAF_DEFAULT_REALM)==null) {
517 access.setProperty(Config.AAF_DEFAULT_REALM, "people.osaaf.org");
518 aafsso.addProp(Config.AAF_DEFAULT_REALM, "people.osaaf.org");
521 AAFcli aafcli = new AAFcli(access,env, new OutputStreamWriter(System.out),
522 aafcon.hman(), aafcon.securityInfo(), aafcon.securityInfo().defSS);
523 // new HBasicAuthSS(si,aafsso.user(), access.decrypt(aafsso.enc_pass(),false)));
526 File delay = new File("aafcli.delay");
527 if (delay.exists()) {
528 BufferedReader br = new BufferedReader(new FileReader(delay));
530 globalDelay = Integer.parseInt(br.readLine());
531 } catch (Exception e) {
532 access.log(Level.DEBUG,e);
540 System.out.println("Type 'help' for short help or 'help -d' for detailed help with aafcli commands");
541 System.out.println("Type '?' for help with command line editing");
542 System.out.println("Type 'q', 'quit', or 'exit' to quit aafcli\n");
544 ConsoleReader reader = new ConsoleReader();
546 reader.setPrompt("aafcli > ");
549 while ((line = reader.readLine()) != null) {
550 showDetails = (line.contains("-d"));
552 if ("quit".equalsIgnoreCase(line) || "q".equalsIgnoreCase(line) || "exit".equalsIgnoreCase(line)) {
554 } else if ("--help -d".equalsIgnoreCase(line) || "help -d".equalsIgnoreCase(line)
555 || "help".equalsIgnoreCase(line)) {
557 } else if ("cls".equalsIgnoreCase(line)) {
558 reader.clearScreen();
560 } else if ("?".equalsIgnoreCase(line)) {
567 } catch (Exception e) {
568 pw.println(e.getMessage());
575 } else if (rdr != null) {
576 BufferedReader br = new BufferedReader(rdr);
579 while ((line = br.readLine()) != null) {
580 if (!aafcli.eval(line) && exitOnFailure) {
588 } else { // just run the command line
589 aafcli.verbose(false);
590 if (sb.length() == 0) {
593 rv = aafcli.eval(sb.toString()) ? 0 : 1;
599 // Don't close if No Reader, or it's a Reader of Standard In
600 if (rdr != null && !(rdr instanceof InputStreamReader)) {
608 StringBuilder err = aafsso.err();
610 err.append("to continue...");
611 System.err.println(err);
619 } catch (MessageException e) {
620 System.out.println("MessageException caught");
622 System.err.println(e.getMessage());
623 } catch (Exception e) {
624 e.printStackTrace(System.err);
629 public boolean isTest() {
630 return AAFcli.isTest;
633 public boolean isDetailed() {
634 return AAFcli.showDetails;
637 public String typeString(Class<?> cls, boolean json) {
638 return "application/" + cls.getSimpleName() + "+" + (json ? "json" : "xml");//+ ";version=" + hman.apiVersion();
641 public String forceString() {
645 public boolean addRequest() {
649 public void clearSingleLineProperties() {
655 public void gui(boolean b) {