syms.c 25 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177
  1. #include <stdlib.h>
  2. #include <stdio.h>
  3. #include <string.h>
  4. #include <assert.h>
  5. #include <syslog.h>
  6. #include <faux/faux.h>
  7. #include <faux/str.h>
  8. #include <faux/argv.h>
  9. #include <faux/list.h>
  10. #include <faux/error.h>
  11. #include <klish/khelper.h>
  12. #include <klish/kplugin.h>
  13. #include <klish/kentry.h>
  14. #include <klish/kscheme.h>
  15. #include <klish/kcontext.h>
  16. #include <klish/kpargv.h>
  17. #include <sysrepo.h>
  18. #include <sysrepo/xpath.h>
  19. #include <sysrepo/values.h>
  20. #include "pline.h"
  21. #include "private.h"
  22. static faux_argv_t *param2argv(const faux_argv_t *cur_path,
  23. const kpargv_t *pargv, const char *entry_name)
  24. {
  25. faux_list_node_t *iter = NULL;
  26. faux_list_t *pargs = NULL;
  27. faux_argv_t *args = NULL;
  28. kparg_t *parg = NULL;
  29. assert(pargv);
  30. if (!pargv)
  31. return NULL;
  32. pargs = kpargv_find_multi(pargv, entry_name);
  33. if (cur_path)
  34. args = faux_argv_dup(cur_path);
  35. else
  36. args = faux_argv_new();
  37. iter = faux_list_head(pargs);
  38. while ((parg = (kparg_t *)faux_list_each(&iter))) {
  39. faux_argv_add(args, kparg_value(parg));
  40. }
  41. faux_list_free(pargs);
  42. return args;
  43. }
  44. // Candidate from pargv contains possible begin of current word (that must be
  45. // completed). kpargv's list don't contain candidate but only already parsed
  46. // words.
  47. static int srp_compl_or_help(kcontext_t *context, bool_t help)
  48. {
  49. faux_argv_t *args = NULL;
  50. pline_t *pline = NULL;
  51. sr_conn_ctx_t *conn = NULL;
  52. sr_session_ctx_t *sess = NULL;
  53. const char *entry_name = NULL;
  54. faux_argv_t *cur_path = NULL;
  55. assert(context);
  56. if (sr_connect(SR_CONN_DEFAULT, &conn))
  57. return -1;
  58. if (sr_session_start(conn, SRP_REPO_EDIT, &sess)) {
  59. sr_disconnect(conn);
  60. return -1;
  61. }
  62. cur_path = (faux_argv_t *)srp_udata_path(context);
  63. entry_name = kentry_name(kcontext_candidate_entry(context));
  64. args = param2argv(cur_path, kcontext_parent_pargv(context), entry_name);
  65. pline = pline_parse(sess, args, srp_udata_flags(context));
  66. faux_argv_free(args);
  67. pline_print_completions(pline, help);
  68. pline_free(pline);
  69. sr_disconnect(conn);
  70. return 0;
  71. }
  72. int srp_compl(kcontext_t *context)
  73. {
  74. return srp_compl_or_help(context, BOOL_FALSE);
  75. }
  76. int srp_help(kcontext_t *context)
  77. {
  78. return srp_compl_or_help(context, BOOL_TRUE);
  79. }
  80. int srp_prompt_edit_path(kcontext_t *context)
  81. {
  82. faux_argv_t *cur_path = NULL;
  83. char *path = NULL;
  84. assert(context);
  85. cur_path = (faux_argv_t *)srp_udata_path(context);
  86. if (cur_path)
  87. path = faux_argv_line(cur_path);
  88. printf("[edit%s%s]\n", path ? " " : "", path ? path : "");
  89. faux_str_free(path);
  90. return 0;
  91. }
  92. static int srp_check_type(kcontext_t *context,
  93. pt_e not_accepted_nodes,
  94. size_t max_expr_num)
  95. {
  96. int ret = -1;
  97. faux_argv_t *args = NULL;
  98. pline_t *pline = NULL;
  99. sr_conn_ctx_t *conn = NULL;
  100. sr_session_ctx_t *sess = NULL;
  101. const char *entry_name = NULL;
  102. const char *value = NULL;
  103. pexpr_t *expr = NULL;
  104. size_t expr_num = 0;
  105. faux_argv_t *cur_path = NULL;
  106. assert(context);
  107. if (sr_connect(SR_CONN_DEFAULT, &conn))
  108. return -1;
  109. if (sr_session_start(conn, SRP_REPO_EDIT, &sess)) {
  110. sr_disconnect(conn);
  111. return -1;
  112. }
  113. cur_path = (faux_argv_t *)srp_udata_path(context);
  114. entry_name = kentry_name(kcontext_candidate_entry(context));
  115. value = kcontext_candidate_value(context);
  116. args = param2argv(cur_path, kcontext_parent_pargv(context), entry_name);
  117. if (value)
  118. faux_argv_add(args, value);
  119. pline = pline_parse(sess, args, srp_udata_flags(context));
  120. faux_argv_free(args);
  121. if (pline->invalid)
  122. goto err;
  123. expr_num = faux_list_len(pline->exprs);
  124. if (expr_num < 1)
  125. goto err;
  126. if ((max_expr_num > 0) && // '0' means unlimited
  127. (expr_num > max_expr_num))
  128. goto err;
  129. expr = pline_current_expr(pline);
  130. if (expr->pat & not_accepted_nodes)
  131. goto err;
  132. ret = 0;
  133. err:
  134. pline_free(pline);
  135. sr_disconnect(conn);
  136. return ret;
  137. }
  138. int srp_PLINE_SET(kcontext_t *context)
  139. {
  140. return srp_check_type(context, PT_NOT_SET, 0);
  141. }
  142. int srp_PLINE_DEL(kcontext_t *context)
  143. {
  144. return srp_check_type(context, PT_NOT_DEL, 1);
  145. }
  146. int srp_PLINE_EDIT(kcontext_t *context)
  147. {
  148. return srp_check_type(context, PT_NOT_EDIT, 1);
  149. }
  150. int srp_PLINE_INSERT_FROM(kcontext_t *context)
  151. {
  152. return srp_check_type(context, PT_NOT_INSERT, 1);
  153. }
  154. static faux_argv_t *assemble_insert_to(sr_session_ctx_t *sess, const kpargv_t *pargv,
  155. faux_argv_t *cur_path, const char *candidate_value, uint32_t flags)
  156. {
  157. faux_argv_t *args = NULL;
  158. faux_argv_t *insert_to = NULL;
  159. pline_t *pline = NULL;
  160. pexpr_t *expr = NULL;
  161. size_t i = 0;
  162. assert(sess);
  163. args = param2argv(cur_path, pargv, "from_path");
  164. pline = pline_parse(sess, args, flags);
  165. expr = pline_current_expr(pline);
  166. for (i = 0; i < (expr->args_num - expr->list_pos); i++) {
  167. faux_argv_node_t *iter = faux_argv_iterr(args);
  168. faux_argv_del(args, iter);
  169. }
  170. insert_to = param2argv(args, pargv, "to_path");
  171. faux_argv_free(args);
  172. if (candidate_value)
  173. faux_argv_add(insert_to, candidate_value);
  174. pline_free(pline);
  175. return insert_to;
  176. }
  177. int srp_PLINE_INSERT_TO(kcontext_t *context)
  178. {
  179. int ret = -1;
  180. faux_argv_t *args = NULL;
  181. pline_t *pline = NULL;
  182. sr_conn_ctx_t *conn = NULL;
  183. sr_session_ctx_t *sess = NULL;
  184. const char *value = NULL;
  185. pexpr_t *expr = NULL;
  186. size_t expr_num = 0;
  187. faux_argv_t *cur_path = NULL;
  188. assert(context);
  189. if (sr_connect(SR_CONN_DEFAULT, &conn))
  190. return -1;
  191. if (sr_session_start(conn, SRP_REPO_EDIT, &sess)) {
  192. sr_disconnect(conn);
  193. return -1;
  194. }
  195. cur_path = (faux_argv_t *)srp_udata_path(context);
  196. value = kcontext_candidate_value(context);
  197. args = assemble_insert_to(sess, kcontext_parent_pargv(context),
  198. cur_path, value, srp_udata_flags(context));
  199. pline = pline_parse(sess, args, srp_udata_flags(context));
  200. faux_argv_free(args);
  201. if (pline->invalid)
  202. goto err;
  203. expr_num = faux_list_len(pline->exprs);
  204. if (expr_num != 1)
  205. goto err;
  206. expr = pline_current_expr(pline);
  207. if (expr->pat & PT_NOT_INSERT)
  208. goto err;
  209. ret = 0;
  210. err:
  211. pline_free(pline);
  212. sr_disconnect(conn);
  213. return ret;
  214. }
  215. static int srp_compl_or_help_insert_to(kcontext_t *context, bool_t help)
  216. {
  217. faux_argv_t *args = NULL;
  218. pline_t *pline = NULL;
  219. sr_conn_ctx_t *conn = NULL;
  220. sr_session_ctx_t *sess = NULL;
  221. faux_argv_t *cur_path = NULL;
  222. assert(context);
  223. if (sr_connect(SR_CONN_DEFAULT, &conn))
  224. return -1;
  225. if (sr_session_start(conn, SRP_REPO_EDIT, &sess)) {
  226. sr_disconnect(conn);
  227. return -1;
  228. }
  229. cur_path = (faux_argv_t *)srp_udata_path(context);
  230. args = assemble_insert_to(sess, kcontext_parent_pargv(context),
  231. cur_path, NULL, srp_udata_flags(context));
  232. pline = pline_parse(sess, args, srp_udata_flags(context));
  233. faux_argv_free(args);
  234. pline_print_completions(pline, help);
  235. pline_free(pline);
  236. sr_disconnect(conn);
  237. return 0;
  238. }
  239. int srp_compl_insert_to(kcontext_t *context)
  240. {
  241. return srp_compl_or_help_insert_to(context, BOOL_FALSE);
  242. }
  243. int srp_help_insert_to(kcontext_t *context)
  244. {
  245. return srp_compl_or_help_insert_to(context, BOOL_TRUE);
  246. }
  247. int srp_set(kcontext_t *context)
  248. {
  249. int ret = 0;
  250. faux_argv_t *args = NULL;
  251. pline_t *pline = NULL;
  252. sr_conn_ctx_t *conn = NULL;
  253. sr_session_ctx_t *sess = NULL;
  254. faux_list_node_t *iter = NULL;
  255. pexpr_t *expr = NULL;
  256. size_t err_num = 0;
  257. faux_argv_t *cur_path = NULL;
  258. assert(context);
  259. if (sr_connect(SR_CONN_DEFAULT, &conn))
  260. return -1;
  261. if (sr_session_start(conn, SRP_REPO_EDIT, &sess)) {
  262. sr_disconnect(conn);
  263. return -1;
  264. }
  265. cur_path = (faux_argv_t *)srp_udata_path(context);
  266. args = param2argv(cur_path, kcontext_pargv(context), "path");
  267. pline = pline_parse(sess, args, srp_udata_flags(context));
  268. faux_argv_free(args);
  269. if (pline->invalid) {
  270. fprintf(stderr, "Invalid set request\n");
  271. ret = -1;
  272. goto cleanup;
  273. }
  274. iter = faux_list_head(pline->exprs);
  275. while ((expr = (pexpr_t *)faux_list_each(&iter))) {
  276. if (!(expr->pat & PT_SET)) {
  277. err_num++;
  278. fprintf(stderr, "Illegal expression for set operation\n");
  279. break;
  280. }
  281. if (sr_set_item_str(sess, expr->xpath, expr->value, NULL, 0) !=
  282. SR_ERR_OK) {
  283. err_num++;
  284. fprintf(stderr, "Can't set data\n");
  285. break;
  286. }
  287. }
  288. if (sr_has_changes(sess)) {
  289. if (err_num > 0)
  290. sr_discard_changes(sess);
  291. else
  292. sr_apply_changes(sess, 0);
  293. }
  294. if (err_num > 0)
  295. ret = -1;
  296. cleanup:
  297. pline_free(pline);
  298. sr_disconnect(conn);
  299. return ret;
  300. }
  301. int srp_del(kcontext_t *context)
  302. {
  303. int ret = -1;
  304. faux_argv_t *args = NULL;
  305. pline_t *pline = NULL;
  306. sr_conn_ctx_t *conn = NULL;
  307. sr_session_ctx_t *sess = NULL;
  308. pexpr_t *expr = NULL;
  309. size_t err_num = 0;
  310. faux_argv_t *cur_path = NULL;
  311. assert(context);
  312. if (sr_connect(SR_CONN_DEFAULT, &conn))
  313. return -1;
  314. if (sr_session_start(conn, SRP_REPO_EDIT, &sess)) {
  315. sr_disconnect(conn);
  316. return -1;
  317. }
  318. cur_path = (faux_argv_t *)srp_udata_path(context);
  319. args = param2argv(cur_path, kcontext_pargv(context), "path");
  320. pline = pline_parse(sess, args, srp_udata_flags(context));
  321. faux_argv_free(args);
  322. if (pline->invalid) {
  323. fprintf(stderr, "Invalid 'del' request\n");
  324. goto err;
  325. }
  326. if (faux_list_len(pline->exprs) > 1) {
  327. fprintf(stderr, "Can't delete more than one object\n");
  328. goto err;
  329. }
  330. expr = (pexpr_t *)faux_list_data(faux_list_head(pline->exprs));
  331. if (!(expr->pat & PT_DEL)) {
  332. fprintf(stderr, "Illegal expression for 'del' operation\n");
  333. goto err;
  334. }
  335. if (sr_delete_item(sess, expr->xpath, 0) != SR_ERR_OK) {
  336. fprintf(stderr, "Can't delete data\n");
  337. goto err;
  338. }
  339. sr_apply_changes(sess, 0);
  340. ret = 0;
  341. err:
  342. pline_free(pline);
  343. sr_disconnect(conn);
  344. return ret;
  345. }
  346. int srp_edit(kcontext_t *context)
  347. {
  348. int ret = -1;
  349. faux_argv_t *args = NULL;
  350. pline_t *pline = NULL;
  351. sr_conn_ctx_t *conn = NULL;
  352. sr_session_ctx_t *sess = NULL;
  353. pexpr_t *expr = NULL;
  354. size_t err_num = 0;
  355. faux_argv_t *cur_path = NULL;
  356. assert(context);
  357. if (sr_connect(SR_CONN_DEFAULT, &conn))
  358. return -1;
  359. if (sr_session_start(conn, SRP_REPO_EDIT, &sess)) {
  360. sr_disconnect(conn);
  361. return -1;
  362. }
  363. cur_path = (faux_argv_t *)srp_udata_path(context);
  364. args = param2argv(cur_path, kcontext_pargv(context), "path");
  365. pline = pline_parse(sess, args, srp_udata_flags(context));
  366. if (pline->invalid) {
  367. fprintf(stderr, "Invalid 'edit' request\n");
  368. goto err;
  369. }
  370. if (faux_list_len(pline->exprs) > 1) {
  371. fprintf(stderr, "Can't process more than one object\n");
  372. goto err;
  373. }
  374. expr = (pexpr_t *)faux_list_data(faux_list_head(pline->exprs));
  375. if (!(expr->pat & PT_EDIT)) {
  376. fprintf(stderr, "Illegal expression for 'edit' operation\n");
  377. goto err;
  378. }
  379. if (sr_set_item_str(sess, expr->xpath, NULL, NULL, 0) != SR_ERR_OK) {
  380. fprintf(stderr, "Can't set editing data\n");
  381. goto err;
  382. }
  383. sr_apply_changes(sess, 0);
  384. // Set new current path
  385. srp_udata_set_path(context, args);
  386. ret = 0;
  387. err:
  388. if (ret < 0)
  389. faux_argv_free(args);
  390. pline_free(pline);
  391. sr_disconnect(conn);
  392. return ret;
  393. }
  394. int srp_top(kcontext_t *context)
  395. {
  396. assert(context);
  397. srp_udata_set_path(context, NULL);
  398. return 0;
  399. }
  400. int srp_up(kcontext_t *context)
  401. {
  402. sr_conn_ctx_t *conn = NULL;
  403. sr_session_ctx_t *sess = NULL;
  404. faux_argv_t *cur_path = NULL;
  405. faux_argv_node_t *iter = NULL;
  406. assert(context);
  407. cur_path = (faux_argv_t *)srp_udata_path(context);
  408. if (!cur_path)
  409. return -1; // It's top level and can't level up
  410. if (sr_connect(SR_CONN_DEFAULT, &conn))
  411. return -1;
  412. if (sr_session_start(conn, SRP_REPO_EDIT, &sess)) {
  413. sr_disconnect(conn);
  414. return -1;
  415. }
  416. // Remove last arguments one by one and wait for legal edit-like pline
  417. while (faux_argv_len(cur_path) > 0) {
  418. pline_t *pline = NULL;
  419. pexpr_t *expr = NULL;
  420. size_t len = 0;
  421. iter = faux_argv_iterr(cur_path);
  422. faux_argv_del(cur_path, iter);
  423. pline = pline_parse(sess, cur_path, srp_udata_flags(context));
  424. if (pline->invalid) {
  425. pline_free(pline);
  426. continue;
  427. }
  428. len = faux_list_len(pline->exprs);
  429. if (len != 1) {
  430. pline_free(pline);
  431. continue;
  432. }
  433. expr = (pexpr_t *)faux_list_data(faux_list_head(pline->exprs));
  434. if (!(expr->pat & PT_EDIT)) {
  435. pline_free(pline);
  436. continue;
  437. }
  438. // Here new path is ok
  439. pline_free(pline);
  440. break;
  441. }
  442. // Don't store empty path
  443. if (faux_argv_len(cur_path) == 0)
  444. srp_udata_set_path(context, NULL);
  445. sr_disconnect(conn);
  446. return 0;
  447. }
  448. int srp_insert(kcontext_t *context)
  449. {
  450. int ret = -1;
  451. pline_t *pline = NULL;
  452. pline_t *pline_to = NULL;
  453. sr_conn_ctx_t *conn = NULL;
  454. sr_session_ctx_t *sess = NULL;
  455. faux_list_node_t *iter = NULL;
  456. pexpr_t *expr = NULL;
  457. pexpr_t *expr_to = NULL;
  458. faux_argv_t *cur_path = NULL;
  459. faux_argv_t *insert_from = NULL;
  460. faux_argv_t *insert_to = NULL;
  461. sr_move_position_t position = SR_MOVE_LAST;
  462. kpargv_t *pargv = NULL;
  463. const char *list_keys = NULL;
  464. const char *leaflist_value = NULL;
  465. assert(context);
  466. if (sr_connect(SR_CONN_DEFAULT, &conn))
  467. return -1;
  468. if (sr_session_start(conn, SRP_REPO_EDIT, &sess)) {
  469. sr_disconnect(conn);
  470. return -1;
  471. }
  472. cur_path = (faux_argv_t *)srp_udata_path(context);
  473. pargv = kcontext_pargv(context);
  474. // 'from' argument
  475. insert_from = param2argv(cur_path, pargv, "from_path");
  476. pline = pline_parse(sess, insert_from, srp_udata_flags(context));
  477. faux_argv_free(insert_from);
  478. if (pline->invalid) {
  479. fprintf(stderr, "Invalid 'from' expression\n");
  480. goto err;
  481. }
  482. if (faux_list_len(pline->exprs) > 1) {
  483. fprintf(stderr, "Can't process more than one object\n");
  484. goto err;
  485. }
  486. expr = (pexpr_t *)faux_list_data(faux_list_head(pline->exprs));
  487. if (!(expr->pat & PT_INSERT)) {
  488. fprintf(stderr, "Illegal 'from' expression for 'insert' operation\n");
  489. goto err;
  490. }
  491. // Position
  492. if (kpargv_find(pargv, "first"))
  493. position = SR_MOVE_FIRST;
  494. else if (kpargv_find(pargv, "last"))
  495. position = SR_MOVE_LAST;
  496. else if (kpargv_find(pargv, "before"))
  497. position = SR_MOVE_BEFORE;
  498. else if (kpargv_find(pargv, "after"))
  499. position = SR_MOVE_AFTER;
  500. else {
  501. fprintf(stderr, "Illegal 'position' argument\n");
  502. goto err;
  503. }
  504. // 'to' argument
  505. if ((SR_MOVE_BEFORE == position) || (SR_MOVE_AFTER == position)) {
  506. insert_to = assemble_insert_to(sess, pargv, cur_path,
  507. NULL, srp_udata_flags(context));
  508. pline_to = pline_parse(sess, insert_to, srp_udata_flags(context));
  509. faux_argv_free(insert_to);
  510. if (pline_to->invalid) {
  511. fprintf(stderr, "Invalid 'to' expression\n");
  512. goto err;
  513. }
  514. if (faux_list_len(pline_to->exprs) > 1) {
  515. fprintf(stderr, "Can't process more than one object\n");
  516. goto err;
  517. }
  518. expr_to = (pexpr_t *)faux_list_data(faux_list_head(pline_to->exprs));
  519. if (!(expr_to->pat & PT_INSERT)) {
  520. fprintf(stderr, "Illegal 'to' expression for 'insert' operation\n");
  521. goto err;
  522. }
  523. if (PAT_LIST_KEY == expr_to->pat)
  524. list_keys = expr_to->last_keys;
  525. else // PATH_LEAFLIST_VALUE
  526. leaflist_value = expr_to->last_keys;
  527. }
  528. if (sr_move_item(sess, expr->xpath, position,
  529. list_keys, leaflist_value, NULL, 0) != SR_ERR_OK) {
  530. fprintf(stderr, "Can't move element\n");
  531. goto err;
  532. }
  533. sr_apply_changes(sess, 0);
  534. ret = 0;
  535. err:
  536. pline_free(pline);
  537. pline_free(pline_to);
  538. sr_disconnect(conn);
  539. return ret;
  540. }
  541. int srp_verify(kcontext_t *context)
  542. {
  543. int ret = -1;
  544. sr_conn_ctx_t *conn = NULL;
  545. sr_session_ctx_t *sess = NULL;
  546. assert(context);
  547. if (sr_connect(SR_CONN_DEFAULT, &conn)) {
  548. fprintf(stderr, "Can't connect to data repository\n");
  549. return -1;
  550. }
  551. if (sr_session_start(conn, SRP_REPO_EDIT, &sess)) {
  552. fprintf(stderr, "Can't connect to candidate data store\n");
  553. goto err;
  554. }
  555. // Validate candidate config
  556. if (sr_validate(sess, NULL, 0) != SR_ERR_OK) {
  557. fprintf(stderr, "Invalid candidate configuration\n");
  558. goto err;
  559. }
  560. ret = 0;
  561. err:
  562. sr_disconnect(conn);
  563. return ret;
  564. }
  565. int srp_commit(kcontext_t *context)
  566. {
  567. int ret = -1;
  568. sr_conn_ctx_t *conn = NULL;
  569. sr_session_ctx_t *sess = NULL;
  570. assert(context);
  571. if (sr_connect(SR_CONN_DEFAULT, &conn)) {
  572. fprintf(stderr, "Can't connect to data repository\n");
  573. return -1;
  574. }
  575. if (sr_session_start(conn, SRP_REPO_EDIT, &sess)) {
  576. fprintf(stderr, "Can't connect to candidate data store\n");
  577. goto err;
  578. }
  579. // Copy candidate to running-config
  580. if (sr_session_switch_ds(sess, SR_DS_RUNNING)) {
  581. fprintf(stderr, "Can't connect to running-config data store\n");
  582. goto err;
  583. }
  584. if (sr_copy_config(sess, NULL, SRP_REPO_EDIT, 0) != SR_ERR_OK) {
  585. fprintf(stderr, "Can't commit to running-config\n");
  586. goto err;
  587. }
  588. // Copy running-config to startup-config
  589. if (sr_session_switch_ds(sess, SR_DS_STARTUP)) {
  590. fprintf(stderr, "Can't connect to startup-config data store\n");
  591. goto err;
  592. }
  593. if (sr_copy_config(sess, NULL, SR_DS_RUNNING, 0) != SR_ERR_OK) {
  594. fprintf(stderr, "Can't store data to startup-config\n");
  595. goto err;
  596. }
  597. ret = 0;
  598. err:
  599. sr_disconnect(conn);
  600. return ret;
  601. }
  602. int srp_rollback(kcontext_t *context)
  603. {
  604. int ret = -1;
  605. sr_conn_ctx_t *conn = NULL;
  606. sr_session_ctx_t *sess = NULL;
  607. assert(context);
  608. if (sr_connect(SR_CONN_DEFAULT, &conn)) {
  609. fprintf(stderr, "Can't connect to data repository\n");
  610. return -1;
  611. }
  612. // Copy running-config to candidate config
  613. if (sr_session_start(conn, SRP_REPO_EDIT, &sess)) {
  614. fprintf(stderr, "Can't connect to candidate data store\n");
  615. goto err;
  616. }
  617. if (sr_copy_config(sess, NULL, SR_DS_RUNNING, 0) != SR_ERR_OK) {
  618. fprintf(stderr, "Can't rollback to running-config\n");
  619. goto err;
  620. }
  621. ret = 0;
  622. err:
  623. sr_disconnect(conn);
  624. return ret;
  625. }
  626. int srp_show_xml(kcontext_t *context)
  627. {
  628. int ret = -1;
  629. faux_argv_t *args = NULL;
  630. pline_t *pline = NULL;
  631. sr_conn_ctx_t *conn = NULL;
  632. sr_session_ctx_t *sess = NULL;
  633. pexpr_t *expr = NULL;
  634. faux_argv_t *cur_path = NULL;
  635. sr_data_t *data = NULL;
  636. struct ly_out *out = NULL;
  637. struct lyd_node *child = NULL;
  638. assert(context);
  639. if (sr_connect(SR_CONN_DEFAULT, &conn))
  640. return -1;
  641. if (sr_session_start(conn, SRP_REPO_EDIT, &sess)) {
  642. sr_disconnect(conn);
  643. return -1;
  644. }
  645. cur_path = (faux_argv_t *)srp_udata_path(context);
  646. args = param2argv(cur_path, kcontext_pargv(context), "path");
  647. pline = pline_parse(sess, args, srp_udata_flags(context));
  648. faux_argv_free(args);
  649. if (pline->invalid) {
  650. fprintf(stderr, "Invalid 'show' request\n");
  651. goto err;
  652. }
  653. if (faux_list_len(pline->exprs) > 1) {
  654. fprintf(stderr, "Can't process more than one object\n");
  655. goto err;
  656. }
  657. expr = (pexpr_t *)faux_list_data(faux_list_head(pline->exprs));
  658. if (!(expr->pat & PT_EDIT)) {
  659. fprintf(stderr, "Illegal expression for 'show' operation\n");
  660. goto err;
  661. }
  662. if (sr_get_subtree(sess, expr->xpath, 0, &data) != SR_ERR_OK) {
  663. fprintf(stderr, "Can't get specified subtree\n");
  664. goto err;
  665. }
  666. ly_out_new_file(stdout, &out);
  667. lyd_print_tree(out, data->tree, LYD_XML, 0);
  668. ly_out_free(out, NULL, 0);
  669. // child = lyd_child(data->tree);
  670. // if (child) {
  671. // ly_out_new_file(stdout, &out);
  672. // lyd_print_all(out, child, LYD_XML, 0);
  673. // }
  674. struct lyd_meta *meta = lyd_find_meta(data->tree->meta, NULL, "junos-configuration-metadata:active");
  675. if (meta)
  676. printf("META\n");
  677. sr_release_data(data);
  678. ret = 0;
  679. err:
  680. pline_free(pline);
  681. sr_disconnect(conn);
  682. return ret;
  683. }
  684. static int show(kcontext_t *context, sr_datastore_t ds)
  685. {
  686. int ret = -1;
  687. faux_argv_t *args = NULL;
  688. pline_t *pline = NULL;
  689. sr_conn_ctx_t *conn = NULL;
  690. sr_session_ctx_t *sess = NULL;
  691. pexpr_t *expr = NULL;
  692. faux_argv_t *cur_path = NULL;
  693. char *xpath = NULL;
  694. assert(context);
  695. if (sr_connect(SR_CONN_DEFAULT, &conn))
  696. return -1;
  697. if (sr_session_start(conn, ds, &sess)) {
  698. sr_disconnect(conn);
  699. return -1;
  700. }
  701. cur_path = (faux_argv_t *)srp_udata_path(context);
  702. if (kpargv_find(kcontext_pargv(context), "path") || cur_path) {
  703. args = param2argv(cur_path, kcontext_pargv(context), "path");
  704. pline = pline_parse(sess, args, srp_udata_flags(context));
  705. faux_argv_free(args);
  706. if (pline->invalid) {
  707. fprintf(stderr, "Invalid 'show' request\n");
  708. goto err;
  709. }
  710. if (faux_list_len(pline->exprs) > 1) {
  711. fprintf(stderr, "Can't process more than one object\n");
  712. goto err;
  713. }
  714. if (!(expr = (pexpr_t *)faux_list_data(faux_list_head(pline->exprs)))) {
  715. fprintf(stderr, "Can't get expression\n");
  716. goto err;
  717. }
  718. if (!(expr->pat & PT_EDIT)) {
  719. fprintf(stderr, "Illegal expression for 'show' operation\n");
  720. goto err;
  721. }
  722. if (!expr->xpath) {
  723. fprintf(stderr, "Empty expression for 'show' operation\n");
  724. goto err;
  725. }
  726. xpath = expr->xpath;
  727. }
  728. show_xpath(sess, xpath, srp_udata_flags(context));
  729. ret = 0;
  730. err:
  731. pline_free(pline);
  732. sr_disconnect(conn);
  733. return ret;
  734. }
  735. int srp_show(kcontext_t *context)
  736. {
  737. return show(context, SRP_REPO_EDIT);
  738. }
  739. int srp_show_running(kcontext_t *context)
  740. {
  741. sr_conn_ctx_t *conn = NULL;
  742. sr_session_ctx_t *sess = NULL;
  743. if (sr_connect(SR_CONN_DEFAULT, &conn))
  744. return -1;
  745. if (sr_session_start(conn, SR_DS_RUNNING, &sess)) {
  746. sr_disconnect(conn);
  747. return -1;
  748. }
  749. show_xpath(sess, NULL, srp_udata_flags(context));
  750. sr_disconnect(conn);
  751. return 0;
  752. }
  753. int srp_deactivate(kcontext_t *context)
  754. {
  755. int ret = -1;
  756. faux_argv_t *args = NULL;
  757. pline_t *pline = NULL;
  758. sr_conn_ctx_t *conn = NULL;
  759. sr_session_ctx_t *sess = NULL;
  760. pexpr_t *expr = NULL;
  761. faux_argv_t *cur_path = NULL;
  762. sr_data_t *data = NULL;
  763. const struct ly_ctx *ctx = NULL;
  764. assert(context);
  765. if (sr_connect(SR_CONN_DEFAULT, &conn))
  766. return -1;
  767. if (sr_session_start(conn, SRP_REPO_EDIT, &sess)) {
  768. sr_disconnect(conn);
  769. return -1;
  770. }
  771. cur_path = (faux_argv_t *)srp_udata_path(context);
  772. args = param2argv(cur_path, kcontext_pargv(context), "path");
  773. pline = pline_parse(sess, args, srp_udata_flags(context));
  774. faux_argv_free(args);
  775. if (pline->invalid) {
  776. fprintf(stderr, "Invalid 'show' request\n");
  777. goto err;
  778. }
  779. if (faux_list_len(pline->exprs) > 1) {
  780. fprintf(stderr, "Can't process more than one object\n");
  781. goto err;
  782. }
  783. expr = (pexpr_t *)faux_list_data(faux_list_head(pline->exprs));
  784. if (!(expr->pat & PT_DEL)) {
  785. fprintf(stderr, "Illegal expression for 'show' operation\n");
  786. goto err;
  787. }
  788. if (sr_get_subtree(sess, expr->xpath, 0, &data) != SR_ERR_OK) {
  789. fprintf(stderr, "Can't get specified subtree\n");
  790. goto err;
  791. }
  792. if (lyd_new_meta(LYD_CTX(data->tree), data->tree, NULL,
  793. "junos-configuration-metadata:active", "false", 0, NULL)) {
  794. fprintf(stderr, "Can't deactivate\n");
  795. goto err;
  796. }
  797. struct lyd_meta *meta = lyd_find_meta(data->tree->meta, NULL, "junos-configuration-metadata:active");
  798. if (meta)
  799. printf("META\n");
  800. if (sr_has_changes(sess))
  801. fprintf(stderr, "Has changes\n");
  802. if (sr_apply_changes(sess, 0)) {
  803. fprintf(stderr, "Can't apply changes\n");
  804. }
  805. sr_release_data(data);
  806. if (sr_get_subtree(sess, expr->xpath, 0, &data) != SR_ERR_OK) {
  807. fprintf(stderr, "Can't get specified subtree\n");
  808. goto err;
  809. }
  810. struct ly_out *out = NULL;
  811. ly_out_new_file(stdout, &out);
  812. lyd_print_tree(out, data->tree, LYD_XML, 0);
  813. ly_out_free(out, NULL, 0);
  814. sr_release_data(data);
  815. ret = 0;
  816. err:
  817. pline_free(pline);
  818. sr_disconnect(conn);
  819. return ret;
  820. }
  821. int srp_diff(kcontext_t *context)
  822. {
  823. int ret = -1;
  824. pline_t *pline = NULL;
  825. sr_conn_ctx_t *conn = NULL;
  826. sr_session_ctx_t *sess1 = NULL;
  827. sr_session_ctx_t *sess2 = NULL;
  828. sr_data_t *data1 = NULL;
  829. sr_data_t *data2 = NULL;
  830. faux_argv_t *cur_path = NULL;
  831. const char *xpath = NULL;
  832. struct lyd_node *diff = NULL;
  833. assert(context);
  834. if (sr_connect(SR_CONN_DEFAULT, &conn))
  835. return -1;
  836. if (sr_session_start(conn, SR_DS_RUNNING, &sess1)) {
  837. sr_disconnect(conn);
  838. return -1;
  839. }
  840. if (sr_session_start(conn, SRP_REPO_EDIT, &sess2)) {
  841. sr_disconnect(conn);
  842. return -1;
  843. }
  844. cur_path = (faux_argv_t *)srp_udata_path(context);
  845. if (kpargv_find(kcontext_pargv(context), "path") || cur_path) {
  846. faux_argv_t *args = NULL;
  847. pexpr_t *expr = NULL;
  848. args = param2argv(cur_path, kcontext_pargv(context), "path");
  849. pline = pline_parse(sess2, args, srp_udata_flags(context));
  850. faux_argv_free(args);
  851. if (pline->invalid) {
  852. fprintf(stderr, "Invalid 'show' request\n");
  853. goto err;
  854. }
  855. if (faux_list_len(pline->exprs) > 1) {
  856. fprintf(stderr, "Can't process more than one object\n");
  857. goto err;
  858. }
  859. if (!(expr = (pexpr_t *)faux_list_data(faux_list_head(pline->exprs)))) {
  860. fprintf(stderr, "Can't get expression\n");
  861. goto err;
  862. }
  863. if (!(expr->pat & PT_EDIT)) {
  864. fprintf(stderr, "Illegal expression for 'show' operation\n");
  865. goto err;
  866. }
  867. if (!expr->xpath) {
  868. fprintf(stderr, "Empty expression for 'show' operation\n");
  869. goto err;
  870. }
  871. xpath = expr->xpath;
  872. }
  873. if (!xpath)
  874. xpath = "/*";
  875. if (sr_get_data(sess1, xpath, 0, 0, 0, &data1) != SR_ERR_OK) {
  876. fprintf(stderr, "Can't get specified subtree\n");
  877. goto err;
  878. }
  879. if (sr_get_data(sess2, xpath, 0, 0, 0, &data2) != SR_ERR_OK) {
  880. fprintf(stderr, "Can't get specified subtree\n");
  881. goto err;
  882. }
  883. if (lyd_diff_siblings(data1->tree, data2->tree, 0, &diff) != LY_SUCCESS) {
  884. fprintf(stderr, "Can't generate diff\n");
  885. goto err;
  886. }
  887. show_subtree(diff, 0, DIFF_OP_NONE, srp_udata_flags(context));
  888. lyd_free_siblings(diff);
  889. ret = 0;
  890. err:
  891. if (data1)
  892. sr_release_data(data1);
  893. if (data2)
  894. sr_release_data(data2);
  895. pline_free(pline);
  896. sr_disconnect(conn);
  897. return ret;
  898. }
  899. static int srp_compl_xpath(kcontext_t *context, const sr_datastore_t datastore)
  900. {
  901. sr_conn_ctx_t *conn = NULL;
  902. sr_session_ctx_t *sess = NULL;
  903. sr_val_t *vals = NULL;
  904. size_t val_num = 0;
  905. size_t i = 0;
  906. const char *script = NULL;
  907. assert(context);
  908. script = kcontext_script(context);
  909. if (faux_str_is_empty(script))
  910. return -1;
  911. if (sr_connect(SR_CONN_DEFAULT, &conn))
  912. return -1;
  913. if (sr_session_start(conn, datastore, &sess)) {
  914. sr_disconnect(conn);
  915. return -1;
  916. }
  917. sr_get_items(sess, script, 0, 0, &vals, &val_num);
  918. for (i = 0; i < val_num; i++) {
  919. char *tmp = sr_val_to_str(&vals[i]);
  920. if (!tmp)
  921. continue;
  922. printf("%s\n", tmp);
  923. free(tmp);
  924. }
  925. sr_free_values(vals, val_num);
  926. sr_disconnect(conn);
  927. return 0;
  928. }
  929. int srp_compl_xpath_running(kcontext_t *context)
  930. {
  931. return srp_compl_xpath(context, SR_DS_RUNNING);
  932. }
  933. int srp_compl_xpath_candidate(kcontext_t *context)
  934. {
  935. return srp_compl_xpath(context, SR_DS_CANDIDATE);
  936. }