Fix TCP/IP API
Change-Id: I302bf1ff97d2560e6f68402e0fae609e1435e257
diff --git a/mbtk/test/mbtk_tcpip_test.c b/mbtk/test/mbtk_tcpip_test.c
new file mode 100755
index 0000000..5bd39f1
--- /dev/null
+++ b/mbtk/test/mbtk_tcpip_test.c
@@ -0,0 +1,313 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <errno.h>
+#include <sys/socket.h>
+#include <sys/un.h>
+#include <netinet/in.h>
+#include <pthread.h>
+#include <sys/epoll.h>
+#include <string.h>
+#include <fcntl.h>
+#include <signal.h>
+
+#include "mbtk_tcpip.h"
+#include "mbtk_log.h"
+
+/*
+ int link_id;
+ char ser_addr[256];
+ int ser_port;
+ mbtk_sock_type prot_type; // TCP/UDP
+ mbtk_tcpip_type_enum tcpip_type; // Only support client.
+ int local_port;
+ bool ack_support;
+ bool ssl_support;
+ bool ignore_cert;
+ uint32 heartbeat_time;
+ uint32 delay_time;
+
+ mbtk_tcpip_read_callback_func read_cb;
+
+*/
+static void help()
+{
+ printf("\n************************************************************************\n");
+ printf("net_open: Open network.\n");
+ printf("net_close: Close network.\n");
+ printf("link_open <link_id> <ser_addr> <ser_port> <loc_port> <TCP/UDP> <CLI/SER> <ack> <ssl> <ignore_cert> <heartbeat_time> <delay_time> <read_cb>: Open link.\n");
+ printf("link_close <link_id>: Close link.\n");
+ printf("send <link_id> <data>:Send data.\n");
+ printf("recv <link_id>:Recv data.\n");
+ printf("traffic_reset <link_id>:Traffic reset.\n");
+ printf("traffic_get <link_id>:Traffic get.\n");
+ printf("state_get <link_id>:Link state get.\n");
+ printf("\n************************************************************************\n");
+}
+
+static void sig_process(int sig)
+{
+ LOGI("I got signal %d\n", sig);
+ switch(sig)
+ {
+ case SIGINT: // Ctrl + C
+ {
+ LOGI("Exit by SIGINT.\n");
+ mbtk_tcpip_err_enum err = mbtk_tcpip_net_close();
+ if(err == MBTK_TCPIP_ERR_SUCCESS) {
+ printf("Net close success.\n");
+ } else {
+ printf("Net close fail:%d\n", err);
+ }
+ exit(0);
+ }
+ case SIGQUIT: // Ctrl + \ (类似 SIGINT ，但要产生core文件)
+ {
+ LOGI("Exit by SIGQUIT.\n");
+ mbtk_tcpip_err_enum err = mbtk_tcpip_net_close();
+ if(err == MBTK_TCPIP_ERR_SUCCESS) {
+ printf("Net close success.\n");
+ } else {
+ printf("Net close fail:%d\n", err);
+ }
+ exit(0);
+ }
+ case SIGTERM:// 默认kill (同 SIGKILL ，但 SIGKILL 不可捕获)
+ {
+ LOGI("Exit by SIGTERM.\n");
+ mbtk_tcpip_err_enum err = mbtk_tcpip_net_close();
+ if(err == MBTK_TCPIP_ERR_SUCCESS) {
+ printf("Net close success.\n");
+ } else {
+ printf("Net close fail:%d\n", err);
+ }
+ exit(0);
+ }
+ case SIGTSTP:// Ctrl + Z (同 SIGSTOP ，但 SIGSTOP 不可捕获)
+ {
+ LOGI("Exit by SIGTSTP.\n");
+ exit(0);
+ }
+ case SIGSEGV: // 如空指针
+ {
+ LOGI("Exit by SIGSEGV.\n");
+ exit(0);
+ }
+ default:
+ {
+ LOGI("Unknown sig:%d\n",sig);
+ break;
+ }
+ }
+}
+
+void tcpip_read_cb(int link_id, const char* data, int data_len)
+{
+ printf("\nRECV(%d-%d):%s\n", link_id, data_len, data);
+}
+
+void tcpip_net_callback_func(int state, const char* addr)
+{
+ if(state) {
+ printf("Net conncect, IP : %s\n", addr);
+ } else {
+ printf("Net disconnect.\n");
+ }
+}
+
+void tcpip_sock_callback_func(int link_id, int state)
+{
+ if(state == 0) {
+ printf("Link[%d] disconnected.\n", link_id);
+ }
+}
+
+int main(int argc, char *argv[])
+{
+ signal(SIGINT, sig_process);
+ signal(SIGQUIT, sig_process);
+ signal(SIGTERM, sig_process);
+ //signal(SIGTSTP, sig_process);
+ //signal(SIGSEGV, sig_process);
+
+ mbtk_log_init("radio","MBTK_TCPIP");
+
+ printf(">>>>>>>>>>>>>>>>>>>>>>>>Enter cmd:\n");
+ char cmd[100];
+ while(1)
+ {
+ memset(cmd, 0, 100);
+ mbtk_tcpip_err_enum err;
+ if(fgets(cmd, 100, stdin))
+ {
+ char *ptr = cmd + strlen(cmd) - 1;
+ while(ptr >= cmd && (*ptr == '\r' || *ptr == '\n'))
+ {
+ *ptr-- = '\0';
+ }
+ // net_open
+ if(!strncasecmp(cmd, "net_open", 8)){
+ err = mbtk_tcpip_net_open(tcpip_net_callback_func, tcpip_sock_callback_func);
+ if(err == MBTK_TCPIP_ERR_SUCCESS) {
+ printf("Net open success.\n");
+ } else {
+ printf("Net open fail:%d\n", err);
+ }
+ } else if(!strncasecmp(cmd, "net_close", 9)){ // net_close
+ err = mbtk_tcpip_net_close();
+ if(err == MBTK_TCPIP_ERR_SUCCESS) {
+ printf("Net close success.\n");
+ } else {
+ printf("Net close fail:%d\n", err);
+ }
+ }
+ // link_open <link_id> <ser_addr> <ser_port> <loc_port> <TCP/UDP> <CLI/SER> <ack> <ssl> <ignore_cert> <heartbeat_time> <delay_time> <read_cb>
+ else if(!strncasecmp(cmd, "link_open", 9)){
+ mbtk_tcpip_info_t info;
+ char prot_type[10] = {0};
+ char type[10] = {0};
+ int read_cb_set;
+ memset(&info, 0x0, sizeof(mbtk_tcpip_info_t));
+ int count = sscanf(cmd, "link_open %d %s %d %d %s %s %d %d %d %d %d %d", &(info.link_id),
+ info.ser_addr, &(info.ser_port), &(info.local_port), prot_type, type,
+ &(info.ack_support), &(info.ssl_support), &(info.ignore_cert), &(info.heartbeat_time),
+ &(info.delay_time), &read_cb_set);
+ if(count == 12) {
+ if(!strncasecmp(prot_type, "UDP", 3)) {
+ info.prot_type = MBTK_SOCK_UDP;
+ } else {
+ info.prot_type = MBTK_SOCK_TCP;
+ }
+
+ if(!strncasecmp(type, "SER", 3)) {
+ info.tcpip_type = MBTK_TCPIP_TYPE_SERVER;
+ } else {
+ info.tcpip_type = MBTK_TCPIP_TYPE_CLIENT;
+ }
+
+ if(read_cb_set) {
+ info.read_cb = tcpip_read_cb;
+ } else {
+ info.read_cb = NULL;
+ }
+
+ err = mbtk_tcpip_sock_open((const mbtk_tcpip_info_t*)(&info));
+ if(err == MBTK_TCPIP_ERR_SUCCESS) {
+ printf("Link open success.\n");
+ } else {
+ printf("Link open fail:%d\n", err);
+ }
+ } else {
+ printf("ARG error.\n");
+ }
+ } else if(!strncasecmp(cmd, "link_close", 10)){ // link_close <link_id>
+ int link_id;
+ int count = sscanf(cmd, "link_close %d", &link_id);
+ if(count == 1) {
+ err = mbtk_tcpip_sock_close(link_id);
+ if(err == MBTK_TCPIP_ERR_SUCCESS) {
+ printf("Link close success.\n");
+ } else {
+ printf("Link close fail:%d\n", err);
+ }
+ } else {
+ printf("ARG error.\n");
+ }
+ } else if(!strncasecmp(cmd, "send", 4)){ // send <link_id> <data>
+ int link_id;
+ char data[100] = {0};
+ int count = sscanf(cmd, "send %d %s", &link_id, data);
+ if(count == 2) {
+ int len = mbtk_tcpip_send(link_id, data, strlen(data), NULL, 0);
+ if(len == strlen(data)) {
+ printf("Send success:%d.\n", len);
+ } else {
+ printf("Send fail:%d/%d\n", len, strlen(data));
+ }
+ } else {
+ printf("ARG error.\n");
+ }
+ } else if(!strncasecmp(cmd, "recv", 4)){ // recv <link_id>
+ int link_id;
+ int count = sscanf(cmd, "recv %d", &link_id);
+ if(count == 1) {
+ char buff[2048] = {0};
+ int len = mbtk_tcpip_read(link_id, buff, 2048);
+ if(len > 0) {
+ printf("RECV[%d]:%s\n", len, buff);
+ } else {
+ printf("RECV fail:%d\n", len);
+ }
+ } else {
+ printf("ARG error.\n");
+ }
+ } else if(!strncasecmp(cmd, "traffic_reset", 13)){ // traffic_reset <link_id>
+ int link_id;
+ int count = sscanf(cmd, "traffic_reset %d", &link_id);
+ if(count == 1) {
+ err = mbtk_tcpip_data_traffic_reset(link_id);
+ if(err == MBTK_TCPIP_ERR_SUCCESS) {
+ printf("Traffic reset success.\n");
+ } else {
+ printf("Traffic reset fail:%d\n", err);
+ }
+ } else {
+ printf("ARG error.\n");
+ }
+ } else if(!strncasecmp(cmd, "traffic_get", 11)){ // traffic_get <link_id>
+ int link_id;
+ int count = sscanf(cmd, "traffic_get %d", &link_id);
+ if(count == 1) {
+ int traffic = mbtk_tcpip_data_traffic_get(link_id);
+ if(traffic >= 0) {
+ printf("Traffic : %d\n", traffic);
+ } else {
+ printf("Get raffic fail:%d\n", traffic);
+ }
+ } else {
+ printf("ARG error.\n");
+ }
+ } else if(!strncasecmp(cmd, "state_get", 9)){ // state_get <link_id>
+ int link_id;
+ int count = sscanf(cmd, "state_get %d", &link_id);
+ if(count == 1) {
+ int state = mbtk_tcpip_link_state_get(link_id);
+ if(state == 1) {
+ printf("Link %d connected.\n", link_id);
+ } else if(state == 0) {
+ printf("Link %d disconnected.\n", link_id);
+ } else {
+ printf("Get link state fail:%d\n", state);
+ }
+ } else {
+ printf("ARG error.\n");
+ }
+ }
+ else if(!strcasecmp(cmd, "h") || !strcasecmp(cmd, "help")) {
+ help();
+ } else if(!strcasecmp(cmd, "q")) {
+ err = mbtk_tcpip_net_close();
+ if(err == MBTK_TCPIP_ERR_SUCCESS) {
+ printf("Net close success.\n");
+ } else {
+ printf("Net close fail:%d\n", err);
+ }
+ break;
+ } else {
+ printf("\n");
+ }
+ }
+ }
+
+ LOGD("Client exec complete.");
+#if 1
+ while(1)
+ {
+ sleep(1000 * 365 * 24 * 60 * 60);
+ }
+#else
+ sleep(1);
+#endif
+ return 0;
+}
+