mirror of
https://gitee.com/wanwujie/sub2api
synced 2026-04-06 00:10:21 +08:00
Compare commits
6 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
e9ec2280ec | ||
|
|
bb7bfb6980 | ||
|
|
b66f97c100 | ||
|
|
b51ad0d893 | ||
|
|
4eb22d8ee9 | ||
|
|
2392e7cf99 |
1
.gitignore
vendored
1
.gitignore
vendored
@@ -92,3 +92,4 @@ backend/data/
|
|||||||
tests
|
tests
|
||||||
CLAUDE.md
|
CLAUDE.md
|
||||||
.claude
|
.claude
|
||||||
|
scripts
|
||||||
@@ -32,7 +32,8 @@ func RestartService() error {
|
|||||||
|
|
||||||
// The sub2api user has NOPASSWD sudo access for systemctl commands
|
// The sub2api user has NOPASSWD sudo access for systemctl commands
|
||||||
// (configured by install.sh in /etc/sudoers.d/sub2api).
|
// (configured by install.sh in /etc/sudoers.d/sub2api).
|
||||||
cmd := exec.Command("sudo", "systemctl", "restart", serviceName)
|
// Use -n (non-interactive) to prevent sudo from waiting for password input
|
||||||
|
cmd := exec.Command("sudo", "-n", "systemctl", "restart", serviceName)
|
||||||
if err := cmd.Start(); err != nil {
|
if err := cmd.Start(); err != nil {
|
||||||
return fmt.Errorf("failed to initiate service restart: %w", err)
|
return fmt.Errorf("failed to initiate service restart: %w", err)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -35,25 +35,25 @@ const (
|
|||||||
|
|
||||||
// allowedHeaders 白名单headers(参考CRS项目)
|
// allowedHeaders 白名单headers(参考CRS项目)
|
||||||
var allowedHeaders = map[string]bool{
|
var allowedHeaders = map[string]bool{
|
||||||
"accept": true,
|
"accept": true,
|
||||||
"x-stainless-retry-count": true,
|
"x-stainless-retry-count": true,
|
||||||
"x-stainless-timeout": true,
|
"x-stainless-timeout": true,
|
||||||
"x-stainless-lang": true,
|
"x-stainless-lang": true,
|
||||||
"x-stainless-package-version": true,
|
"x-stainless-package-version": true,
|
||||||
"x-stainless-os": true,
|
"x-stainless-os": true,
|
||||||
"x-stainless-arch": true,
|
"x-stainless-arch": true,
|
||||||
"x-stainless-runtime": true,
|
"x-stainless-runtime": true,
|
||||||
"x-stainless-runtime-version": true,
|
"x-stainless-runtime-version": true,
|
||||||
"x-stainless-helper-method": true,
|
"x-stainless-helper-method": true,
|
||||||
"anthropic-dangerous-direct-browser-access": true,
|
"anthropic-dangerous-direct-browser-access": true,
|
||||||
"anthropic-version": true,
|
"anthropic-version": true,
|
||||||
"x-app": true,
|
"x-app": true,
|
||||||
"anthropic-beta": true,
|
"anthropic-beta": true,
|
||||||
"accept-language": true,
|
"accept-language": true,
|
||||||
"sec-fetch-mode": true,
|
"sec-fetch-mode": true,
|
||||||
"accept-encoding": true,
|
"accept-encoding": true,
|
||||||
"user-agent": true,
|
"user-agent": true,
|
||||||
"content-type": true,
|
"content-type": true,
|
||||||
}
|
}
|
||||||
|
|
||||||
// ClaudeUsage 表示Claude API返回的usage信息
|
// ClaudeUsage 表示Claude API返回的usage信息
|
||||||
@@ -418,13 +418,19 @@ func (s *GatewayService) Forward(ctx context.Context, c *gin.Context, account *m
|
|||||||
}
|
}
|
||||||
|
|
||||||
// 构建上游请求
|
// 构建上游请求
|
||||||
upstreamReq, err := s.buildUpstreamRequest(ctx, c, account, body, token, tokenType)
|
upstreamResult, err := s.buildUpstreamRequest(ctx, c, account, body, token, tokenType)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 选择使用的client:如果有代理则使用独立的client,否则使用共享的httpClient
|
||||||
|
httpClient := s.httpClient
|
||||||
|
if upstreamResult.Client != nil {
|
||||||
|
httpClient = upstreamResult.Client
|
||||||
|
}
|
||||||
|
|
||||||
// 发送请求
|
// 发送请求
|
||||||
resp, err := s.httpClient.Do(upstreamReq)
|
resp, err := httpClient.Do(upstreamResult.Request)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("upstream request failed: %w", err)
|
return nil, fmt.Errorf("upstream request failed: %w", err)
|
||||||
}
|
}
|
||||||
@@ -437,11 +443,16 @@ func (s *GatewayService) Forward(ctx context.Context, c *gin.Context, account *m
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("token refresh failed: %w", err)
|
return nil, fmt.Errorf("token refresh failed: %w", err)
|
||||||
}
|
}
|
||||||
upstreamReq, err = s.buildUpstreamRequest(ctx, c, account, body, token, tokenType)
|
upstreamResult, err = s.buildUpstreamRequest(ctx, c, account, body, token, tokenType)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
resp, err = s.httpClient.Do(upstreamReq)
|
// 重试时也需要使用正确的client
|
||||||
|
httpClient = s.httpClient
|
||||||
|
if upstreamResult.Client != nil {
|
||||||
|
httpClient = upstreamResult.Client
|
||||||
|
}
|
||||||
|
resp, err = httpClient.Do(upstreamResult.Request)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("retry request failed: %w", err)
|
return nil, fmt.Errorf("retry request failed: %w", err)
|
||||||
}
|
}
|
||||||
@@ -480,7 +491,13 @@ func (s *GatewayService) Forward(ctx context.Context, c *gin.Context, account *m
|
|||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *GatewayService) buildUpstreamRequest(ctx context.Context, c *gin.Context, account *model.Account, body []byte, token, tokenType string) (*http.Request, error) {
|
// buildUpstreamRequestResult contains the request and optional custom client for proxy
|
||||||
|
type buildUpstreamRequestResult struct {
|
||||||
|
Request *http.Request
|
||||||
|
Client *http.Client // nil means use default s.httpClient
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *GatewayService) buildUpstreamRequest(ctx context.Context, c *gin.Context, account *model.Account, body []byte, token, tokenType string) (*buildUpstreamRequestResult, error) {
|
||||||
// 确定目标URL
|
// 确定目标URL
|
||||||
targetURL := claudeAPIURL
|
targetURL := claudeAPIURL
|
||||||
if account.Type == model.AccountTypeApiKey {
|
if account.Type == model.AccountTypeApiKey {
|
||||||
@@ -549,7 +566,8 @@ func (s *GatewayService) buildUpstreamRequest(ctx context.Context, c *gin.Contex
|
|||||||
req.Header.Set("anthropic-beta", s.getBetaHeader(body, c.GetHeader("anthropic-beta")))
|
req.Header.Set("anthropic-beta", s.getBetaHeader(body, c.GetHeader("anthropic-beta")))
|
||||||
}
|
}
|
||||||
|
|
||||||
// 配置代理
|
// 配置代理 - 创建独立的client避免并发修改共享httpClient
|
||||||
|
var customClient *http.Client
|
||||||
if account.ProxyID != nil && account.Proxy != nil {
|
if account.ProxyID != nil && account.Proxy != nil {
|
||||||
proxyURL := account.Proxy.URL()
|
proxyURL := account.Proxy.URL()
|
||||||
if proxyURL != "" {
|
if proxyURL != "" {
|
||||||
@@ -566,12 +584,18 @@ func (s *GatewayService) buildUpstreamRequest(ctx context.Context, c *gin.Contex
|
|||||||
IdleConnTimeout: 90 * time.Second,
|
IdleConnTimeout: 90 * time.Second,
|
||||||
ResponseHeaderTimeout: responseHeaderTimeout,
|
ResponseHeaderTimeout: responseHeaderTimeout,
|
||||||
}
|
}
|
||||||
s.httpClient.Transport = transport
|
// 创建独立的client,避免并发时修改共享的s.httpClient.Transport
|
||||||
|
customClient = &http.Client{
|
||||||
|
Transport: transport,
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return req, nil
|
return &buildUpstreamRequestResult{
|
||||||
|
Request: req,
|
||||||
|
Client: customClient,
|
||||||
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// getBetaHeader 处理anthropic-beta header
|
// getBetaHeader 处理anthropic-beta header
|
||||||
|
|||||||
@@ -483,9 +483,24 @@ download_and_extract() {
|
|||||||
create_user() {
|
create_user() {
|
||||||
if id "$SERVICE_USER" &>/dev/null; then
|
if id "$SERVICE_USER" &>/dev/null; then
|
||||||
print_info "$(msg 'user_exists'): $SERVICE_USER"
|
print_info "$(msg 'user_exists'): $SERVICE_USER"
|
||||||
|
# Fix: Ensure existing user has /bin/sh shell for sudo to work
|
||||||
|
# Previous versions used /bin/false which prevents sudo execution
|
||||||
|
local current_shell
|
||||||
|
current_shell=$(getent passwd "$SERVICE_USER" 2>/dev/null | cut -d: -f7)
|
||||||
|
if [ "$current_shell" = "/bin/false" ] || [ "$current_shell" = "/sbin/nologin" ]; then
|
||||||
|
print_info "Fixing user shell for sudo compatibility..."
|
||||||
|
if usermod -s /bin/sh "$SERVICE_USER" 2>/dev/null; then
|
||||||
|
print_success "User shell updated to /bin/sh"
|
||||||
|
else
|
||||||
|
print_warning "Failed to update user shell. Service restart may not work automatically."
|
||||||
|
print_warning "Manual fix: sudo usermod -s /bin/sh $SERVICE_USER"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
else
|
else
|
||||||
print_info "$(msg 'creating_user') $SERVICE_USER..."
|
print_info "$(msg 'creating_user') $SERVICE_USER..."
|
||||||
useradd -r -s /bin/false -d "$INSTALL_DIR" "$SERVICE_USER"
|
# Use /bin/sh instead of /bin/false to allow sudo execution
|
||||||
|
# The user still cannot login interactively (no password set)
|
||||||
|
useradd -r -s /bin/sh -d "$INSTALL_DIR" "$SERVICE_USER"
|
||||||
print_success "$(msg 'user_created')"
|
print_success "$(msg 'user_created')"
|
||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
@@ -510,18 +525,18 @@ setup_directories() {
|
|||||||
setup_sudoers() {
|
setup_sudoers() {
|
||||||
print_info "$(msg 'setting_up_sudoers')"
|
print_info "$(msg 'setting_up_sudoers')"
|
||||||
|
|
||||||
# Check if sudoers file exists in install dir
|
# Always generate sudoers file from script (not from tar.gz)
|
||||||
if [ -f "$INSTALL_DIR/sub2api-sudoers" ]; then
|
# This ensures the latest configuration is used even with older releases
|
||||||
cp "$INSTALL_DIR/sub2api-sudoers" /etc/sudoers.d/sub2api
|
# Support both /bin/systemctl and /usr/bin/systemctl for different distros
|
||||||
else
|
cat > /etc/sudoers.d/sub2api << 'EOF'
|
||||||
# Create sudoers file
|
|
||||||
cat > /etc/sudoers.d/sub2api << 'EOF'
|
|
||||||
# Sudoers configuration for Sub2API
|
# Sudoers configuration for Sub2API
|
||||||
sub2api ALL=(ALL) NOPASSWD: /bin/systemctl restart sub2api
|
sub2api ALL=(ALL) NOPASSWD: /bin/systemctl restart sub2api
|
||||||
sub2api ALL=(ALL) NOPASSWD: /bin/systemctl stop sub2api
|
sub2api ALL=(ALL) NOPASSWD: /bin/systemctl stop sub2api
|
||||||
sub2api ALL=(ALL) NOPASSWD: /bin/systemctl start sub2api
|
sub2api ALL=(ALL) NOPASSWD: /bin/systemctl start sub2api
|
||||||
|
sub2api ALL=(ALL) NOPASSWD: /usr/bin/systemctl restart sub2api
|
||||||
|
sub2api ALL=(ALL) NOPASSWD: /usr/bin/systemctl stop sub2api
|
||||||
|
sub2api ALL=(ALL) NOPASSWD: /usr/bin/systemctl start sub2api
|
||||||
EOF
|
EOF
|
||||||
fi
|
|
||||||
|
|
||||||
# Set correct permissions (required for sudoers files)
|
# Set correct permissions (required for sudoers files)
|
||||||
chmod 440 /etc/sudoers.d/sub2api
|
chmod 440 /etc/sudoers.d/sub2api
|
||||||
|
|||||||
@@ -8,6 +8,10 @@
|
|||||||
# SECURITY NOTE: This grants limited sudo access only for service management
|
# SECURITY NOTE: This grants limited sudo access only for service management
|
||||||
|
|
||||||
# Allow sub2api user to restart the service without password
|
# Allow sub2api user to restart the service without password
|
||||||
|
# Support both /bin/systemctl (Debian/Ubuntu) and /usr/bin/systemctl (RHEL/CentOS)
|
||||||
sub2api ALL=(ALL) NOPASSWD: /bin/systemctl restart sub2api
|
sub2api ALL=(ALL) NOPASSWD: /bin/systemctl restart sub2api
|
||||||
sub2api ALL=(ALL) NOPASSWD: /bin/systemctl stop sub2api
|
sub2api ALL=(ALL) NOPASSWD: /bin/systemctl stop sub2api
|
||||||
sub2api ALL=(ALL) NOPASSWD: /bin/systemctl start sub2api
|
sub2api ALL=(ALL) NOPASSWD: /bin/systemctl start sub2api
|
||||||
|
sub2api ALL=(ALL) NOPASSWD: /usr/bin/systemctl restart sub2api
|
||||||
|
sub2api ALL=(ALL) NOPASSWD: /usr/bin/systemctl stop sub2api
|
||||||
|
sub2api ALL=(ALL) NOPASSWD: /usr/bin/systemctl start sub2api
|
||||||
|
|||||||
Reference in New Issue
Block a user