1
0

shell_tinyxml_read.cpp 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629
  1. ////////////////////////////////////////
  2. // shell_tinyxml_read.cpp
  3. //
  4. // This file implements the means to read an XML encoded file and populate the
  5. // CLI tree based on the contents.
  6. ////////////////////////////////////////
  7. extern "C" {
  8. #include "private.h"
  9. #include "lub/string.h"
  10. #include "lub/ctype.h"
  11. }
  12. /*lint +libh(tinyxml/tinyxml.h) Add this to the library file list */
  13. #include "tinyxml/tinyxml.h"
  14. #include <string.h>
  15. #include <assert.h>
  16. typedef void (PROCESS_FN) (clish_shell_t * instance,
  17. TiXmlElement * element, void *parent);
  18. // Define a control block for handling the decode of an XML file
  19. typedef struct clish_xml_cb_s clish_xml_cb_t;
  20. struct clish_xml_cb_s {
  21. const char *element;
  22. PROCESS_FN *handler;
  23. };
  24. // forward declare the handler functions
  25. static PROCESS_FN
  26. process_clish_module,
  27. process_startup,
  28. process_view,
  29. process_command,
  30. process_param,
  31. process_action,
  32. process_ptype,
  33. process_overview, process_detail, process_namespace, process_config;
  34. static clish_xml_cb_t xml_elements[] = {
  35. {"CLISH_MODULE", process_clish_module},
  36. {"STARTUP", process_startup},
  37. {"VIEW", process_view},
  38. {"COMMAND", process_command},
  39. {"PARAM", process_param},
  40. {"ACTION", process_action},
  41. {"PTYPE", process_ptype},
  42. {"OVERVIEW", process_overview},
  43. {"DETAIL", process_detail},
  44. {"NAMESPACE", process_namespace},
  45. {"CONFIG", process_config},
  46. {NULL, NULL}
  47. };
  48. ///////////////////////////////////////
  49. // This function reads an element from the XML stream and processes it.
  50. ///////////////////////////////////////
  51. static void process_node(clish_shell_t * shell, TiXmlNode * node, void *parent)
  52. {
  53. switch (node->Type()) {
  54. case TiXmlNode::DOCUMENT:
  55. break;
  56. case TiXmlNode::ELEMENT:
  57. clish_xml_cb_t * cb;
  58. for (cb = &xml_elements[0]; cb->element; cb++) {
  59. if (0 == strcmp(node->Value(), cb->element)) {
  60. TiXmlElement *element = (TiXmlElement *) node;
  61. #ifdef DEBUG
  62. fprintf(stderr, "NODE:");
  63. element->Print(stderr, 0);
  64. fprintf(stderr, "\n***\n");
  65. #endif
  66. // process the elements at this level
  67. cb->handler(shell, element, parent);
  68. break;
  69. }
  70. }
  71. break;
  72. case TiXmlNode::COMMENT:
  73. case TiXmlNode::TEXT:
  74. case TiXmlNode::DECLARATION:
  75. case TiXmlNode::TYPECOUNT:
  76. case TiXmlNode::UNKNOWN:
  77. default:
  78. break;
  79. }
  80. }
  81. ///////////////////////////////////////
  82. static void
  83. process_children(clish_shell_t * shell,
  84. TiXmlElement * element, void *parent = NULL)
  85. {
  86. for (TiXmlNode * node = element->FirstChild();
  87. NULL != node; node = element->IterateChildren(node)) {
  88. // Now deal with all the contained elements
  89. process_node(shell, node, parent);
  90. }
  91. }
  92. ///////////////////////////////////////
  93. static void
  94. process_clish_module(clish_shell_t * shell, TiXmlElement * element, void *)
  95. {
  96. // create the global view
  97. if (NULL == shell->global) {
  98. shell->global =
  99. clish_shell_find_create_view(shell, "global", "");
  100. }
  101. process_children(shell, element, shell->global);
  102. }
  103. ///////////////////////////////////////
  104. static void process_view(clish_shell_t * shell, TiXmlElement * element, void *)
  105. {
  106. clish_view_t *view;
  107. const char *name = element->Attribute("name");
  108. const char *prompt = element->Attribute("prompt");
  109. const char *depth = element->Attribute("depth");
  110. const char *restore = element->Attribute("restore");
  111. // re-use a view if it already exists
  112. view = clish_shell_find_create_view(shell, name, prompt);
  113. if ((depth != NULL) && (*depth != '\0') && (lub_ctype_isdigit(*depth))) {
  114. unsigned res = atoi(depth);
  115. clish_view__set_depth(view, res);
  116. }
  117. if (restore) {
  118. if (!lub_string_nocasecmp(restore, "depth"))
  119. clish_view__set_restore(view, CLISH_RESTORE_DEPTH);
  120. else if (!lub_string_nocasecmp(restore, "view"))
  121. clish_view__set_restore(view, CLISH_RESTORE_VIEW);
  122. else
  123. clish_view__set_restore(view, CLISH_RESTORE_NONE);
  124. }
  125. process_children(shell, element, view);
  126. }
  127. ///////////////////////////////////////
  128. static void process_ptype(clish_shell_t * shell, TiXmlElement * element, void *)
  129. {
  130. clish_ptype_method_e method;
  131. clish_ptype_preprocess_e preprocess;
  132. clish_ptype_t *ptype;
  133. const char *name = element->Attribute("name");
  134. const char *help = element->Attribute("help");
  135. const char *pattern = element->Attribute("pattern");
  136. const char *method_name = element->Attribute("method");
  137. const char *preprocess_name = element->Attribute("preprocess");
  138. assert(name);
  139. assert(pattern);
  140. method = clish_ptype_method_resolve(method_name);
  141. preprocess = clish_ptype_preprocess_resolve(preprocess_name);
  142. ptype = clish_shell_find_create_ptype(shell,
  143. name,
  144. help,
  145. pattern, method, preprocess);
  146. assert(ptype);
  147. }
  148. ///////////////////////////////////////
  149. static void
  150. process_overview(clish_shell_t * shell, TiXmlElement * element, void *)
  151. {
  152. // read the following text element
  153. TiXmlNode *text = element->FirstChild();
  154. if (NULL != text) {
  155. assert(TiXmlNode::TEXT == text->Type());
  156. // set the overview text for this view
  157. assert(NULL == shell->overview);
  158. // store the overview
  159. shell->overview = lub_string_dup(text->Value());
  160. }
  161. }
  162. ////////////////////////////////////////
  163. static void
  164. process_command(clish_shell_t * shell, TiXmlElement * element, void *parent)
  165. {
  166. clish_view_t *v = (clish_view_t *) parent;
  167. clish_command_t *cmd = NULL;
  168. const char *access = element->Attribute("access");
  169. bool allowed = true;
  170. if (NULL != access) {
  171. allowed = false; // err on the side of caution
  172. if (shell->client_hooks->access_fn) {
  173. // get the client to authenticate
  174. allowed =
  175. shell->client_hooks->access_fn(shell,
  176. access) ? true :
  177. false;
  178. }
  179. }
  180. if (allowed) {
  181. const char *name = element->Attribute("name");
  182. const char *help = element->Attribute("help");
  183. const char *view = element->Attribute("view");
  184. const char *viewid = element->Attribute("viewid");
  185. const char *escape_chars = element->Attribute("escape_chars");
  186. const char *args_name = element->Attribute("args");
  187. const char *args_help = element->Attribute("args_help");
  188. const char *lock = element->Attribute("lock");
  189. clish_command_t *old = clish_view_find_command(v, name, BOOL_FALSE);
  190. // check this command doesn't already exist
  191. if (NULL != old) {
  192. // flag the duplication then ignore further definition
  193. printf("DUPLICATE COMMAND: %s\n",
  194. clish_command__get_name(old));
  195. } else {
  196. assert(name);
  197. assert(help);
  198. /* create a command */
  199. cmd = clish_view_new_command(v, name, help);
  200. assert(cmd);
  201. clish_command__set_pview(cmd, v);
  202. if (NULL != escape_chars) {
  203. /* define some specialist escape characters */
  204. clish_command__set_escape_chars(cmd,
  205. escape_chars);
  206. }
  207. if (NULL != args_name) {
  208. /* define a "rest of line" argument */
  209. clish_param_t *param;
  210. clish_ptype_t *tmp = NULL;
  211. assert(NULL != args_help);
  212. tmp = clish_shell_find_create_ptype(shell,
  213. "internal_ARGS",
  214. "Arguments", "[^\\]+",
  215. CLISH_PTYPE_REGEXP,
  216. CLISH_PTYPE_NONE);
  217. assert(tmp);
  218. param =
  219. clish_param_new(args_name, args_help, tmp);
  220. clish_command__set_args(cmd, param);
  221. }
  222. // define the view which this command changes to
  223. if (NULL != view) {
  224. clish_view_t *next =
  225. clish_shell_find_create_view(shell, view,
  226. NULL);
  227. // reference the next view
  228. clish_command__set_view(cmd, next);
  229. }
  230. // define the view id which this command changes to
  231. if (NULL != viewid) {
  232. clish_command__set_viewid(cmd, viewid);
  233. }
  234. /* lock field */
  235. if (lock && (lub_string_nocasecmp(lock, "false") == 0))
  236. clish_command__set_lock(cmd, BOOL_FALSE);
  237. else
  238. clish_command__set_lock(cmd, BOOL_TRUE);
  239. process_children(shell, element, cmd);
  240. }
  241. }
  242. }
  243. ///////////////////////////////////////
  244. static void
  245. process_startup(clish_shell_t * shell, TiXmlElement * element, void *parent)
  246. {
  247. clish_view_t *v = (clish_view_t *) parent;
  248. clish_command_t *cmd = NULL;
  249. const char *view = element->Attribute("view");
  250. const char *viewid = element->Attribute("viewid");
  251. const char *default_shebang = element->Attribute("default_shebang");
  252. assert(NULL == shell->startup);
  253. assert(view);
  254. /* create a command with NULL help */
  255. cmd = clish_view_new_command(v, "startup", NULL);
  256. clish_command__set_lock(cmd, BOOL_FALSE);
  257. // define the view which this command changes to
  258. clish_view_t *next = clish_shell_find_create_view(shell, view, NULL);
  259. // reference the next view
  260. clish_command__set_view(cmd, next);
  261. // define the view id which this command changes to
  262. if (NULL != viewid) {
  263. clish_command__set_viewid(cmd, viewid);
  264. }
  265. if (NULL != default_shebang)
  266. clish_shell__set_default_shebang(shell, default_shebang);
  267. // remember this command
  268. shell->startup = cmd;
  269. process_children(shell, element, cmd);
  270. }
  271. ///////////////////////////////////////
  272. static void
  273. process_param(clish_shell_t * shell, TiXmlElement * element, void *parent)
  274. {
  275. clish_command_t *cmd = NULL;
  276. clish_param_t *p_param = NULL;
  277. if (0 == lub_string_nocasecmp(element->Parent()->Value(), "PARAM"))
  278. p_param = (clish_param_t *) parent;
  279. else
  280. cmd = (clish_command_t *) parent;
  281. if ((NULL != cmd) || (NULL != p_param)) {
  282. assert((!cmd) || (cmd != shell->startup));
  283. const char *name = element->Attribute("name");
  284. const char *help = element->Attribute("help");
  285. const char *ptype = element->Attribute("ptype");
  286. const char *prefix = element->Attribute("prefix");
  287. const char *defval = element->Attribute("default");
  288. const char *mode = element->Attribute("mode");
  289. const char *optional = element->Attribute("optional");
  290. const char *value = element->Attribute("value");
  291. const char *hidden = element->Attribute("hidden");
  292. const char *test = element->Attribute("test");
  293. clish_param_t *param;
  294. clish_ptype_t *tmp = NULL;
  295. // create a command
  296. assert(name);
  297. assert(help);
  298. assert(ptype);
  299. if (ptype && *ptype) {
  300. tmp = clish_shell_find_create_ptype(shell, ptype,
  301. NULL, NULL,
  302. CLISH_PTYPE_REGEXP,
  303. CLISH_PTYPE_NONE);
  304. assert(tmp);
  305. }
  306. param = clish_param_new(name, help, tmp);
  307. /* If prefix is set clish will emulate old optional
  308. * command syntax over newer optional command mechanism.
  309. * It will create nested PARAM.
  310. */
  311. if (NULL != prefix) {
  312. const char *ptype_name = "__SUBCOMMAND";
  313. clish_param_t *opt_param = NULL;
  314. /* Create a ptype for prefix-named subcommand that
  315. * will contain the nested optional parameter. The
  316. * name of ptype is hardcoded. It's not good but
  317. * it's only the service ptype.
  318. */
  319. tmp = (clish_ptype_t *)lub_bintree_find(
  320. &shell->ptype_tree, ptype_name);
  321. if (!tmp)
  322. tmp = clish_shell_find_create_ptype(shell,
  323. ptype_name, "Option", "[^\\]+",
  324. CLISH_PTYPE_REGEXP, CLISH_PTYPE_NONE);
  325. assert(tmp);
  326. opt_param = clish_param_new(prefix, help, tmp);
  327. clish_param__set_mode(opt_param,
  328. CLISH_PARAM_SUBCOMMAND);
  329. clish_param__set_optional(opt_param, BOOL_TRUE);
  330. if (NULL != test)
  331. clish_param__set_test(opt_param, test);
  332. if (cmd)
  333. // add the parameter to the command
  334. clish_command_insert_param(cmd, opt_param);
  335. if (p_param)
  336. // add the parameter to the param
  337. clish_param_insert_param(p_param, opt_param);
  338. /* Unset cmd and set parent param to opt_param */
  339. cmd = NULL;
  340. p_param = opt_param;
  341. }
  342. if (NULL != defval) {
  343. clish_param__set_default(param, defval);
  344. }
  345. if (hidden && (lub_string_nocasecmp(hidden, "true") == 0))
  346. clish_param__set_hidden(param, BOOL_TRUE);
  347. else
  348. clish_param__set_hidden(param, BOOL_FALSE);
  349. if (NULL != mode) {
  350. if (!lub_string_nocasecmp(mode, "switch")) {
  351. clish_param__set_mode(param,
  352. CLISH_PARAM_SWITCH);
  353. /* Force hidden attribute */
  354. clish_param__set_hidden(param, BOOL_TRUE);
  355. } else if (!lub_string_nocasecmp(mode, "subcommand"))
  356. clish_param__set_mode(param,
  357. CLISH_PARAM_SUBCOMMAND);
  358. else
  359. clish_param__set_mode(param,
  360. CLISH_PARAM_COMMON);
  361. }
  362. if (optional && (lub_string_nocasecmp(optional, "true") == 0))
  363. clish_param__set_optional(param, BOOL_TRUE);
  364. else
  365. clish_param__set_optional(param, BOOL_FALSE);
  366. if (NULL != value) {
  367. clish_param__set_value(param, value);
  368. /* Force mode to subcommand */
  369. clish_param__set_mode(param,
  370. CLISH_PARAM_SUBCOMMAND);
  371. }
  372. if ((NULL != test) && (NULL == prefix))
  373. clish_param__set_test(param, test);
  374. if (cmd)
  375. // add the parameter to the command
  376. clish_command_insert_param(cmd, param);
  377. if (p_param)
  378. // add the parameter to the param
  379. clish_param_insert_param(p_param, param);
  380. process_children(shell, element, param);
  381. }
  382. }
  383. ////////////////////////////////////////
  384. static void
  385. process_action(clish_shell_t * shell, TiXmlElement * element, void *parent)
  386. {
  387. clish_command_t *cmd = (clish_command_t *) parent;
  388. if (NULL != cmd) {
  389. // read the following text element
  390. TiXmlNode *text = element->FirstChild();
  391. const char *builtin = element->Attribute("builtin");
  392. const char *shebang = element->Attribute("shebang");
  393. if (NULL != text) {
  394. assert(TiXmlNode::TEXT == text->Type());
  395. // store the action
  396. clish_command__set_action(cmd, text->Value());
  397. }
  398. if (NULL != builtin) {
  399. // store the action
  400. clish_command__set_builtin(cmd, builtin);
  401. }
  402. if (NULL != shebang)
  403. clish_command__set_shebang(cmd, shebang);
  404. }
  405. }
  406. ////////////////////////////////////////
  407. static void
  408. process_detail(clish_shell_t * shell, TiXmlElement * element, void *parent)
  409. {
  410. clish_command_t *cmd = (clish_command_t *) parent;
  411. // read the following text element
  412. TiXmlNode *text = element->FirstChild();
  413. if ((NULL != cmd) && (NULL != text)) {
  414. assert(TiXmlNode::TEXT == text->Type());
  415. // store the action
  416. clish_command__set_detail(cmd, text->Value());
  417. }
  418. }
  419. ///////////////////////////////////////
  420. static void
  421. process_namespace(clish_shell_t * shell, TiXmlElement * element, void *parent)
  422. {
  423. clish_view_t *v = (clish_view_t *) parent;
  424. clish_nspace_t *nspace = NULL;
  425. const char *view = element->Attribute("ref");
  426. const char *prefix = element->Attribute("prefix");
  427. const char *prefix_help = element->Attribute("prefix_help");
  428. const char *help = element->Attribute("help");
  429. const char *completion = element->Attribute("completion");
  430. const char *context_help = element->Attribute("context_help");
  431. const char *inherit = element->Attribute("inherit");
  432. assert(view);
  433. clish_view_t *ref_view =
  434. clish_shell_find_create_view(shell, view, NULL);
  435. assert(ref_view);
  436. nspace = clish_nspace_new(ref_view);
  437. assert(nspace);
  438. clish_view_insert_nspace(v, nspace);
  439. if (NULL != prefix) {
  440. clish_nspace__set_prefix(nspace, prefix);
  441. if (prefix_help)
  442. clish_nspace_create_prefix_cmd(nspace,
  443. "prefix",
  444. prefix_help);
  445. else
  446. clish_nspace_create_prefix_cmd(nspace,
  447. "prefix",
  448. "Prefix for the imported commands.");
  449. }
  450. if (help && (lub_string_nocasecmp(help, "true") == 0))
  451. clish_nspace__set_help(nspace, BOOL_TRUE);
  452. else
  453. clish_nspace__set_help(nspace, BOOL_FALSE);
  454. if (completion && (lub_string_nocasecmp(completion, "false") == 0))
  455. clish_nspace__set_completion(nspace, BOOL_FALSE);
  456. else
  457. clish_nspace__set_completion(nspace, BOOL_TRUE);
  458. if (context_help && (lub_string_nocasecmp(context_help, "true") == 0))
  459. clish_nspace__set_context_help(nspace, BOOL_TRUE);
  460. else
  461. clish_nspace__set_context_help(nspace, BOOL_FALSE);
  462. if (inherit && (lub_string_nocasecmp(inherit, "false") == 0))
  463. clish_nspace__set_inherit(nspace, BOOL_FALSE);
  464. else
  465. clish_nspace__set_inherit(nspace, BOOL_TRUE);
  466. }
  467. ////////////////////////////////////////
  468. static void
  469. process_config(clish_shell_t * shell, TiXmlElement * element, void *parent)
  470. {
  471. clish_command_t *cmd = (clish_command_t *) parent;
  472. if (NULL == cmd)
  473. return;
  474. // read the following text element
  475. const char *operation = element->Attribute("operation");
  476. const char *priority = element->Attribute("priority");
  477. const char *pattern = element->Attribute("pattern");
  478. const char *file = element->Attribute("file");
  479. const char *splitter = element->Attribute("splitter");
  480. const char *seq = element->Attribute("sequence");
  481. const char *unique = element->Attribute("unique");
  482. const char *cfg_depth = element->Attribute("depth");
  483. if (operation && !lub_string_nocasecmp(operation, "unset"))
  484. clish_command__set_cfg_op(cmd, CLISH_CONFIG_UNSET);
  485. else if (operation && !lub_string_nocasecmp(operation, "none"))
  486. clish_command__set_cfg_op(cmd, CLISH_CONFIG_NONE);
  487. else if (operation && !lub_string_nocasecmp(operation, "dump"))
  488. clish_command__set_cfg_op(cmd, CLISH_CONFIG_DUMP);
  489. else {
  490. clish_command__set_cfg_op(cmd, CLISH_CONFIG_SET);
  491. /* The priority if no clearly specified */
  492. clish_command__set_priority(cmd, 0x7f00);
  493. }
  494. if ((priority != NULL) && (*priority != '\0')) {
  495. long val = 0;
  496. char *endptr;
  497. unsigned short pri;
  498. val = strtol(priority, &endptr, 0);
  499. if (endptr == priority)
  500. pri = 0;
  501. else if (val > 0xffff)
  502. pri = 0xffff;
  503. else if (val < 0)
  504. pri = 0;
  505. else
  506. pri = (unsigned short)val;
  507. clish_command__set_priority(cmd, pri);
  508. }
  509. if (pattern != NULL)
  510. clish_command__set_pattern(cmd, pattern);
  511. else
  512. clish_command__set_pattern(cmd, "^${__cmd}");
  513. if (file != NULL)
  514. clish_command__set_file(cmd, file);
  515. if (splitter && (lub_string_nocasecmp(splitter, "false") == 0))
  516. clish_command__set_splitter(cmd, BOOL_FALSE);
  517. else
  518. clish_command__set_splitter(cmd, BOOL_TRUE);
  519. if (unique && (lub_string_nocasecmp(unique, "false") == 0))
  520. clish_command__set_unique(cmd, BOOL_FALSE);
  521. else
  522. clish_command__set_unique(cmd, BOOL_TRUE);
  523. if (seq != NULL)
  524. clish_command__set_seq(cmd, seq);
  525. else
  526. /* The entries without sequence cannot be non-unique */
  527. clish_command__set_unique(cmd, BOOL_TRUE);
  528. if (cfg_depth != NULL)
  529. clish_command__set_cfg_depth(cmd, cfg_depth);
  530. }
  531. ///////////////////////////////////////
  532. int clish_shell_xml_read(clish_shell_t * shell, const char *filename)
  533. {
  534. int ret = -1;
  535. TiXmlDocument doc;
  536. // keep the white space
  537. TiXmlBase::SetCondenseWhiteSpace(false);
  538. if (doc.LoadFile(filename)) {
  539. TiXmlNode *child = 0;
  540. while ((child = doc.IterateChildren(child))) {
  541. process_node(shell, child, NULL);
  542. }
  543. ret = 0;
  544. } else {
  545. printf("Unable to open %s\n", filename);
  546. }
  547. return ret;
  548. }
  549. ///////////////////////////////////////