【网络编程】——linux socket demo

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <unistd.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#if 0
#define UDP
#else
#define TCP
#endif

int sockfd;
char* IP = "10.8.2.60";
//char *IP = "127.0.0.1";
#ifdef UDP
short PORT = ;
#endif
#ifdef TCP
short PORT = ;
#endif
typedef struct sockaddr SA;

void init(){
#ifdef UDP
    sockfd = socket(AF_INET, SOCK_DGRAM, );
#endif

#ifdef TCP
    sockfd = socket(AF_INET, SOCK_STREAM, );
    struct sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_port = htons(PORT);
    addr.sin_addr.s_addr = inet_addr(IP);
    if(connect(sockfd, (SA*)&addr, sizeof(addr)) == -){
        printf("connect failed!\n");
        exit(-);
    }
#endif
}

int main (int argc, char *argv[]) {
    //char buf[2048];
    char buf[];
    int index, i = ;

    if (argc == ) {
        index = atoi(argv[]);
    } else {
        printf("Usage: ./client <times>\n");
        return -;
    } 

    printf("start init ....\n");
    init();
    printf("connected...\n");
    memset(buf, , sizeof(buf));
#ifdef UDP
    struct sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_port = htons(PORT);
    addr.sin_addr.s_addr = inet_addr(IP);
#endif

    while () {
        //printf("please input something:\n");
        //scanf("%s", buf);
        //puts(buf);
        if (i == index)
            break;
#ifdef TCP
        send(sockfd, buf, sizeof(buf), );
        printf("send 2048! index:%d\n", i);
#endif
#ifdef UDP
        printf("sendto 2048! index:%d\n", i);
        if (sendto(sockfd, buf, sizeof(buf), , (struct sockaddr *)&addr, sizeof(struct sockaddr)) == -){
            printf("sendto error!\n");
            return -;
        }
#endif
        //sleep(1);
        i++;
    }

    close(sockfd);

    return ;
}

client。

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <unistd.h>
#include <netinet/in.h>
#include <arpa/inet.h>

int sockfd;
char* IP = "10.8.2.56";
//char *IP = "127.0.0.1";
typedef struct sockaddr SA;

#if 0
#define UDP
#else
#define TCP
#endif
#ifdef UDP
short PORT = ;
#else
short PORT = ;
#endif

void init(){
#ifdef TCP
    sockfd = socket(AF_INET, SOCK_STREAM, );    //tcp
#endif
#ifdef UDP
    sockfd = socket(AF_INET, SOCK_DGRAM, );    //udp
#endif
    if(sockfd == -){
        printf("socket failed!\n");
        exit(-);
    }

    struct sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_port = htons(PORT);
    addr.sin_addr.s_addr = inet_addr(IP);
    if(bind(sockfd, (SA *)&addr, sizeof(addr)) == -) {
        printf("bind failed!\n");
        exit(-);
    }
#ifdef TCP
    if(listen(sockfd, ) == -) {
        printf("listen failed!\n");
        exit(-);
    }
#endif
}

int main (int argc, char *argv[]) {
    int fd;
    int ret;
    char buf[];
    int index = ;
    init();

    while () {
#ifdef TCP
        struct sockaddr_in fromaddr;
        socklen_t len = sizeof(fromaddr);
        fd = accept(sockfd, (SA *)&fromaddr, &len);
        if (fd < ) {
            printf("accept failed!\n");
            continue;
        }
        printf("fd:%d\n", fd);
#endif
        memset(buf, , sizeof(buf));
        while () {
#ifdef TCP
            if ((ret = recv(fd, buf, sizeof(buf), MSG_WAITALL)) > ) {
#endif
#ifdef UDP
            printf("udp!\n");
            if ((ret = recvfrom(sockfd, buf, sizeof(buf), MSG_WAITALL, NULL, NULL)) > ) {
#endif
                printf("ret %d\n", ret);
            } else {
                printf("recvfrom failed! ret:%d\n", ret);
                break;
            }
            printf("index:%d\n", index);
            index++;
        }
    }

    return ;
}

server。