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