APUE Learning Example Source Code
guowenxue
2020-06-07 8a383b8715fc07145fdc3f3e3dbe6ecd93f72989
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h> 
#include <sys/socket.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <getopt.h>
 
#define MSG_STR "Hello LingYun IoT Studio Client\n"
 
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                  sockfd = -1;
    int                  rv = -1;
    struct sockaddr_in   servaddr;
    struct sockaddr_in   cliaddr;
    socklen_t            len;
    int                  port = 0;
    char                 buf[1024];
    int                  clifd;
    int                  ch;
    int                  on = 1;
 
           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;
    }
 
    sockfd=socket(AF_INET, SOCK_STREAM, 0);
    if(sockfd < 0)
    {
        printf("Create socket failure: %s\n", strerror(errno));
        return -1;
    }
    printf("Create socket[%d] successfully!\n", sockfd);
 
    setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
 
    memset(&servaddr, 0, sizeof(servaddr));
    servaddr.sin_family=AF_INET;
    servaddr.sin_port = htons(port);
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    //inet_aton("192.168.0.16", &servaddr.sin_addr);
 
    rv=bind(sockfd, (struct sockaddr *)&servaddr, sizeof(servaddr));
    if(rv < 0)
    {
        printf("Socket[%d] bind on port[%d] failure: %s\n", 
                sockfd, port, strerror(errno));
        return -2;
    }
 
    listen(sockfd, 13);
    printf("Start to listen on port [%d]\n", port);
 
    while(1)
    { 
        printf("Start accept new client incoming...\n");
 
        clifd=accept(sockfd, (struct sockaddr *)&cliaddr, &len);
        if(clifd < 0)
        {
            printf("Accept new client failure: %s\n", strerror(errno));
            continue;
        }
 
        printf("Accept new client[%s:%d] successfully\n", 
                inet_ntoa(cliaddr.sin_addr), ntohs(cliaddr.sin_port));
 
        memset(buf, 0, sizeof(buf));
        rv=read(clifd, buf, sizeof(buf));
               if( rv < 0)
               {
                   printf("Read data from client sockfd[%d] failure: %s\n", 
                    clifd, strerror(errno));
            close(clifd);
            continue;
               }
               else if( rv == 0)
               { 
            printf("Socket[%d] get disconnected\n", clifd);
            close(clifd);
            continue;
        }
        else if( rv > 0 )
        {
            printf("Read %d bytes data from Server: %s\n", 
                    rv, buf);
        }
 
        rv=write(clifd, MSG_STR, strlen(MSG_STR));
               if(rv < 0)
               {
                   printf("Write to client by sockfd[%d] failure: %s\n", 
                    sockfd, strerror(errno));
            close(clifd);
            continue;
               }
 
        printf("Close client socket[%d]\n", clifd);
        close(clifd);
    }
 
 
    close(sockfd);
}