117 lines
3.4 KiB
C
117 lines
3.4 KiB
C
#include <stdio.h>
|
||
#include <stdlib.h>
|
||
#include <string.h>
|
||
#include <unistd.h>
|
||
#include <sys/socket.h>
|
||
#include <bits/fcntl-linux.h>
|
||
#include <errno.h>
|
||
|
||
#include "tools/log/log.h"
|
||
#include "http_rel.h"
|
||
#include <netinet/in.h>
|
||
|
||
/* 接收状态辅助结构 */
|
||
|
||
const char *http_get_body(const char *buf);
|
||
char *recv_http_request(int cfd);
|
||
|
||
/* http_get_body 无需修改，保持原样 */
|
||
const char *http_get_body(const char *buf)
|
||
{
|
||
if (!buf) return NULL;
|
||
const char *sep = strstr(buf, "\r\n\r\n");
|
||
if (!sep) return NULL;
|
||
const char *body = sep + 4;
|
||
if (*body == '\0') return NULL;
|
||
return body;
|
||
}
|
||
|
||
/**
|
||
* @brief 初始化HTTP监听socket，所有错误通过logmanager记录
|
||
* @param port 监听端口
|
||
* @param logger 日志管理器实例指针
|
||
* @return 成功返回监听fd，失败返回-1并记录日志
|
||
*/
|
||
int init_http_network(int port, log_manager *logger)
|
||
{
|
||
logs *log;
|
||
int fd;
|
||
|
||
/* 1. 创建socket */
|
||
fd = socket(AF_INET, SOCK_STREAM, 0);
|
||
if (fd == -1) {
|
||
log = malloc(sizeof(logs));
|
||
// cppcheck-suppress uninitdata
|
||
snprintf(log->log, sizeof(log->log),
|
||
"[FATAL] socket() failed: %s", strerror(errno));
|
||
logger->in_log(log, logger);
|
||
return -1;
|
||
}
|
||
|
||
/* 2. 设置SO_REUSEADDR，避免TIME_WAIT状态导致bind失败 */
|
||
int opt = 1;
|
||
if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt)) == -1) {
|
||
log = malloc(sizeof(logs));
|
||
snprintf(log->log, sizeof(log->log),
|
||
"[ERROR] setsockopt(SO_REUSEADDR) on fd=%d failed: %s",
|
||
fd, strerror(errno));
|
||
logger->in_log(log, logger);
|
||
close(fd);
|
||
return -1;
|
||
}
|
||
|
||
/* 3. 设置为非阻塞模式（配合epoll使用） */
|
||
int flags = fcntl(fd, F_GETFL, 0);
|
||
if (flags == -1) {
|
||
log = malloc(sizeof(logs));
|
||
snprintf(log->log, sizeof(log->log),
|
||
"[ERROR] fcntl(F_GETFL) on fd=%d failed: %s", fd, strerror(errno));
|
||
logger->in_log(log, logger);
|
||
|
||
close(fd);
|
||
return -1;
|
||
}
|
||
if (fcntl(fd, F_SETFL, flags | O_NONBLOCK) == -1) {
|
||
log = malloc(sizeof(logs));
|
||
snprintf(log->log, sizeof(log->log),
|
||
"[ERROR] fcntl(O_NONBLOCK) on fd=%d failed: %s", fd, strerror(errno));
|
||
logger->in_log(log, logger);
|
||
close(fd);
|
||
return -1;
|
||
}
|
||
|
||
/* 4. 绑定到指定端口 */
|
||
struct sockaddr_in addr = {0};
|
||
addr.sin_family = AF_INET;
|
||
addr.sin_port = htons(port);
|
||
addr.sin_addr.s_addr = htonl(INADDR_ANY); // 监听所有网卡
|
||
|
||
if (bind(fd, (struct sockaddr *)&addr, sizeof(addr)) == -1) {
|
||
log = malloc(sizeof(logs));
|
||
snprintf(log->log, sizeof(log->log),
|
||
"[FATAL] bind(port %d) failed: %s (fd=%d)",
|
||
port, strerror(errno), fd);
|
||
logger->in_log(log, logger);
|
||
close(fd);
|
||
return -1;
|
||
}
|
||
|
||
/* 5. 开始监听 */
|
||
if (listen(fd, 10) == -1) {
|
||
log = malloc(sizeof(logs));
|
||
snprintf(log->log, sizeof(log->log),
|
||
"[FATAL] listen(fd=%d, backlog=10) failed: %s",
|
||
fd, strerror(errno));
|
||
logger->in_log(log, logger);
|
||
close(fd);
|
||
return -1;
|
||
}
|
||
|
||
/* 6. 成功日志 */
|
||
log = malloc(sizeof(logs));
|
||
snprintf(log->log, sizeof(log->log),
|
||
"[HTTP] Successfully listening on port %d (fd=%d)", port, fd);
|
||
logger->in_log(log, logger);
|
||
return fd;
|
||
}
|