/*********************************************************************************
|
* Copyright: (C) 2018 LingYun IoT Studio
|
* All rights reserved.
|
*
|
* Filename: socket_server.c
|
* Description: This file is TCP socket server example source code.
|
*
|
* 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 <errno.h>
|
#include <unistd.h>
|
#include <sys/types.h>
|
#include <sys/socket.h>
|
#include <netinet/in.h>
|
#include <arpa/inet.h>
|
#include <getopt.h>
|
|
#define BACKLOG 13
|
|
void print_usage(char *progname)
|
{
|
printf("%s usage: \n", progname);
|
printf("-p(--port): sepcify server listen port.\n");
|
printf("-h(--Help): print this help information.\n");
|
|
return ;
|
}
|
|
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;
|
int port = 8889;
|
socklen_t addr_len = sizeof(cli_addr);
|
char buf[1024];
|
int reuse = 1;
|
|
int ch;
|
struct option opts[] = {
|
{"port", required_argument, NULL, 'p'},
|
{"help", no_argument, NULL, 'h'},
|
{NULL, 0, NULL, 0}
|
};
|
|
while( (ch=getopt_long(argc, argv, "p:h", opts, NULL)) != -1 )
|
{
|
switch(ch)
|
{
|
case 'p':
|
port=atoi(optarg);
|
break;
|
|
case 'h':
|
print_usage(argv[0]);
|
return 0;
|
}
|
|
}
|
|
if( !port )
|
{
|
print_usage(argv[0]);
|
return 0;
|
}
|
|
|
|
/* create TCP socket */
|
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);
|
|
/* Fix port can not be used when stop server program and run it again immediately. */
|
setsockopt(listen_fd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse));
|
|
/* bind socket server information */
|
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);
|
|
/* start listen socket */
|
listen(listen_fd, BACKLOG);
|
|
while(1)
|
{
|
/* accept new incoming socket client */
|
printf("\nStart waiting and accept new client connect...\n");
|
client_fd = accept(listen_fd, (struct sockaddr*)&cli_addr, &addr_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);
|
|
while(1)
|
{
|
/* receive message from socket client */
|
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);
|
break;
|
}
|
else if( rv == 0 )
|
{
|
printf("client socket[%d] disconnected\n", client_fd);
|
close(client_fd);
|
break;
|
}
|
|
printf("Read [%d] bytes data from socket client[%d]: %s\n", rv, client_fd, buf);
|
|
/* send message to socket client */
|
printf("Reply back to socket client...\n");
|
snprintf(buf, sizeof(buf), "Hello, socket client [%s:%d]\n", inet_ntoa(cli_addr.sin_addr), ntohs(cli_addr.sin_port));
|
if( write(client_fd, buf, strlen(buf)) < 0 )
|
{
|
printf("Reply back to socket client[%d] failure: %s\n", client_fd, strerror(errno));
|
close(client_fd);
|
}
|
printf("\n");
|
}
|
|
close(client_fd);
|
}
|
|
close(listen_fd);
|
return 0;
|
}
|
