From ba7f217b4121935439aec339529b1c39df9e1bba Mon Sep 17 00:00:00 2001
From: guowenxue <guowenxue@gmail.com>
Date: Tue, 30 Oct 2018 10:30:11 +0800
Subject: [PATCH] Add TCP and UDP socket server sample code in 4.Socket_Server

---
 apue/4.Socket_Server/tcp_server.c |  108 +++++++++++++++++++++++++++
 apue/4.Socket_Server/makefile     |   13 +++
 apue/4.Socket_Server/udp_server.c |  102 +++++++++++++++++++++++++
 3 files changed, 223 insertions(+), 0 deletions(-)

diff --git a/apue/4.Socket_Server/makefile b/apue/4.Socket_Server/makefile
new file mode 100644
index 0000000..91f1d13
--- /dev/null
+++ b/apue/4.Socket_Server/makefile
@@ -0,0 +1,13 @@
+BIN_FILES=tcp_server udp_server
+
+all: clean ${BIN_FILES}
+
+tcp_server:
+	gcc $@.c  -o $@
+	
+udp_server:
+	gcc $@.c  -o $@
+
+clean:
+	@rm -f ${BIN_FILES}
+
diff --git a/apue/4.Socket_Server/tcp_server.c b/apue/4.Socket_Server/tcp_server.c
new file mode 100644
index 0000000..006a044
--- /dev/null
+++ b/apue/4.Socket_Server/tcp_server.c
@@ -0,0 +1,108 @@
+/*********************************************************************************
+ *      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 DEF_LISTEN_PORT      8060
+#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]; 
+    int                     port = DEF_LISTEN_PORT;
+    int                     reuse = 1;
+
+    printf("You can use '%s [port]' to change default port[%d].\n", argv[0], DEF_LISTEN_PORT);
+
+    if( 2== argc )
+    {
+        port = atoi(argv[1]);
+    }
+
+    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);
+
+    setsockopt(listen_fd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse));
+
+    memset(&serv_addr, 0, sizeof(serv_addr));
+    serv_addr.sin_family = AF_INET;        
+    serv_addr.sin_port = htons(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, 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);
+        }
+	
+        nanosleep(1000);
+        close(client_fd); 
+    } 
+
+    close(listen_fd);
+}
+
diff --git a/apue/4.Socket_Server/udp_server.c b/apue/4.Socket_Server/udp_server.c
new file mode 100644
index 0000000..d033150
--- /dev/null
+++ b/apue/4.Socket_Server/udp_server.c
@@ -0,0 +1,102 @@
+/*********************************************************************************
+ *      Copyright:  (C) 2018 LingYun IoT Studio
+ *                  All rights reserved.
+ *
+ *       Filename:  udp_server.c
+ *    Description:  This file is UDP 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 <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <ctype.h>
+#include <errno.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+
+#define DEF_SERV_PORT 8059
+
+void dump_buf(char *buf, int size)
+{
+    int  i;
+
+    for(i=0; i<size; i++)
+    {
+        printf("0x%02x ", buf[i]);
+    }
+
+    printf("\n");
+}
+
+int main(int argc, char **argv)
+{
+    struct sockaddr_in      servaddr, cliaddr;
+    socklen_t               cliaddr_len;
+    int                     sockfd;
+    char                    buf[1024];
+    char                    ip[INET_ADDRSTRLEN];
+    int                     i, len;
+    int                     port = DEF_SERV_PORT;
+
+    printf("You can use '%s [port]' to change default port[%d].\n", argv[0], DEF_SERV_PORT);
+
+    if( argc == 2 )
+    {
+        port = atoi(argv[1]);
+    }
+
+    sockfd = socket(AF_INET, SOCK_DGRAM, 0);
+
+    bzero(&servaddr, sizeof(servaddr));
+    servaddr.sin_family = AF_INET;
+    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
+    servaddr.sin_port = htons(port);
+
+    printf("Start listen on port[%d]\n", port);
+
+    bind(sockfd, (struct sockaddr*)&servaddr, sizeof(servaddr));
+    printf("Start Accepting connections ...\n");
+
+    while(1)
+    {
+        cliaddr_len = sizeof(cliaddr);
+
+        memset(buf, 0, sizeof(buf));
+        len = recvfrom(sockfd, buf, sizeof(buf), 0, (struct sockaddr*)&cliaddr, &cliaddr_len);
+        if(len == -1)
+        {
+            printf("recvfrom error: %s\n", strerror(errno));
+        }
+        printf("Received %d bytes datas from %s at PORT %d\n", len, inet_ntop(AF_INET, &cliaddr.sin_addr, ip, sizeof(ip)), 
+                ntohs(cliaddr.sin_port));
+
+        printf("%s\n", buf);
+        dump_buf(buf, len);
+
+#if 0
+        for(i = 0; i < len; i++)
+        {
+            buf[i] = toupper(buf[i]);
+        }
+        dump_buf(buf, len);
+#endif
+
+        len = sendto(sockfd, buf, len, 0, (struct sockaddr*)&cliaddr, sizeof(cliaddr));
+        if(len == -1)
+        {
+            printf("sendto error: %s\n", strerror(errno));
+        }
+    }
+
+    close(sockfd);
+    return 0;
+}

--
Gitblit v1.9.1