X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=src%2Fvlibapi%2Fvat_helper_macros.h;h=5e7f10839474d7203f6a1138ee7250010a27210a;hb=59b2565cd91a67ced650739f36129650830211ac;hp=dc91c9f5023757871a22ac8e85d42d03f952d107;hpb=2d6b2d6d1bbb130921ec525a1cc6e88f42717c79;p=vpp.git

diff --git a/src/vlibapi/vat_helper_macros.h b/src/vlibapi/vat_helper_macros.h
index dc91c9f5023..5e7f1083947 100644
--- a/src/vlibapi/vat_helper_macros.h
+++ b/src/vlibapi/vat_helper_macros.h
@@ -20,54 +20,115 @@
 #define __vat_helper_macros_h__
 
 /* M: construct, but don't yet send a message */
-#define M(T,t)                                                  \
+#define M(T, mp)                                                \
 do {                                                            \
+    socket_client_main_t *scm = &vam->socket_client_main;	\
     vam->result_ready = 0;                                      \
-    mp = vl_msg_api_alloc_as_if_client(sizeof(*mp));            \
+    if (scm->socket_enable)                                     \
+      {                                                         \
+        mp = (void *)scm->socket_tx_buffer;                     \
+        scm->socket_tx_nbytes = sizeof (*mp);                   \
+      }                                                         \
+    else                                                        \
+      mp = vl_msg_api_alloc_as_if_client(sizeof(*mp));          \
     memset (mp, 0, sizeof (*mp));                               \
     mp->_vl_msg_id = ntohs (VL_API_##T+__plugin_msg_base);      \
     mp->client_index = vam->my_client_index;                    \
 } while(0);
 
-#define M2(T,t,n)                                               \
+/* MPING: construct a control-ping message, don't send it yet */
+#define MPING(T, mp)                                            \
 do {                                                            \
+    socket_client_main_t *scm = &vam->socket_client_main;	\
     vam->result_ready = 0;                                      \
-    mp = vl_msg_api_alloc_as_if_client(sizeof(*mp)+(n));        \
+    if (scm->socket_enable)                                     \
+      {                                                         \
+        mp = (void *)scm->socket_tx_buffer;                     \
+        scm->socket_tx_nbytes = sizeof (*mp);                   \
+      }                                                         \
+    else                                                        \
+      mp = vl_msg_api_alloc_as_if_client(sizeof(*mp));          \
+    memset (mp, 0, sizeof (*mp));                               \
+    mp->_vl_msg_id = ntohs (VL_API_##T+__plugin_msg_base);      \
+    mp->client_index = vam->my_client_index;                    \
+    scm->control_pings_outstanding++;                           \
+} while(0);
+
+#define M2(T, mp, n)                                            \
+do {                                                            \
+    socket_client_main_t *scm = &vam->socket_client_main;	\
+    vam->result_ready = 0;                                      \
+    if (scm->socket_enable)                                     \
+      {                                                         \
+        mp = (void *)scm->socket_tx_buffer;                     \
+        scm->socket_tx_nbytes = sizeof (*mp) + n;               \
+      }                                                         \
+    else                                                        \
+      mp = vl_msg_api_alloc_as_if_client(sizeof(*mp) + n);      \
     memset (mp, 0, sizeof (*mp));                               \
     mp->_vl_msg_id = ntohs (VL_API_##T+__plugin_msg_base);      \
     mp->client_index = vam->my_client_index;                    \
 } while(0);
 
 /* S: send a message */
-#define S (vl_msg_api_send_shmem (vam->vl_input_queue, (u8 *)&mp))
+#define S(mp)                                                   \
+do {                                                            \
+  int n;                                                        \
+  socket_client_main_t *scm = &vam->socket_client_main;         \
+  if (scm->socket_enable)                                       \
+    {                                                           \
+      msgbuf_t msgbuf;                                          \
+                                                                \
+      msgbuf.q = 0;                                             \
+      msgbuf.gc_mark_timestamp = 0;                             \
+      msgbuf.data_len = ntohl(scm->socket_tx_nbytes);           \
+                                                                \
+      n = write (scm->socket_fd, &msgbuf, sizeof (msgbuf));     \
+      if (n < sizeof (msgbuf))                                  \
+        clib_unix_warning ("socket write (msgbuf)");            \
+                                                                \
+      n = write (scm->socket_fd, scm->socket_tx_buffer,         \
+                   scm->socket_tx_nbytes);                      \
+      if (n < scm->socket_tx_nbytes)                            \
+        clib_unix_warning ("socket write (msg)");               \
+    }                                                           \
+  else                                                          \
+    vl_msg_api_send_shmem (vam->vl_input_queue, (u8 *)&mp);     \
+ } while (0);
 
 /* W: wait for results, with timeout */
-#define W                                       \
-do {                                            \
-    timeout = vat_time_now (vam) + 1.0;         \
-                                                \
-    while (vat_time_now (vam) < timeout) {      \
-        if (vam->result_ready == 1) {           \
-            return (vam->retval);               \
-        }                                       \
-        vat_suspend (vam->vlib_main, 1e-5);     \
-    }                                           \
-    return -99;                                 \
+#define W(ret)                                                  \
+do {                                                            \
+    f64 timeout = vat_time_now (vam) + 1.0;                     \
+    socket_client_main_t *scm = &vam->socket_client_main;	\
+    ret = -99;                                                  \
+                                                                \
+    vl_socket_client_read_reply (scm);                          \
+    while (vat_time_now (vam) < timeout) {                      \
+        if (vam->result_ready == 1) {                           \
+            ret = vam->retval;                                  \
+            break;                                              \
+        }                                                       \
+        vat_suspend (vam->vlib_main, 1e-5);                     \
+    }                                                           \
 } while(0);
 
 /* W2: wait for results, with timeout */
-#define W2(body)                                \
-do {                                            \
-    timeout = vat_time_now (vam) + 1.0;         \
-                                                \
-    while (vat_time_now (vam) < timeout) {      \
-        if (vam->result_ready == 1) {           \
-	  (body);                               \
-	  return (vam->retval);                 \
-        }                                       \
-        vat_suspend (vam->vlib_main, 1e-5);     \
-    }                                           \
-    return -99;                                 \
+#define W2(ret, body)                                           \
+do {                                                            \
+    f64 timeout = vat_time_now (vam) + 1.0;                     \
+    socket_client_main_t *scm = &vam->socket_client_main;	\
+    ret = -99;                                                  \
+                                                                \
+    vl_socket_client_read_reply (scm);                          \
+    while (vat_time_now (vam) < timeout) {                      \
+        if (vam->result_ready == 1) {                           \
+	  (body);                                               \
+	  ret = vam->retval;                                    \
+          break;                                                \
+        }                                                       \
+        vat_suspend (vam->vlib_main, 1e-5);                     \
+    }                                                           \
 } while(0);