522685e55b
Some checks failed
CI / build (push) Failing after 1m0s
- 新增 throttle.c/throttle.h：Token Bucket 双层限速实现 - 支持 per-connection（单连接限速）和 global（服务器总带宽限速） - 使用 fd 映射表自动注册/清理，侵入性最小 - cocoon.h: 配置结构体新增 throttle_conn_rate、throttle_global_rate - server.c: 连接创建时自动初始化 per-connection throttle；server_destroy 释放 global throttle - static.c: send_all() 自动查表限速 - platform.c: cocoon_file_send() 限速启用时回退到 read+write 循环，逐 chunk 限速 - proxy.c: send_all_fd() 加入限速 - config.c: 解析 JSON 配置中的 throttle 对象（conn_rate / global_rate） - config.h: 更新 config_merge 签名 - Makefile: 将 throttle.c 加入 SRCS 和所有单元测试编译规则 编译零警告，109 项集成测试通过（6 个失败为已知环境反向代理 404 问题）
196 lines
8.7 KiB
Makefile
196 lines
8.7 KiB
Makefile
# Cocoon - 基于 coco 协程库的静态资源 Web 服务器
|
||
|
||
# coco 库路径（默认使用 submodule，可覆盖）
|
||
COCO_DIR ?= coco
|
||
COCO_INCLUDE ?= $(COCO_DIR)/include
|
||
COCO_LIB ?= $(COCO_DIR)/build
|
||
|
||
CC = gcc
|
||
CFLAGS = -Wall -Wextra -O2 -std=c11 -D_GNU_SOURCE -I$(COCO_INCLUDE) -I/usr/include/x86_64-linux-gnu
|
||
LDFLAGS = -L$(COCO_LIB) -lcoco -lpthread -lm -luring -lz -lbrotlienc -lssl -lcrypto -lnghttp2 -ldl -lcurl
|
||
|
||
# 调试模式
|
||
DEBUG ?= 0
|
||
ifeq ($(DEBUG),1)
|
||
CFLAGS += -g -O0 -DCOCOON_DEBUG
|
||
endif
|
||
|
||
# 安装路径
|
||
PREFIX ?= /usr/local
|
||
BINDIR = $(PREFIX)/bin
|
||
|
||
# 源文件
|
||
SRCS = main.c server.c http.c static.c log.c config.c multipart.c tls.c http2.c access_log.c websocket.c platform.c middleware.c plugin.c proxy.c proxy_tls.c healthcheck.c middleware_ext.c load_balance.c grpc.c http3.c sse.c fastcgi.c fcgi_handler.c cache.c dashboard.c acme.c throttle.c
|
||
OBJS = $(SRCS:.c=.o)
|
||
TARGET = cocoon
|
||
|
||
# Windows (MinGW) 下链接 Winsock 库
|
||
ifeq ($(OS),Windows_NT)
|
||
LDFLAGS += -lws2_32
|
||
endif
|
||
|
||
# 单元测试
|
||
UNITY_SRC = tests/unity/unity.c
|
||
UNIT_TEST_DIR = tests/unit
|
||
UNIT_TEST_SRCS = $(wildcard $(UNIT_TEST_DIR)/test_*.c)
|
||
UNIT_TEST_OBJS = $(UNIT_TEST_SRCS:.c=.o)
|
||
UNIT_TEST_BINS = $(UNIT_TEST_SRCS:.c=)
|
||
|
||
# 默认目标
|
||
all: $(TARGET)
|
||
|
||
# 链接
|
||
$(TARGET): $(OBJS)
|
||
$(CC) $(OBJS) -o $@ $(LDFLAGS)
|
||
|
||
# 构建 coco 依赖（如果 submodule 已初始化）
|
||
deps:
|
||
@if [ ! -d "$(COCO_DIR)/build" ]; then \
|
||
echo "[Cocoon] 构建 coco 依赖..."; \
|
||
cd $(COCO_DIR) && mkdir -p build && cd build && cmake .. && $(MAKE); \
|
||
else \
|
||
echo "[Cocoon] coco 已构建，跳过"; \
|
||
fi
|
||
|
||
# 完整构建（先构建依赖，再构建项目）
|
||
build-all: deps $(TARGET)
|
||
|
||
# 集成测试（curl + bash）
|
||
test: $(TARGET)
|
||
@echo "[Cocoon] 运行集成测试..."
|
||
@./tests/integration_test.sh
|
||
|
||
# 集成测试别名（更直观）
|
||
integration-test: test
|
||
|
||
# 一键运行所有测试（单元 + 集成）
|
||
test-all: unit-test test
|
||
@echo "[Cocoon] 全部测试完成 ✓"
|
||
|
||
# 性能基准
|
||
bench: $(TARGET)
|
||
@echo "[Cocoon] 运行性能基准..."
|
||
@./tests/benchmark.sh
|
||
|
||
# 单元测试
|
||
unit-test: $(UNIT_TEST_BINS)
|
||
@echo "[Cocoon] 运行单元测试..."
|
||
@failed=0; \
|
||
for bin in $(UNIT_TEST_BINS); do \
|
||
echo "--- $$bin ---"; \
|
||
if ! ./$$bin; then \
|
||
failed=$$((failed + 1)); \
|
||
fi; \
|
||
done; \
|
||
if [ $$failed -gt 0 ]; then \
|
||
echo "[Cocoon] $$failed 个单元测试失败"; \
|
||
exit 1; \
|
||
else \
|
||
echo "[Cocoon] 全部单元测试通过 ✓"; \
|
||
fi
|
||
|
||
# 单元测试编译规则
|
||
$(UNIT_TEST_DIR)/test_server: $(UNIT_TEST_DIR)/test_server.c server.c http.c static.c log.c config.c multipart.c tls.c http2.c access_log.c websocket.c platform.c middleware.c plugin.c proxy.c proxy_tls.c healthcheck.c sse.c fastcgi.c fcgi_handler.c cache.c dashboard.c acme.c throttle.c $(UNITY_SRC)
|
||
$(CC) $(CFLAGS) -I. -I$(UNIT_TEST_DIR)/../unity -o $@ $(UNIT_TEST_DIR)/test_server.c server.c http.c static.c log.c config.c multipart.c tls.c http2.c access_log.c websocket.c platform.c middleware.c plugin.c proxy.c proxy_tls.c healthcheck.c sse.c fastcgi.c fcgi_handler.c cache.c dashboard.c acme.c throttle.c $(UNITY_SRC) $(LDFLAGS)
|
||
|
||
$(UNIT_TEST_DIR)/test_multipart: $(UNIT_TEST_DIR)/test_multipart.c multipart.c $(UNITY_SRC)
|
||
$(CC) $(CFLAGS) -I. -I$(UNIT_TEST_DIR)/../unity -o $@ $(UNIT_TEST_DIR)/test_multipart.c multipart.c $(UNITY_SRC) -lm
|
||
|
||
$(UNIT_TEST_DIR)/test_http: $(UNIT_TEST_DIR)/test_http.c http.c log.c $(UNITY_SRC)
|
||
$(CC) $(CFLAGS) -I. -I$(UNIT_TEST_DIR)/../unity -o $@ $(UNIT_TEST_DIR)/test_http.c http.c log.c $(UNITY_SRC) -lm
|
||
|
||
$(UNIT_TEST_DIR)/test_static: $(UNIT_TEST_DIR)/test_static.c http.c log.c tls.c access_log.c platform.c cache.c acme.c throttle.c $(UNITY_SRC)
|
||
$(CC) $(CFLAGS) -I. -I$(UNIT_TEST_DIR)/../unity -o $@ $(UNIT_TEST_DIR)/test_static.c http.c log.c tls.c access_log.c platform.c cache.c acme.c throttle.c $(UNITY_SRC) $(LDFLAGS)
|
||
|
||
$(UNIT_TEST_DIR)/test_websocket: $(UNIT_TEST_DIR)/test_websocket.c websocket.c log.c platform.c $(UNITY_SRC)
|
||
$(CC) $(CFLAGS) -I. -I$(UNIT_TEST_DIR)/../unity -o $@ $(UNIT_TEST_DIR)/test_websocket.c websocket.c log.c platform.c $(UNITY_SRC) $(LDFLAGS)
|
||
|
||
$(UNIT_TEST_DIR)/test_log: $(UNIT_TEST_DIR)/test_log.c log.c $(UNITY_SRC)
|
||
$(CC) $(CFLAGS) -I. -I$(UNIT_TEST_DIR)/../unity -o $@ $(UNIT_TEST_DIR)/test_log.c log.c $(UNITY_SRC) -lm
|
||
|
||
# 单元测试编译规则
|
||
$(UNIT_TEST_DIR)/test_healthcheck: $(UNIT_TEST_DIR)/test_healthcheck.c proxy.c proxy_tls.c http.c log.c platform.c throttle.c $(UNITY_SRC)
|
||
$(CC) $(CFLAGS) -I. -I$(UNIT_TEST_DIR)/../unity -o $@ $(UNIT_TEST_DIR)/test_healthcheck.c proxy.c proxy_tls.c http.c log.c platform.c throttle.c $(UNITY_SRC) $(LDFLAGS)
|
||
|
||
$(UNIT_TEST_DIR)/test_config: $(UNIT_TEST_DIR)/test_config.c config.c log.c access_log.c http.c throttle.c $(UNITY_SRC)
|
||
$(CC) $(CFLAGS) -I. -I$(UNIT_TEST_DIR)/../unity -o $@ $(UNIT_TEST_DIR)/test_config.c config.c log.c access_log.c http.c throttle.c $(UNITY_SRC) -lm
|
||
|
||
$(UNIT_TEST_DIR)/test_proxy: $(UNIT_TEST_DIR)/test_proxy.c proxy_tls.c http.c log.c platform.c throttle.c $(UNITY_SRC)
|
||
$(CC) $(CFLAGS) -I. -I$(UNIT_TEST_DIR)/../unity -o $@ $(UNIT_TEST_DIR)/test_proxy.c proxy_tls.c http.c log.c platform.c throttle.c $(UNITY_SRC) $(LDFLAGS)
|
||
|
||
# 扩展中间件测试
|
||
$(UNIT_TEST_DIR)/test_middleware_ext: $(UNIT_TEST_DIR)/test_middleware_ext.c middleware_ext.c http.c log.c platform.c throttle.c $(UNITY_SRC)
|
||
$(CC) $(CFLAGS) -I. -I$(UNIT_TEST_DIR)/../unity -o $@ $(UNIT_TEST_DIR)/test_middleware_ext.c middleware_ext.c http.c log.c platform.c throttle.c $(UNITY_SRC) $(LDFLAGS)
|
||
|
||
# gRPC 测试
|
||
$(UNIT_TEST_DIR)/test_grpc: $(UNIT_TEST_DIR)/test_grpc.c grpc.c http.c log.c platform.c $(UNITY_SRC)
|
||
$(CC) $(CFLAGS) -I. -I$(UNIT_TEST_DIR)/../unity -o $@ $(UNIT_TEST_DIR)/test_grpc.c grpc.c http.c log.c platform.c $(UNITY_SRC) -lm
|
||
|
||
# 负载均衡测试
|
||
$(UNIT_TEST_DIR)/test_load_balance: $(UNIT_TEST_DIR)/test_load_balance.c load_balance.c proxy.c proxy_tls.c http.c log.c platform.c throttle.c $(UNITY_SRC)
|
||
$(CC) $(CFLAGS) -I. -I$(UNIT_TEST_DIR)/../unity -o $@ $(UNIT_TEST_DIR)/test_load_balance.c load_balance.c proxy.c proxy_tls.c http.c log.c platform.c throttle.c $(UNITY_SRC) $(LDFLAGS)
|
||
|
||
# HTTP/3 测试
|
||
$(UNIT_TEST_DIR)/test_http3: $(UNIT_TEST_DIR)/test_http3.c http3.c http.c log.c platform.c $(UNITY_SRC)
|
||
$(CC) $(CFLAGS) -I. -I$(UNIT_TEST_DIR)/../unity -o $@ $(UNIT_TEST_DIR)/test_http3.c http3.c http.c log.c platform.c $(UNITY_SRC) $(LDFLAGS)
|
||
|
||
# SSE 测试
|
||
$(UNIT_TEST_DIR)/test_sse: $(UNIT_TEST_DIR)/test_sse.c sse.c log.c $(UNITY_SRC)
|
||
$(CC) $(CFLAGS) -I. -I$(UNIT_TEST_DIR)/../unity -o $@ $(UNIT_TEST_DIR)/test_sse.c sse.c log.c $(UNITY_SRC) $(LDFLAGS)
|
||
|
||
# FastCGI 测试
|
||
$(UNIT_TEST_DIR)/test_fastcgi: $(UNIT_TEST_DIR)/test_fastcgi.c fastcgi.c log.c $(UNITY_SRC)
|
||
$(CC) $(CFLAGS) -I. -I$(UNIT_TEST_DIR)/../unity -o $@ $(UNIT_TEST_DIR)/test_fastcgi.c fastcgi.c log.c $(UNITY_SRC) $(LDFLAGS)
|
||
|
||
# 缓存测试
|
||
$(UNIT_TEST_DIR)/test_cache: $(UNIT_TEST_DIR)/test_cache.c cache.c log.c $(UNITY_SRC)
|
||
$(CC) $(CFLAGS) -I. -I$(UNIT_TEST_DIR)/../unity -o $@ $(UNIT_TEST_DIR)/test_cache.c cache.c log.c $(UNITY_SRC) $(LDFLAGS)
|
||
|
||
# Dashboard 测试
|
||
$(UNIT_TEST_DIR)/test_dashboard: $(UNIT_TEST_DIR)/test_dashboard.c dashboard.c sse.c log.c $(UNITY_SRC)
|
||
$(CC) $(CFLAGS) -I. -I$(UNIT_TEST_DIR)/../unity -o $@ $(UNIT_TEST_DIR)/test_dashboard.c dashboard.c sse.c log.c $(UNITY_SRC) $(LDFLAGS)
|
||
|
||
# ACME 测试
|
||
$(UNIT_TEST_DIR)/test_acme: $(UNIT_TEST_DIR)/test_acme.c acme.c log.c platform.c $(UNITY_SRC)
|
||
$(CC) $(CFLAGS) -I. -I$(UNIT_TEST_DIR)/../unity -o $@ $(UNIT_TEST_DIR)/test_acme.c acme.c log.c platform.c $(UNITY_SRC) $(LDFLAGS)
|
||
|
||
# ACME 配置集成测试
|
||
$(UNIT_TEST_DIR)/test_acme_config: $(UNIT_TEST_DIR)/test_acme_config.c config.c acme.c log.c access_log.c http.c platform.c throttle.c $(UNITY_SRC)
|
||
$(CC) $(CFLAGS) -I. -I$(UNIT_TEST_DIR)/../unity -o $@ $(UNIT_TEST_DIR)/test_acme_config.c config.c acme.c log.c access_log.c http.c platform.c throttle.c $(UNITY_SRC) $(LDFLAGS)
|
||
|
||
# 安装
|
||
install: $(TARGET)
|
||
install -d $(BINDIR)
|
||
install -m 755 $(TARGET) $(BINDIR)/
|
||
|
||
# 卸载
|
||
uninstall:
|
||
rm -f $(BINDIR)/$(TARGET)
|
||
|
||
# 清理
|
||
clean:
|
||
rm -f $(OBJS) $(TARGET)
|
||
rm -f $(UNIT_TEST_OBJS) $(UNIT_TEST_BINS)
|
||
|
||
# 安装 systemd 服务文件
|
||
install-systemd: cocoon.service
|
||
install -d /etc/cocoon
|
||
install -d /var/log/cocoon
|
||
install -d /var/www
|
||
install -m 644 cocoon.service /etc/systemd/system/
|
||
@echo "[Cocoon] systemd 服务已安装"
|
||
@echo " 1. 编辑 /etc/cocoon/cocoon.json 配置服务器"
|
||
@echo " 2. 运行 'systemctl daemon-reload' 重新加载 systemd"
|
||
@echo " 3. 运行 'systemctl enable --now cocoon' 启动并启用服务"
|
||
|
||
# 卸载 systemd 服务
|
||
uninstall-systemd:
|
||
systemctl disable cocoon 2>/dev/null || true
|
||
rm -f /etc/systemd/system/cocoon.service
|
||
@echo "[Cocoon] systemd 服务已卸载"
|
||
|
||
# 重新构建
|
||
rebuild: clean all
|
||
|
||
.PHONY: all clean install uninstall rebuild deps build-all test integration-test test-all bench unit-test install-systemd uninstall-systemd
|