From fb2ca9bfee72fa856717f98a0c4da0722fb53678 Mon Sep 17 00:00:00 2001
From: Guo Wenxue <guowenxue@gmail.com>
Date: Tue, 23 Oct 2018 22:08:55 +0800
Subject: [PATCH] Add ch2_socket sample code

---
 ch2_socket/socket_client.c |   77 +++++++++++++++++++++++++
 ch2_socket/socket_server.c |   95 +++++++++++++++++++++++++++++++
 2 files changed, 172 insertions(+), 0 deletions(-)

diff --git a/ch2_socket/socket_client.c b/ch2_socket/socket_client.c
new file mode 100644
index 0000000..fbe77b3
--- /dev/null
+++ b/ch2_socket/socket_client.c
@@ -0,0 +1,77 @@
+/*********************************************************************************
+ *      Copyright:  (C) 2018 LingYun IoT Studio
+ *                  All rights reserved.
+ *
+ *       Filename:  socket_client.c
+ *    Description:  This file is socket client sample source code, it will connect
+ *                  socket server and send to/receive from it.
+ *                 
+ *        Version:  1.0.0(10/23/2018)
+ *         Author:  Guo Wenxue <guowenxue@gmail.com>
+ *      ChangeLog:  1, Release initial version on "2018-10-23 01:38:08 PM"
+ *                 
+ ********************************************************************************/
+#include <stdio.h>
+#include <errno.h>
+#include <string.h>
+#include <unistd.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+
+#define SERVER_IP          "127.0.0.1"
+#define SERVER_PORT        8889
+#define MSG_STR            "Hello, Unix Network Program World!"
+
+int main(int argc, char **argv)
+{
+    int                     conn_fd = -1;
+    int                     rv = -1;    
+    char                    buf[1024]; 
+    struct sockaddr_in      serv_addr;
+
+    conn_fd = socket(AF_INET, SOCK_STREAM, 0);
+    if(conn_fd < 0)
+    {
+        printf("create socket failure: %s\n", strerror(errno));
+        return -1;
+    }
+
+    memset(&serv_addr, 0, sizeof(serv_addr));
+    serv_addr.sin_family = AF_INET;
+    serv_addr.sin_port = htons(SERVER_PORT);
+    inet_aton( SERVER_IP, &serv_addr.sin_addr );
+
+    if( connect(conn_fd, (struct sockaddr *)&serv_addr, sizeof(serv_addr))  < 0)
+    {
+        printf("connect to server [%s:%d] failure: %s\n", SERVER_IP, SERVER_PORT, strerror(errno));
+        return 0;
+    }
+    
+    if( write(conn_fd, MSG_STR, strlen(MSG_STR)) < 0 )
+    {
+        printf("Write data to server [%s:%d] failure: %s\n", SERVER_IP, SERVER_PORT, strerror(errno));
+	goto cleanup;
+    }
+    
+    memset(buf, 0, sizeof(buf));
+    rv = read(conn_fd, buf, sizeof(buf));    
+    if(rv < 0)
+    {
+	    printf("Read data from server failure: %s\n", strerror(errno));
+	    goto cleanup;
+    }
+    else if( 0 == rv )
+    {
+	    printf("Client connect to server get disconnected\n");
+	    goto cleanup;
+    }
+
+    printf("Read %d bytes data from server: '%s'\n", rv, buf);
+
+
+cleanup:
+    close(conn_fd);
+}
+
diff --git a/ch2_socket/socket_server.c b/ch2_socket/socket_server.c
new file mode 100644
index 0000000..78c27ea
--- /dev/null
+++ b/ch2_socket/socket_server.c
@@ -0,0 +1,95 @@
+/*********************************************************************************
+ *      Copyright:  (C) 2018 LingYun IoT Studio
+ *                  All rights reserved.
+ *
+ *       Filename:  socket_server.c
+ *    Description:  This file is socket server sample source code, it will echo back
+ *                  the receive data from client.
+ *                 
+ *        Version:  1.0.0(10/23/2018)
+ *         Author:  Guo Wenxue <guowenxue@gmail.com>
+ *      ChangeLog:  1, Release initial version on "2017-10-23 01:41:05 PM"
+ *                 
+ ********************************************************************************/
+
+#include <stdio.h>
+#include <string.h>
+#include <errno.h>
+#include <unistd.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+
+#define LISTEN_PORT          8889
+#define BACKLOG              13
+
+int main(int argc, char **argv)
+{
+    int                     rv = -1;
+    int                     listen_fd,  client_fd = -1;
+    struct sockaddr_in      serv_addr;             
+    struct sockaddr_in      cli_addr;
+    socklen_t               cliaddr_len;
+    char                    buf[1024]; 
+    
+    listen_fd = socket(AF_INET, SOCK_STREAM, 0);
+    if(listen_fd < 0 )
+    {
+        printf("create socket failure: %s\n", strerror(errno));
+        return -1;
+    }
+    printf("socket create fd[%d]\n", listen_fd);
+
+    memset(&serv_addr, 0, sizeof(serv_addr));
+    serv_addr.sin_family = AF_INET;        
+    serv_addr.sin_port = htons(LISTEN_PORT);
+    serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
+    if( bind(listen_fd, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0 )
+    {
+        printf("create socket failure: %s\n", strerror(errno));
+        return -2;
+    }
+    printf("socket[%d] bind on port[%d] for all IP address ok\n", listen_fd, LISTEN_PORT);
+
+    listen(listen_fd, BACKLOG); 
+    
+    while(1)   
+    {
+        printf("\nStart waiting and accept new client connect...\n", listen_fd);
+        client_fd = accept(listen_fd, (struct sockaddr*)&cli_addr, &cliaddr_len);
+        if(client_fd < 0)
+        {
+            printf("accept new socket failure: %s\n", strerror(errno));
+            return -2;
+        }
+        printf("Accept new client[%s:%d] with fd [%d]\n", inet_ntoa(cli_addr.sin_addr), ntohs(cli_addr.sin_port), client_fd);
+
+
+        memset(buf, 0, sizeof(buf));        
+        if( (rv=read(client_fd, buf, sizeof(buf))) < 0)
+	{
+		printf("Read data from client socket[%d] failure: %s\n", client_fd, strerror(errno));
+		close(client_fd);
+		continue;
+	}
+	else if( rv == 0 )
+	{
+		printf("client socket[%d] disconnected\n", client_fd);
+		close(client_fd);
+		continue;
+	}
+	printf("read %d bytes data from client[%d] and echo it back: '%s'\n", rv, client_fd, buf);
+
+	if( write(client_fd, buf, rv) < 0 )
+	{
+	       	printf("Write %d bytes data back to client[%d] failure: %s\n", rv, client_fd, strerror(errno));
+		close(client_fd);
+	}
+	
+        sleep(1);
+        close(client_fd); 
+    } 
+    close(listen_fd);
+}
+

--
Gitblit v1.9.1