update TCP/UDP socket example code

guowenxue

2023-11-06 d241def81514078e2f3650218781bc4353c4d4bd

update TCP/UDP socket example code

 ch2_socket/makefile

New file
@@ -0,0 +1,10 @@
SRCFILES = $(wildcard *.c)
BINARIES=$(SRCFILES:%.c=%)

all: ${BINARIES}

%: %.c 
    $(CC) $(CFLAGS) -o $@ $< $(LDFLAGS)

clean:
    rm -f ${BINARIES}

 ch2_socket/socket_client.c

File was deleted

 ch2_socket/socket_server.c

File was deleted

 ch2_socket/socktcp_client.c

New file
@@ -0,0 +1,90 @@
/*********************************************************************************
 *        Copyright:    (C) 2018 LingYun IoT Studio
 *                    All rights reserved.
 *
 *         Filename:    socktcp_client.c
 *      Description:    This file is TCP socket client example source code.
 *				
 *          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 <math.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


int main(int argc, char **argv)
{
    int                        conn_fd = -1;
    int                        rv = -1;	
    char                    buf[1024];
    struct sockaddr_in        serv_addr;
    float                    temp;

    /* create TCP socket */
    conn_fd = socket(AF_INET, SOCK_STREAM, 0);
    if(conn_fd < 0)
    {
        printf("create socket failure: %s\n", strerror(errno));
        return -1;
    }

    /* set socket server information */
    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 );

    /* connect to socket server */
    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;
    }

    while(1)
    {
        /* send message to socket server */
        strncpy(buf, "Hello, LingYun IoT socket server!", sizeof(buf));
        if( write(conn_fd, buf, strlen(buf)) < 0 )
        {
            printf("Write data to server [%s:%d] failure: %s\n", SERVER_IP, SERVER_PORT, strerror(errno));
            goto cleanup;
        }
        printf("Send message to socket server okay\n");

        /* receive message from socket server */
        memset(buf, 0, sizeof(buf));
        printf("Wating reply from socket server...\n");
        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("Receive [%d] bytes data from server: %s", rv, buf);
        printf("\n");

        sleep(3);
    }

cleanup:
    close(conn_fd);
    return 0;
}

 ch2_socket/socktcp_server.c

New file
@@ -0,0 +1,111 @@
/*********************************************************************************
 *        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 <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                addr_len = sizeof(cli_addr);
    char                    buf[1024];
    int                        reuse = 1;

    /* 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(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);

    /* 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;
}


 ch2_socket/sockudp_client.c

New file
@@ -0,0 +1,78 @@
/*********************************************************************************
 *        Copyright:    (C) 2018 LingYun IoT Studio
 *                    All rights reserved.
 *
 *         Filename:    sockudp_client.c
 *      Description:    This file is UDP socket client example source code.
 *
 *          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 <math.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           8890


int main(int argc, char **argv)
{
    int                        sockfd = -1;
    int                        rv = -1;
    char                    buf[1024];
    struct sockaddr_in        serv_addr;
    socklen_t                addrlen = sizeof(serv_addr);
    float                    temp;

    /* create UDP socket */
    sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    if(sockfd < 0)
    {
        printf("create socket failure: %s\n", strerror(errno));
        return -1;
    }

    /* set socket server information  */
    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 );

    while(1)
    {
        /* send message to socket server */
        strncpy(buf, "Hello, LingYun IoT socket server!", sizeof(buf));
        if( sendto(sockfd, buf, strlen(buf), 0, (struct sockaddr *)&serv_addr, addrlen) < 0 )
        {
            printf("Write data to server [%s:%d] failure: %s\n", SERVER_IP, SERVER_PORT, strerror(errno));
            break;
        }
        printf("Send message to socket server okay\n");

        /* receive message from socket server */
        memset(buf, 0, sizeof(buf));
        printf("Wating reply from socket server...\n");
        rv = recvfrom(sockfd, buf, sizeof(buf), 0, (struct sockaddr *)&serv_addr, &addrlen);
        if(rv < 0)
        {
            printf("Read data from server failure: %s\n", strerror(errno));
            break;
        }
        printf("Receive [%d] bytes data from server: %s", rv, buf);
        printf("\n");

        sleep(3);
    }

    close(sockfd);
    return 0;
}

 ch2_socket/sockudp_server.c

New file
@@ -0,0 +1,86 @@
/*********************************************************************************
 *        Copyright:    (C) 2018 LingYun IoT Studio
 *                    All rights reserved.
 *
 *         Filename:    socket_server.c
 *      Description:    This file is UDP 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 <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             8890
#define BACKLOG                 13

int main(int argc, char **argv)
{
    int                        rv = -1;
    int                        sockfd;
    struct sockaddr_in        serv_addr;			
    struct sockaddr_in        cli_addr;
    socklen_t                addr_len = sizeof(cli_addr);
    char                    buf[1024];
    int                        reuse = 1;

    /* create UDP socket */
    sockfd = socket(AF_INET, SOCK_DGRAM, 0);
    if(sockfd < 0 )
    {
        printf("create socket failure: %s\n", strerror(errno));
        return -1;
    }
    printf("socket create fd[%d]\n", sockfd);

    /* fix port can not be used when stop server program and run it again immediately */
    setsockopt(sockfd, 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(LISTEN_PORT);
    serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    if( bind(sockfd, (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", sockfd, LISTEN_PORT);

    while(1)
    {
        /* receive message from socket client */
        memset(buf, 0, sizeof(buf));		
        if( (rv=recvfrom(sockfd, buf, sizeof(buf), 0, (struct sockaddr *)&cli_addr, &addr_len)) < 0)
        {
            printf("Read data from socket[%d] failure: %s\n", sockfd, strerror(errno));
            break;
        }
        printf("Read [%d] bytes data from client[%s:%d]: %s\n",
                rv, inet_ntoa(cli_addr.sin_addr), ntohs(cli_addr.sin_port), 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( (rv=sendto(sockfd, buf, sizeof(buf), 0, (struct sockaddr *)&cli_addr, addr_len)) < 0)
        {
            printf("Reply back to socket client[%d] failure: %s\n", sockfd, strerror(errno));
            break;
        }
        printf("\n");
    }

    close(sockfd);
    return 0;
}