[qca-nss-drv] Update callback handler

Update the callback handler to handle
per core requests

Change-Id: Id0502903ffd26c81620e34427af7e2f626a572b3
Signed-off-by: Thomas Wu <wthomas@codeaurora.org>
diff --git a/nss_oam.c b/nss_oam.c
index 9d3204e..26f5906 100644
--- a/nss_oam.c
+++ b/nss_oam.c
@@ -162,7 +162,9 @@
  */
 void nss_oam_register_handler(void)
 {
-	if (nss_core_register_handler(NSS_OAM_INTERFACE, nss_oam_rx_msg_handler, NULL) != NSS_CORE_STATUS_SUCCESS) {
+	struct nss_ctx_instance *nss_ctx = &nss_top_main.nss[nss_top_main.oam_handler_id];
+
+	if (nss_core_register_handler(nss_ctx, NSS_OAM_INTERFACE, nss_oam_rx_msg_handler, NULL) != NSS_CORE_STATUS_SUCCESS) {
 		nss_warning("OAM handler failed to register");
 	}
 }