socket编程：实现一个TCP服务器

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

#define PORT 4321
#define BACKLOG 1
#define MAXRECVLEN 1024

int main(int argc, char const *argv[])
{
    char buff[MAXRECVLEN];
    int listenfd, connectfd;
    struct sockaddr_in server; // 定义服务端IPv4结构
    struct sockaddr_in client; // 定义客户端IPv4结构
    socklen_t addrlen; // 定义地址长度
    listenfd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    // socket()函数返回-1，则表示创建socket失败
    if (listenfd == -1)
    {
        perror("socket() error. Fail to init socket");
        exit(1); // 抛出异常，退出主进程
    }
    bzero(&server, sizeof(server)); // server IPv4结构体清零

    // 初始化IPv4地址结构体
    server.sin_addr.s_addr = htonl(INADDR_ANY);
    server.sin_family = AF_INET;
    server.sin_port = htons(PORT);

    int bind_status = bind(listenfd, (struct sockaddr*)&server, sizeof(server));
    if (bind_status == -1)
    {
        perror("bind() error. Fail to bind socket");
        exit(1); // 抛出异常，退出主进程
    }

    int listen_status = listen(listenfd, BACKLOG);
    if (listen_status == -1)
    {
        perror("listen() error. Fail to bind socket");
        exit(1); // 抛出异常，退出主进程
    }

    while(1) {
        addrlen = sizeof(client);
        connectfd = accept(listenfd, (struct sockaddr*)&client, &addrlen);
        const char* ip_pression = inet_ntop(AF_INET, &client.sin_addr, buff, sizeof(buff));
        u_int32_t client_port = ntohs(client.sin_port);
        printf("from client:%s:%d\n", ip_pression, client_port);
        write(connectfd, buff, sizeof(buff));
        close(connectfd);
    }
    return 0;
}

Web服务器

#include <sys/socket.h>
#include <sys/un.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <stdio.h>

#define PORT 8080 // 服务器监听端口

int main() {

  int server_socket = socket(AF_INET, SOCK_STREAM, 0);

  struct sockaddr_in server_addr;
  memset(&server_addr, 0, sizeof(server_addr));
  server_addr.sin_family = AF_INET;
  server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
  server_addr.sin_port = htons(PORT);

  bind(server_socket, (struct sockaddr *)&server_addr, sizeof(server_addr));

  listen(server_socket, 5);

  while (1) {
    int client_socket = accept(server_socket, NULL, NULL);

    char buf[1024];
    read(client_socket, buf, 1024);

    printf("%s", buf);
    char status[] = "HTTP/1.0 200 OK\r\n";
    char header[] = "Server: DWBServer\r\nContent-Type: text/html;charset=utf-8\r\n\r\n";
    char body[] = "<html><head><title>hello</title></head><body><h2>hello</h2><p>Hello，World</p></body></html>";

    write(client_socket, status, sizeof(status));
    write(client_socket, header, sizeof(header));
    write(client_socket, body, sizeof(body));

    close(client_socket);
  }

  return 0;
}

xingbofeng / xingbofeng.github.io

socket编程：实现一个TCP服务器 #27