/* rpc parameter is freed after the function call */ static int op_send_recv(ncSessionObject *self, nc_rpc* rpc, char **data) { nc_reply *reply = NULL; int ret = EXIT_SUCCESS; /* send the request and get the reply */ switch (nc_session_send_recv(self->session, rpc, &reply)) { case NC_MSG_UNKNOWN: if (nc_session_get_status(self->session) != NC_SESSION_STATUS_WORKING) { PyErr_SetString(libnetconfError, "Session damaged, closing."); /* free the Session */ nc_session_free(self->session); self->session = NULL; } ret = EXIT_FAILURE; break; case NC_MSG_NONE: /* error occurred, but processed by callback */ break; case NC_MSG_REPLY: switch (nc_reply_get_type(reply)) { case NC_REPLY_OK: break; case NC_REPLY_DATA: if (data) { *data = nc_reply_get_data (reply); } break; case NC_REPLY_ERROR: ret = EXIT_FAILURE; break; default: PyErr_SetString(libnetconfError, "Unexpected operation result."); ret = EXIT_FAILURE; break; } break; default: PyErr_SetString(libnetconfError, "Unknown error occurred."); ret = EXIT_FAILURE; break; } nc_rpc_free(rpc); nc_reply_free(reply); return (ret); }
static void test_send_recv_error(void) { int ret; uint64_t msgid; NC_MSG_TYPE msgtype; struct nc_rpc *rpc; struct nc_reply *reply; struct nc_pollsession *ps; /* client RPC */ rpc = nc_rpc_kill(1); assert_non_null(rpc); msgtype = nc_send_rpc(client_session, rpc, 0, &msgid); assert_int_equal(msgtype, NC_MSG_RPC); /* server RPC, send reply */ ps = nc_ps_new(); assert_non_null(ps); nc_ps_add_session(ps, server_session); ret = nc_ps_poll(ps, 0); assert_int_equal(ret, 1); /* server finished */ nc_ps_free(ps); /* client reply */ msgtype = nc_recv_reply(client_session, rpc, msgid, 0, &reply); assert_int_equal(msgtype, NC_MSG_REPLY); nc_rpc_free(rpc); assert_int_equal(reply->type, NC_RPL_ERROR); assert_string_equal(((struct nc_reply_error *)reply)->err->tag, "operation-not-supported"); nc_reply_free(reply); }
static void test_send_recv_data(void) { int ret; uint64_t msgid; NC_MSG_TYPE msgtype; struct nc_rpc *rpc; struct nc_reply *reply; struct nc_pollsession *ps; /* client RPC */ rpc = nc_rpc_getconfig(NC_DATASTORE_RUNNING, NULL, 0, 0); assert_non_null(rpc); msgtype = nc_send_rpc(client_session, rpc, 0, &msgid); assert_int_equal(msgtype, NC_MSG_RPC); /* server RPC, send reply */ ps = nc_ps_new(); assert_non_null(ps); nc_ps_add_session(ps, server_session); ret = nc_ps_poll(ps, 0); assert_int_equal(ret, 1); /* server finished */ nc_ps_free(ps); /* client reply */ msgtype = nc_recv_reply(client_session, rpc, msgid, 0, &reply); assert_int_equal(msgtype, NC_MSG_REPLY); nc_rpc_free(rpc); assert_int_equal(reply->type, NC_RPL_DATA); nc_reply_free(reply); }
static PyObject *ncProcessRPC(ncSessionObject *self) { NC_MSG_TYPE ret; NC_RPC_TYPE req_type; NC_OP req_op; nc_rpc *rpc = NULL; nc_reply *reply = NULL; struct nc_err* e = NULL; SESSION_CHECK(self); /* receive incoming message */ ret = nc_session_recv_rpc(self->session, -1, &rpc); if (ret != NC_MSG_RPC) { if (nc_session_get_status(self->session) != NC_SESSION_STATUS_WORKING) { /* something really bad happend, and communication is not possible anymore */ nc_session_free(self->session); self->session = NULL; } Py_RETURN_NONE; } /* process it */ req_type = nc_rpc_get_type(rpc); req_op = nc_rpc_get_op(rpc); if (req_type == NC_RPC_SESSION) { /* process operations affectinf session */ switch(req_op) { case NC_OP_CLOSESESSION: /* exit the event loop immediately without processing any following request */ reply = nc_reply_ok(); break; case NC_OP_KILLSESSION: /* todo: kill the requested session */ reply = nc_reply_error(nc_err_new(NC_ERR_OP_NOT_SUPPORTED)); break; default: reply = nc_reply_error(nc_err_new(NC_ERR_OP_NOT_SUPPORTED)); break; } } else if (req_type == NC_RPC_DATASTORE_READ) { /* process operations reading datastore */ switch (req_op) { case NC_OP_GET: case NC_OP_GETCONFIG: reply = ncds_apply_rpc2all(self->session, rpc, NULL); break; default: reply = nc_reply_error(nc_err_new(NC_ERR_OP_NOT_SUPPORTED)); break; } } else if (req_type == NC_RPC_DATASTORE_WRITE) { /* process operations affecting datastore */ switch (req_op) { case NC_OP_LOCK: case NC_OP_UNLOCK: case NC_OP_COPYCONFIG: case NC_OP_DELETECONFIG: case NC_OP_EDITCONFIG: reply = ncds_apply_rpc2all(self->session, rpc, NULL); break; default: reply = nc_reply_error(nc_err_new(NC_ERR_OP_NOT_SUPPORTED)); break; } } else { /* process other operations */ reply = ncds_apply_rpc2all(self->session, rpc, NULL); } /* create reply */ if (reply == NULL) { reply = nc_reply_error(nc_err_new(NC_ERR_OP_FAILED)); } else if (reply == NCDS_RPC_NOT_APPLICABLE) { e = nc_err_new(NC_ERR_OP_FAILED); nc_err_set(e, NC_ERR_PARAM_MSG, "Requested operation cannot be performed on the managed datastore."); reply = nc_reply_error(e); } /* and send the reply to the client */ nc_session_send_reply(self->session, rpc, reply); nc_rpc_free(rpc); nc_reply_free(reply); if (req_op == NC_OP_CLOSESESSION) { /* free the Session */ nc_session_free(self->session); self->session = NULL; } Py_RETURN_NONE; }
int main(int argc, char** argv) { struct nc_session* dummy_session; struct nc_cpblts* capabs; struct ncds_ds* ds; nc_rpc* rpc; nc_reply* reply; char* new_startup_config; xmlDocPtr startup_doc = NULL; int ret = 0, i, j; if (argc < 2 || argv[1][0] == '-') { help(argv[0]); return 1; } /* set message printing callback */ nc_callback_print(my_print); /* init libnetconf for messages from transAPI function */ if (nc_init(NC_INIT_ALL | NC_INIT_MULTILAYER) == -1) { my_print(NC_VERB_ERROR, "Could not initialize libnetconf."); return 1; } /* register the datastore */ if ((ds = ncds_new(NCDS_TYPE_FILE, "./model/ietf-interfaces.yin", NULL)) == NULL) { nc_close(); return 1; } /* add imports and augments */ if (ncds_add_model("./model/ietf-yang-types.yin") != 0 || ncds_add_model("./model/ietf-inet-types.yin") != 0 || ncds_add_model("./model/ietf-ip.yin") != 0) { nc_verb_error("Could not add import and augment models."); nc_close(); return 1; } /* enable features */ for (i = 2; i < argc; ++i) { if (strcmp(argv[i], "ipv4-non-contiguous-netmasks") == 0 || strcmp(argv[i], "ipv6-privacy-autoconf") == 0) { j = ncds_feature_enable("ietf-ip", argv[i]); } else { j = ncds_feature_enable("ietf-interfaces", argv[i]); } if (j != 0) { nc_verb_error("Could not enable feature \"%s\".", argv[i]); nc_close(); return 1; } } /* set the path to the target file */ if (ncds_file_set_path(ds, argv[1]) != 0) { nc_verb_error("Could not set \"%s\" to the datastore.", argv[1]); nc_close(); return 1; } if (ncds_init(ds) < 0) { nc_verb_error("Failed to nitialize datastore."); nc_close(); return 1; } if (ncds_consolidate() != 0) { nc_verb_error("Could not consolidate the datastore."); nc_close(); return 1; } if (transapi_init(&startup_doc) != EXIT_SUCCESS) { nc_close(); return 1; } if (startup_doc == NULL || startup_doc->children == NULL) { /* nothing to do */ nc_close(); return 0; } /* create the dummy session */ capabs = nc_cpblts_new(capabilities); if ((dummy_session = nc_session_dummy("session0", "root", NULL, capabs)) == NULL) { nc_verb_error("Could not create a dummy session."); nc_close(); return 1; } /* dump the new config */ xmlDocDumpMemory(startup_doc, (xmlChar**)&new_startup_config, NULL); xmlFreeDoc(startup_doc); /* apply edit-config rpc on the datastore */ if ((rpc = nc_rpc_editconfig(NC_DATASTORE_STARTUP, NC_DATASTORE_CONFIG, 0, 0, 0, new_startup_config)) == NULL) { nc_verb_error("Could not create edit-config RPC."); nc_close(); return 1; } free(new_startup_config); reply = ncds_apply_rpc2all(dummy_session, rpc, NULL); if (nc_reply_get_type(reply) != NC_REPLY_OK) { nc_verb_error("Edit-config RPC failed."); nc_close(); return 1; } nc_reply_free(reply); nc_rpc_free(rpc); nc_cpblts_free(capabs); nc_session_free(dummy_session); nc_close(); return ret; }