From 77670c40de606acfedfe0ec3d37706ec87dc27a0 Mon Sep 17 00:00:00 2001
From: Jan Breuer <jan.breuer@jaybee.cz>
Date: 周二, 09 11月 2021 21:24:26 +0800
Subject: [PATCH] Merge pull request #126 from MisterHW/clean-up

---
 examples/common/scpi-def.h               |    6 +++
 examples/test-LwIP-netconn/scpi_server.c |   79 ++++++++++++++++++++++++++-------------
 examples/test-LwIP-netconn/scpi_server.h |   17 ++++++++
 3 files changed, 76 insertions(+), 26 deletions(-)

diff --git a/examples/common/scpi-def.h b/examples/common/scpi-def.h
index 3b0acca..9e89f95 100644
--- a/examples/common/scpi-def.h
+++ b/examples/common/scpi-def.h
@@ -28,6 +28,9 @@
 
 #ifndef __SCPI_DEF_H_
 #define __SCPI_DEF_H_
+#ifdef __cplusplus
+extern "C" {
+#endif
 
 #include "scpi/scpi.h"
 
@@ -53,5 +56,8 @@
 
 scpi_result_t SCPI_SystemCommTcpipControlQ(scpi_t * context);
 
+#ifdef __cplusplus
+}
+#endif
 #endif /* __SCPI_DEF_H_ */
 
diff --git a/examples/test-LwIP-netconn/scpi_server.c b/examples/test-LwIP-netconn/scpi_server.c
index a12bbbc..2018bcd 100644
--- a/examples/test-LwIP-netconn/scpi_server.c
+++ b/examples/test-LwIP-netconn/scpi_server.c
@@ -49,10 +49,6 @@
 #include "lwip/tcp.h"
 #include "lwip/inet.h"
 
-
-#define DEVICE_PORT 5025
-#define CONTROL_PORT 5026
-
 #define SCPI_THREAD_PRIO (tskIDLE_PRIORITY + 2)
 
 #define SCPI_MSG_TIMEOUT                0
@@ -112,16 +108,13 @@
 }
 
 int SCPI_Error(scpi_t * context, int_fast16_t err) {
-    (void) context;
-    /* BEEP */
-    iprintf("**ERROR: %ld, \"%s\"\r\n", (int32_t) err, SCPI_ErrorTranslate(err));
+    (void) context;    
     if (err != 0) {
         /* New error */
-        /* Beep */
-        /* Error LED ON */
+        SCPI_Event_ErrorIndicatorOn(context, err);
     } else {
         /* No more errors in the queue */
-        /* Error LED OFF */
+        SCPI_Event_ErrorIndicatorOff(context, err);
     }
     return 0;
 }
@@ -145,14 +138,14 @@
     return SCPI_RES_OK;
 }
 
-scpi_result_t SCPI_Reset(scpi_t * context) {
+scpi_result_t __attribute__((weak)) SCPI_Reset(scpi_t * context) {
     (void) context;
     iprintf("**Reset\r\n");
     return SCPI_RES_OK;
 }
 
 scpi_result_t SCPI_SystemCommTcpipControlQ(scpi_t * context) {
-    SCPI_ResultInt(context, CONTROL_PORT);
+    SCPI_ResultInt(context, SCPI_CONTROL_PORT);
     return SCPI_RES_OK;
 }
 
@@ -242,7 +235,7 @@
     }
 }
 
-static int processIoListen(user_data_t * user_data) {
+static int processIoListen(scpi_t * context, user_data_t * user_data) {
     struct netconn *newconn;
 
     if (netconn_accept(user_data->io_listen, &newconn) == ERR_OK) {
@@ -252,7 +245,7 @@
             netconn_delete(newconn);
         } else {
             /* connection established */
-            iprintf("***Connection established %s\r\n", inet_ntoa(newconn->pcb.ip->remote_ip));
+            SCPI_Event_DeviceConnected(context, newconn);
             user_data->io = newconn;
         }
     }
@@ -260,7 +253,7 @@
     return 0;
 }
 
-static int processSrqIoListen(user_data_t * user_data) {
+static int processSrqIoListen(scpi_t * context, user_data_t * user_data) {
     struct netconn *newconn;
 
     if (netconn_accept(user_data->control_io_listen, &newconn) == ERR_OK) {
@@ -269,7 +262,7 @@
             netconn_delete(newconn);
         } else {
             /* control connection established */
-            iprintf("***Control Connection established %s\r\n", inet_ntoa(newconn->pcb.ip->remote_ip));
+            SCPI_Event_ControlConnected(context, newconn);
             user_data->control_io = newconn;
         }
     }
@@ -277,20 +270,20 @@
     return 0;
 }
 
-static void closeIo(user_data_t * user_data) {
+static void closeIo(scpi_t * context, user_data_t * user_data) {
     /* connection closed */
+    SCPI_Event_DeviceDisconnected(context, user_data->io);
     netconn_close(user_data->io);
     netconn_delete(user_data->io);
     user_data->io = NULL;
-    iprintf("***Connection closed\r\n");
 }
 
-static void closeSrqIo(user_data_t * user_data) {
+static void closeSrqIo(scpi_t * context, user_data_t * user_data) {
     /* control connection closed */
+    SCPI_Event_ControlDisconnected(context, user_data->io);
     netconn_close(user_data->control_io);
     netconn_delete(user_data->control_io);
     user_data->control_io = NULL;
-    iprintf("***Control Connection closed\r\n");
 }
 
 static int processIo(user_data_t * user_data) {
@@ -379,10 +372,10 @@
 
     scpi_context.user_context = &user_data;
 
-    user_data.io_listen = createServer(DEVICE_PORT);
+    user_data.io_listen = createServer(SCPI_DEVICE_PORT);
     LWIP_ASSERT("user_data.io_listen != NULL", user_data.io_listen != NULL);
 
-    user_data.control_io_listen = createServer(CONTROL_PORT);
+    user_data.control_io_listen = createServer(SCPI_CONTROL_PORT);
     LWIP_ASSERT("user_data.control_io_listen != NULL", user_data.control_io_listen != NULL);
 
     while (1) {
@@ -393,19 +386,19 @@
         }
 
         if ((user_data.io_listen != NULL) && (evt.cmd == SCPI_MSG_IO_LISTEN)) {
-            processIoListen(&user_data);
+            processIoListen(&scpi_context, &user_data);
         }
 
         if ((user_data.control_io_listen != NULL) && (evt.cmd == SCPI_MSG_CONTROL_IO_LISTEN)) {
-            processSrqIoListen(&user_data);
+            processSrqIoListen(&scpi_context, &user_data);
         }
 
         if ((user_data.io != NULL) && (evt.cmd == SCPI_MSG_IO)) {
-            processIo(&user_data);
+            processIo(&scpi_context, &user_data);
         }
 
         if ((user_data.control_io != NULL) && (evt.cmd == SCPI_MSG_CONTROL_IO)) {
-            processSrqIo(&user_data);
+            processSrqIo(&scpi_context, &user_data);
         }
 
         if (evt.cmd == SCPI_MSG_SET_ESE_REQ) {
@@ -427,3 +420,37 @@
     xReturned = xTaskCreate(scpi_server_thread, "SCPI", DEFAULT_THREAD_STACKSIZE, NULL, SCPI_THREAD_PRIO, &xHandle);
     LWIP_ASSERT("scpi_server_init failed", xReturned == pdPASS);
 }
+
+/* Called by processIoListen() for additional reporting. Override on demand. */
+void __attribute__((weak)) SCPI_Event_DeviceConnected(scpi_t * context, struct netconn * conn) {
+    iprintf("***Connection established %s\r\n", inet_ntoa(newconn->pcb.ip->remote_ip));
+    /* Remote or Eth LED ON */
+}
+
+/* Called by closeIO() for additional reporting. Override on demand. */
+void __attribute__((weak)) SCPI_Event_DeviceDisconnected(scpi_t * context, struct netconn * conn) {
+    iprintf("***Connection closed\r\n");
+    /* Remote or Eth LED OFF */
+}
+
+/* Called by processIoListen() for additional reporting. Override on demand. */
+void __attribute__((weak)) SCPI_Event_ControlConnected(scpi_t * context, struct netconn * conn) {
+    iprintf("***Control Connection established %s\r\n", inet_ntoa(newconn->pcb.ip->remote_ip));
+}
+
+/* Called by closeIO() for additional reporting. Override on demand. */
+void __attribute__((weak)) SCPI_Event_ControlDisconnected(scpi_t * context, struct netconn * conn) {
+    iprintf("***Control Connection closed\r\n");
+}
+
+/* Called by SCPI_Error() for reporting. Override on demand. */
+void __attribute__((weak)) SCPI_Event_ErrorIndicatorOn(scpi_t * context, int_fast16_t err) {
+    iprintf("**ERROR: %ld, \"%s\"\r\n", (int32_t) err, SCPI_ErrorTranslate(err));
+    /* New error : BEEP, Error LED ON */
+}
+
+/* Called by SCPI_Error() for reporting. Override on demand. */
+void __attribute__((weak)) SCPI_Event_ErrorIndicatorOff(scpi_t * context, int_fast16_t err) {
+    iprintf("**ERROR: %ld, \"%s\"\r\n", (int32_t) err, SCPI_ErrorTranslate(err));
+    /* No more errors in the queue : Error LED OFF */
+}
diff --git a/examples/test-LwIP-netconn/scpi_server.h b/examples/test-LwIP-netconn/scpi_server.h
index 4d2895f..90bdc64 100644
--- a/examples/test-LwIP-netconn/scpi_server.h
+++ b/examples/test-LwIP-netconn/scpi_server.h
@@ -28,14 +28,31 @@
 
 #ifndef _SCPI_SERVER_H_
 #define _SCPI_SERVER_H_
+#ifdef __cplusplus
+extern "C" {
+#endif
 
+#define SCPI_DEVICE_PORT  5025 // scpi-raw standard port
+#define SCPI_CONTROL_PORT 5026 // libscpi control port (not part of the standard)
 
 #include <stdint.h>
+#include "lwip/api.h"
+#include "scpi/types.h"
 
 void scpi_server_init(void);
 
 void SCPI_AddError(int16_t err);
 void SCPI_RequestControl(void);
 
+// optional event handlers
+void SCPI_Event_DeviceConnected(scpi_t * context, struct netconn * conn);
+void SCPI_Event_DeviceDisconnected(scpi_t * context, struct netconn * conn);
+void SCPI_Event_ControlConnected(scpi_t * context, struct netconn * conn);
+void SCPI_Event_ControlDisconnected(scpi_t * context, struct netconn * conn);
+void SCPI_Event_ErrorIndicatorOn(scpi_t * context, int_fast16_t err);
+void SCPI_Event_ErrorIndicatorOff(scpi_t * context, int_fast16_t err);
 
+#ifdef __cplusplus
+}
+#endif
 #endif /* _SCPI_SERVER_H_ */

--
Gitblit v1.9.1