tcp回射程序

服务器端：

#include <stdio.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <stdlib.h>
#include <string.h>


#define LISTENQ 1024
#define SERV_PORT 9877

#define SA struct sockaddr

int main(int argc, char** argv)
{
    int listenfd, connfd;
    pid_t   childpid;
    socklen_t   clilen;
    struct sockaddr_in cliaddr, servaddr;

    listenfd = socket(AF_INET, SOCK_STREAM, 0);
    
    bzero(&servaddr, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servaddr.sin_port = htons(SERV_PORT);

    bind(listenfd, (SA*)&servaddr, sizeof(servaddr));

    listen(listenfd, LISTENQ);

    for(;;){
        clilen = sizeof(cliaddr);
        connfd = accept(listenfd, (SA*)&cliaddr, &clilen);

        if((childpid = fork()) == 0){
            close(listenfd);
            str_echo(connfd);
            exit(0);
        }
        close(connfd);
    }
}

#include <stdio.h>
#include <errno.h>

#define MAXLINE 1024

void str_echo(int sockfd)
{
    ssize_t n;
    char    buf[MAXLINE];

again:
    while((n = read(sockfd, buf, MAXLINE)) > 0)
        writen(sockfd, buf, n);

    if(n < 0 && errno == EINTR){
        goto again;
    }
    else if(n < 0){
        printf("str_echo: read error");
    }
}

#include <stdio.h>
#include <errno.h>

ssize_t writen(int fd, void* vptr, size_t n)
{
    size_t  nleft;
    ssize_t nwritten;
    const char* ptr;

    ptr = vptr;
    nleft = n;

    while(nleft > 0){
        if((nwritten = write(fd, ptr, nleft)) <= 0){
            if(nwritten < 0 && errno == EINTR){
                nwritten = 0;    
            }
            else{
                return(-1);
            }
        }
        nleft -= nwritten;
        ptr += nwritten;
    }
    return(n);
}

客户端程序：

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <stdlib.h>
#include <netinet/in.h>
#include <string.h>

#define SER_PORT 9877
#define SA struct sockaddr

int main(int argc, char** argv)
{
    int sockfd;
    struct sockaddr_in servaddr;

    if(argc != 2){
        printf("param");
    }

    sockfd = socket(AF_INET, SOCK_STREAM, 0);

    bzero(&servaddr, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(SER_PORT);
    
    inet_pton(AF_INET, argv[1], &servaddr.sin_addr);

    connect(sockfd, (SA*)&servaddr, sizeof(servaddr));

    str_cli(stdin, sockfd);

    exit(0);
}

#include <sys/types.h>
#include <stdio.h>
#include <string.h>

#define MAXLINE 1024

void str_cli(FILE* fp, int sockfd)
{
    char sendline[MAXLINE] = {0}, recvline[MAXLINE] = {0};
    while(fgets(sendline, MAXLINE, fp) != NULL){
        writen(sockfd, sendline, strlen(sendline));

        if(readline(sockfd, recvline, MAXLINE) == 0)
            printf("str_cli: server terminated");

        fputs(recvline, stdout);
    }
}

#include <stdio.h>
#include <errno.h>

#define MAXLINE 4096

static int  read_cnt;
static char*    read_ptr;
static char     read_buf[MAXLINE];

static ssize_t my_read(int fd, char* ptr)
{
    if(read_cnt <= 0){
        again:
            if((read_cnt = read(fd, read_buf, sizeof(read_buf))) < 0){
                if(errno == EINTR){
                    goto again;
                }
                return (-1);
            }
            else if(read_cnt == 0){
                return (0);
            }
            read_ptr = read_buf;
    }
    read_cnt--;
    *ptr = *read_ptr++;
    return(1);
}

ssize_t readline(int fd, void* vptr, size_t maxlen)
{
    ssize_t n, rc;
    char    c, *ptr;

    ptr = vptr;

    for(n = 1; n < maxlen; n++){
        if((rc = my_read(fd, &c)) == 1){
            *ptr++ = c;
            if(c == '\n'){
                break;
            }
        }
        else if(rc == 0){
            *ptr = 0;
            return (n -1);
        }
        else{
            return (-1);
        }
    }
    *ptr = 0;
    return(n);
}

ssize_t readlinebuf(void** vptrptr)
{
    if(read_cnt){
        *vptrptr = read_ptr;
    }
    return(read_cnt);
}

Makefile文件

CC = gcc
OBJS = ser01.o str_echo.o writen.o cli01.o str_cli.o readline.o
EXEC = ser01 cli01
 
$(EXEC): $(OBJS)
     $(CC) -g -o ser01 ser01.o str_echo.o writen.o
     $(CC) -g -o cli01 cli01.o str_cli.o writen.o readline.o
 
ser01.o: ser01.c str_echo.c writen.c
     $(CC) -g -c $<
 
cli01.o: cli01.c str_cli.c writen.c readline.c
     $(CC) -g -c $<
 
str_echo.o: str_echo.c writen.c
     $(CC) -g -c $<
 
str_cli.o: str_cli.c writen.c readline.c
     $(CC) -g -c $<
 
writen.o: writen.c
     $(CC) -g -c $<
 
readline.o: readline.c
     $(CC) -g -c $<
 
clean:
     rm -rf $(EXEC) $(EXECLI) *.o