yuesong-feng / 30daymakecppserver Goto Github PK

#include <sys/socket.h>
#include <arpa/inet.h>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <unistd.h>

void errif(bool condition, const char *errmsg)
{
    if (condition)
    {
        perror(errmsg);
        exit(EXIT_FAILURE);
    }
}

int main(int argc, char const *argv[])
{
    int sockfd = socket(AF_INET, SOCK_STREAM, 0);
    errif(sockfd == -1, "sock creat error");

    struct sockaddr_in serv_addr;
    bzero(&serv_addr, sizeof(serv_addr));
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
    serv_addr.sin_port = htons(8888);

    errif(bind(sockfd, (sockaddr *)&serv_addr, sizeof(serv_addr)) == -1, "sock bind error");

    errif(listen(sockfd, SOMAXCONN) == -1, "sock listen error");

    struct sockaddr_in clnt_addr;
    socklen_t clnt_addr_len = sizeof(clnt_addr);
    bzero(&clnt_addr, sizeof(clnt_addr));

    int clnt_sockfd = accept(sockfd, (sockaddr *)&clnt_addr, &clnt_addr_len);
    errif(clnt_sockfd == -1, "sock accpt error");
    printf("new client fd %d! IP: %s Port: %d\n", clnt_sockfd, inet_ntoa(clnt_addr.sin_addr), ntohs(clnt_addr.sin_port));

    while (true)
    {
        char buf[2048];
        bzero(&buf, sizeof(buf));
        ssize_t read_size = read(clnt_sockfd, buf, sizeof(buf));
        if (read_size > 0)
        {
            printf("message from client fd %d: %s\n", clnt_sockfd, buf);
            write(clnt_sockfd, buf, sizeof(buf));
        }
        else if (read_size == 0)
        {
            printf("client fd %d disconnected\n", clnt_sockfd);
            close(clnt_sockfd);
            break;
        }
        else if (read_size == -1)
        {
            close(clnt_sockfd);
            errif(true, "socket read error");
        }
    }
    close(sockfd);
    return 0;
}

#include <sys/socket.h>
#include <arpa/inet.h>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <unistd.h>

void errif(bool condition, const char *errmsg)
{
    if (condition)
    {
        perror(errmsg);
        exit(EXIT_FAILURE);
    }
}

int main(int argc, char const *argv[])
{
    int sockfd = socket(AF_INET, SOCK_STREAM, 0);
    errif(sockfd == -1, "sock creat error");

    struct sockaddr_in serv_addr;
    bzero(&serv_addr, sizeof(serv_addr));
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
    serv_addr.sin_port = htons(8888);

    errif(connect(sockfd, (sockaddr *)&serv_addr, sizeof(serv_addr)) == -1, "connect error");

    while (true)
    {
        char buf[1024];
        bzero(&buf, sizeof(buf));                              //清空缓冲区
        scanf("%s", buf);                                      //从键盘输入要传到服务器的数据
        ssize_t write_bytes = write(sockfd, buf, sizeof(buf)); //发送缓冲区中的数据到服务器socket，返回已发送数据大小
        if (write_bytes == -1)
        { // write返回-1，表示发生错误
            printf("socket already disconnected, can't write any more!\n");
            break;
        }
        bzero(&buf, sizeof(buf));                            //清空缓冲区
        ssize_t read_bytes = read(sockfd, buf, sizeof(buf)); //从服务器socket读到缓冲区，返回已读数据大小
        if (read_bytes > 0)
        {
            printf("message from server: %s\n", buf);
        }
        else if (read_bytes == 0)
        { // read返回0，表示EOF，通常是服务器断开链接，等会儿进行测试
            printf("server socket disconnected!\n");
            break;
        }
        else if (read_bytes == -1)
        { // read返回-1，表示发生错误，按照上文方法进行错误处理
            close(sockfd);
            errif(true, "socket read error");
        }
    }

    return 0;
}

#include <cstdio>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <cstring>
#include <cstdlib>

using namespace std;

void errif(bool condition, const char *errmsg){
    if(condition){
        perror(errmsg);
        exit(EXIT_FAILURE);
    }
}

int main() {
	// 创建 socket
	int sockfd = socket(AF_INET, SOCK_STREAM, 0); 

	// 声明专用sockaddr_in socket 地址
	struct sockaddr_in serv_addr;
	errif(sockfd == -1, "sockfd create error");
	// 初始化地址信息为 0
	bzero(&serv_addr, sizeof(serv_addr));
	// socket地址中 family 表示 IP 地址类型
	serv_addr.sin_family = AF_INET;
	// socket 地址中， sin_addr表示 IPv4 地址结构体,用网络字节序表示
	// inet_addr将用于十进制字符串表示转化为网络字节序整数表示
	serv_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
	// socket 地址中， sin_port表示端口号，要用网络字节序表示
	// htons： host to network short
	serv_addr.sin_port = htons(8888);

	// 将 server 地址和文件描述符绑定
	bind(sockfd, (sockaddr*)&serv_addr, sizeof(serv_addr));
	errif(bind(sockfd, (sockaddr*)&serv_addr, sizeof(serv_addr)) == -1, "bind failed");
	// 监听 socket 端口
	listen(sockfd, SOMAXCONN);
	errif(listen(sockfd, SOMAXCONN) == -1, "listen failed");
	// 声明客户端地址
	struct sockaddr_in clnt_addr;
	// 计算客户端地址所占大小
	socklen_t clnt_addr_len = sizeof(clnt_addr);
	// 初始化客户端地址
	bzero(&clnt_addr, sizeof(clnt_addr));

	// 客户端文件描述符为接收
	int clnt_sockfd = accept(sockfd, (sockaddr*)&clnt_addr, &clnt_addr_len);
	
	printf("new client fd%d! IP:%s Port: %d\n", clnt_sockfd, inet_ntoa(clnt_addr.sin_addr), ntohs(clnt_addr.sin_port));
	return 0;
}