socket编程学习之TCP模式

1、目的：根据TCP编程模式，编写一个可以进行通信的进程。

2、编写文件主要包括：

• 公共头文件：utili.h
• 客户端进程：cli.c
• 服务器进程：ser.c
• Makefile文件：Makefile

3、各个文件详细代码如下：

公共头文件：utili.h

#include<string.h>
#include<stdlib.h>

#include<sys/socket.h>
#include<netinet/in.h>
#include<arpa/inet.h>

#define BUFFER_MAX_SIZE 256
char sendbuf[BUFFER_MAX_SIZE];
char recvbuf[BUFFER_MAX_SIZE];

#define LISTEN_PORT 8080
#define LISTEN_IP "127.0.0.1"

#define LISTEN_BACKLOG 5

int Socket(int domain, int type, int protocol)
{
    int fd = socket(domain, type, protocol);
    if(fd == -1)
    {
        perror("socket.");
        exit(1);
    }
    return fd;
}

int Bind(int sockfd, const struct sockaddr *addr, socklen_t addrlen)
{
    int ret = bind(sockfd, addr, addrlen);
    if(ret == -1)
    {
        perror("bind.");
        exit(1);
    }
    return ret;
}

int Listen(int sockfd, int backlog)
{
    int ret = listen(sockfd, backlog);
    if(ret == -1)
    {
        perror("listen.");
        exit(1);
    }
    return ret;
}

int startup(const char *ip, int port)
{
    int fd = Socket(AF_INET, SOCK_STREAM, 0);
    struct sockaddr_in address;
    address.sin_family = AF_INET;
    address.sin_port = htons(port);
    address.sin_addr.s_addr = inet_addr(ip);

    socklen_t addrlen = sizeof(struct sockaddr);
    Bind(fd, (struct sockaddr*)&address, addrlen);

    Listen(fd, LISTEN_BACKLOG);

    return fd;
}

 

服务器进程代码：ser.c

#include "utili.h"

int main(int argc, char *argv[])
{
    if(argc != 3)
    {
        printf("Usage: <Sername ip port>.\n");
        return -1;
    }
    int sockSer = startup(argv[1], atoi(argv[2]));

    struct sockaddr_in addrCli;
    socklen_t addrlen = sizeof(struct sockaddr);
    int sockConn = accept(sockSer, (struct sockaddr*)&addrCli, &addrlen);
    if(sockConn == -1)
    {
        printf("Server accept Client connect fail.\n");
        close(sockSer);
        exit(1);
    }
    else
    {
        printf("Server accept Client connect success.\n");
        printf("Client ip:>%s\n", inet_ntoa(addrCli.sin_addr));
        printf("Client port:>%d\n", addrCli.sin_port);
    }

    while(1)
    {
        printf("Ser:>");
        scanf("%s", sendbuf);
        send(sockConn, sendbuf, strlen(sendbuf)+1, 0);
        recv(sockConn, recvbuf, BUFFER_MAX_SIZE, 0);
        printf("Cli:>%s\n", recvbuf);
    }

    close(sockSer);
    return 0;
}

 

客户端进程代码：cli.c

 

#include "../utili.h"

int main(int argc, char *argv[])
{
    if(argc != 3)
    {
        printf("Usage: <Cliname ip port>.\n");
        return -1;
    }
    int sockCli = socket(AF_INET, SOCK_STREAM, 0);
    if(sockCli == -1)
    {
        perror("socket.");
        exit(1);
    }

    struct sockaddr_in addrSer;
    addrSer.sin_family = AF_INET;
    addrSer.sin_port = htons(atoi(argv[2]));
    addrSer.sin_addr.s_addr = inet_addr(argv[1]);
    
    socklen_t addrlen = sizeof(struct sockaddr);
    int ret = connect(sockCli, (struct sockaddr*)&addrSer, addrlen);
    if(ret == -1)
    {
        perror("connect.");
        close(sockCli);
        exit(1);
    }

    while(1)
    {
        recv(sockCli, recvbuf, BUFFER_MAX_SIZE, 0);
        printf("Ser:>%s\n", recvbuf);
        printf("Cli:>");
        scanf("%s",sendbuf);
        send(sockCli, sendbuf, strlen(sendbuf)+1, 0);
    }

    close(sockCli);

    return 0;
}

 

Makefile文件代码：Makefile

all: ser cli

ser:ser.c
	gcc -o ser ser.c
cli:cli.c
	gcc -o cli cli.c

.PHONY:clean
clean:
	rm ser cli

 

4、make：生成文件ser和cli

5、分别执行ser和cli进行通信(借助putty)：

服务器：

客户端：