mgmtd: backend subscriptions to notifications

Signed-off-by: Christian Hopps <chopps@labn.net>

1 files changed, 70 insertions, 22 deletions

diff --git a/lib/mgmt_be_client.c b/lib/mgmt_be_client.c
index 070d6627b4..b217ce40ed 100644
--- a/lib/mgmt_be_client.c
+++ b/lib/mgmt_be_client.c
@@ -822,6 +822,12 @@ static int mgmt_be_client_handle_msg(struct mgmt_be_client *client_ctx,
 	case MGMTD__BE_MESSAGE__MESSAGE_SUBSCR_REPLY:
 		MGMTD_BE_CLIENT_DBG("Got SUBSCR_REPLY success %u",
 				    be_msg->subscr_reply->success);
+
+		if (client_ctx->cbs.subscr_done)
+			(*client_ctx->cbs.subscr_done)(client_ctx,
+						       client_ctx->user_data,
+						       be_msg->subscr_reply
+							       ->success);
 		break;
 	case MGMTD__BE_MESSAGE__MESSAGE_TXN_REQ:
 		MGMTD_BE_CLIENT_DBG("Got TXN_REQ %s txn-id: %" PRIu64,
@@ -958,6 +964,31 @@ static void be_client_handle_get_tree(struct mgmt_be_client *client,
 }
 
 /*
+ * Process the notification.
+ */
+static void be_client_handle_notify(struct mgmt_be_client *client, void *msgbuf,
+				    size_t msg_len)
+{
+	struct mgmt_msg_notify_data *notif_msg = msgbuf;
+	struct mgmt_be_client_notification_cb *cb;
+	const char *notif;
+	uint i;
+
+	MGMTD_BE_CLIENT_DBG("Received notification for client %s", client->name);
+
+	/* "{\"modname:notification-name\": ...}" */
+	notif = (const char *)notif_msg->result + 2;
+
+	for (i = 0; i < client->cbs.nnotify_cbs; i++) {
+		cb = &client->cbs.notify_cbs[i];
+		if (strncmp(cb->xpath, notif, strlen(cb->xpath)))
+			continue;
+		cb->callback(client, client->user_data, cb,
+			     (const char *)notif_msg->result);
+	}
+}
+
+/*
  * Handle a native encoded message
  *
  * We don't create transactions with native messaging.
@@ -972,12 +1003,16 @@ static void be_client_handle_native_msg(struct mgmt_be_client *client,
 	case MGMT_MSG_CODE_GET_TREE:
 		be_client_handle_get_tree(client, txn_id, msg, msg_len);
 		break;
+	case MGMT_MSG_CODE_NOTIFY:
+		be_client_handle_notify(client, msg, msg_len);
+		break;
 	default:
 		MGMTD_BE_CLIENT_ERR("unknown native message txn-id %" PRIu64
 				    " req-id %" PRIu64 " code %u to client %s",
 				    txn_id, msg->req_id, msg->code,
 				    client->name);
-		be_client_send_error(client, msg->refer_id, msg->req_id, false, -1,
+		be_client_send_error(client, msg->refer_id, msg->req_id, false,
+				     -1,
 				     "BE cilent %s recv msg unknown txn-id %" PRIu64,
 				     client->name, txn_id);
 		break;
@@ -1011,38 +1046,51 @@ static void mgmt_be_client_process_msg(uint8_t version, uint8_t *data,
 				    len);
 		return;
 	}
-	MGMTD_BE_CLIENT_DBG(
-		"Decoded %zu bytes of message(msg: %u/%u) from server", len,
-		be_msg->message_case, be_msg->message_case);
+	MGMTD_BE_CLIENT_DBG("Decoded %zu bytes of message(msg: %u/%u) from server",
+			    len, be_msg->message_case, be_msg->message_case);
 	(void)mgmt_be_client_handle_msg(client_ctx, be_msg);
 	mgmtd__be_message__free_unpacked(be_msg, NULL);
 }
 
 int mgmt_be_send_subscr_req(struct mgmt_be_client *client_ctx,
-			    bool subscr_xpaths, int num_xpaths,
-			    char **reg_xpaths)
+			    int n_config_xpaths, char **config_xpaths,
+			    int n_oper_xpaths, char **oper_xpaths)
 {
 	Mgmtd__BeMessage be_msg;
 	Mgmtd__BeSubscribeReq subscr_req;
+	const char **notif_xpaths = NULL;
+	int ret;
 
 	mgmtd__be_subscribe_req__init(&subscr_req);
 	subscr_req.client_name = client_ctx->name;
-	subscr_req.n_xpath_reg = num_xpaths;
-	if (num_xpaths)
-		subscr_req.xpath_reg = reg_xpaths;
-	else
-		subscr_req.xpath_reg = NULL;
-	subscr_req.subscribe_xpaths = subscr_xpaths;
+	subscr_req.n_config_xpaths = n_config_xpaths;
+	subscr_req.config_xpaths = config_xpaths;
+	subscr_req.n_oper_xpaths = n_oper_xpaths;
+	subscr_req.oper_xpaths = oper_xpaths;
+
+	/* See if we should register for notifications */
+	subscr_req.n_notif_xpaths = client_ctx->cbs.nnotify_cbs;
+	if (client_ctx->cbs.nnotify_cbs) {
+		struct mgmt_be_client_notification_cb *cb, *ecb;
+
+		cb = client_ctx->cbs.notify_cbs;
+		ecb = cb + client_ctx->cbs.nnotify_cbs;
+		for (; cb < ecb; cb++)
+			*darr_append(notif_xpaths) = cb->xpath;
+	}
+	subscr_req.notif_xpaths = (char **)notif_xpaths;
 
 	mgmtd__be_message__init(&be_msg);
 	be_msg.message_case = MGMTD__BE_MESSAGE__MESSAGE_SUBSCR_REQ;
 	be_msg.subscr_req = &subscr_req;
 
-	MGMTD_BE_CLIENT_DBG("Sending SUBSCR_REQ name: %s subscr_xpaths: %u num_xpaths: %zu",
-			    subscr_req.client_name, subscr_req.subscribe_xpaths,
-			    subscr_req.n_xpath_reg);
+	MGMTD_BE_CLIENT_DBG("Sending SUBSCR_REQ name: %s xpaths: config %zu oper: %zu notif: %zu",
+			    subscr_req.client_name, subscr_req.n_config_xpaths,
+			    subscr_req.n_oper_xpaths, subscr_req.n_notif_xpaths);
 
-	return mgmt_be_client_send_msg(client_ctx, &be_msg);
+	ret = mgmt_be_client_send_msg(client_ctx, &be_msg);
+	darr_free(notif_xpaths);
+	return ret;
 }
 
 static int _notify_conenct_disconnect(struct msg_client *msg_client,
@@ -1054,15 +1102,16 @@ static int _notify_conenct_disconnect(struct msg_client *msg_client,
 
 	if (connected) {
 		assert(msg_client->conn.fd != -1);
-		ret = mgmt_be_send_subscr_req(client, false, 0, NULL);
+		ret = mgmt_be_send_subscr_req(client, 0, NULL, 0, NULL);
 		if (ret)
 			return ret;
 	}
 
 	/* Notify BE client through registered callback (if any) */
 	if (client->cbs.client_connect_notify)
-		(void)(*client->cbs.client_connect_notify)(
-			client, client->user_data, connected);
+		(void)(*client->cbs.client_connect_notify)(client,
+							   client->user_data,
+							   connected);
 
 	/* Cleanup any in-progress TXN on disconnect */
 	if (!connected)
@@ -1100,9 +1149,8 @@ static void mgmt_debug_client_be_set(uint32_t flags, bool set)
 
 DEFPY(debug_mgmt_client_be, debug_mgmt_client_be_cmd,
       "[no] debug mgmt client backend",
-      NO_STR DEBUG_STR MGMTD_STR
-      "client\n"
-      "backend\n")
+      NO_STR DEBUG_STR MGMTD_STR "client\n"
+				 "backend\n")
 {
 	mgmt_debug_client_be_set(DEBUG_NODE2MODE(vty->node), !no);