181 lines
7.6 KiB
Go
181 lines
7.6 KiB
Go
package service
|
||
|
||
import (
|
||
"fmt"
|
||
"net/http"
|
||
"time"
|
||
|
||
"github.com/miaomint/port-manager/pkg/types"
|
||
)
|
||
|
||
// ServiceIdentifier 服务识别器
|
||
type ServiceIdentifier struct {
|
||
wellKnownPorts map[int]types.ServiceMapping
|
||
}
|
||
|
||
// NewServiceIdentifier 创建新的服务识别器
|
||
func NewServiceIdentifier() *ServiceIdentifier {
|
||
return &ServiceIdentifier{
|
||
wellKnownPorts: initWellKnownPorts(),
|
||
}
|
||
}
|
||
|
||
// IdentifyService 识别端口对应的服务
|
||
func (si *ServiceIdentifier) IdentifyService(port *types.PortInfo) *types.PortInfo {
|
||
// 复制端口信息
|
||
result := *port
|
||
|
||
// 查找已知服务
|
||
if mapping, exists := si.wellKnownPorts[port.Port]; exists {
|
||
result.ServiceName = mapping.ServiceName
|
||
|
||
// 如果是HTTP/HTTPS服务，尝试构建访问URL
|
||
if mapping.IsHTTP || mapping.IsHTTPS {
|
||
protocol := "http"
|
||
if mapping.IsHTTPS {
|
||
protocol = "https"
|
||
}
|
||
result.ServiceURL = fmt.Sprintf("%s://localhost:%d", protocol, port.Port)
|
||
|
||
// 验证HTTP服务是否可访问
|
||
if si.isHTTPServiceActive(result.ServiceURL) {
|
||
result.ServiceName = fmt.Sprintf("%s (可访问)", result.ServiceName)
|
||
}
|
||
}
|
||
} else {
|
||
// 尝试探测HTTP服务
|
||
if si.probeHTTPService(port.Port) {
|
||
result.ServiceName = "HTTP服务"
|
||
result.ServiceURL = fmt.Sprintf("http://localhost:%d", port.Port)
|
||
} else if si.probeHTTPSService(port.Port) {
|
||
result.ServiceName = "HTTPS服务"
|
||
result.ServiceURL = fmt.Sprintf("https://localhost:%d", port.Port)
|
||
} else {
|
||
result.ServiceName = "未知服务"
|
||
}
|
||
}
|
||
|
||
return &result
|
||
}
|
||
|
||
// probeHTTPService 探测HTTP服务
|
||
func (si *ServiceIdentifier) probeHTTPService(port int) bool {
|
||
url := fmt.Sprintf("http://localhost:%d", port)
|
||
return si.isHTTPServiceActive(url)
|
||
}
|
||
|
||
// probeHTTPSService 探测HTTPS服务
|
||
func (si *ServiceIdentifier) probeHTTPSService(port int) bool {
|
||
url := fmt.Sprintf("https://localhost:%d", port)
|
||
return si.isHTTPServiceActive(url)
|
||
}
|
||
|
||
// isHTTPServiceActive 检查HTTP服务是否活跃
|
||
func (si *ServiceIdentifier) isHTTPServiceActive(url string) bool {
|
||
client := &http.Client{
|
||
Timeout: 2 * time.Second,
|
||
}
|
||
|
||
resp, err := client.Get(url)
|
||
if err != nil {
|
||
return false
|
||
}
|
||
defer resp.Body.Close()
|
||
|
||
// 任何HTTP响应都认为是活跃的HTTP服务
|
||
return true
|
||
}
|
||
|
||
// initWellKnownPorts 初始化已知端口映射
|
||
func initWellKnownPorts() map[int]types.ServiceMapping {
|
||
return map[int]types.ServiceMapping{
|
||
// Web服务
|
||
80: {Port: 80, ServiceName: "HTTP", Description: "超文本传输协议", IsHTTP: true},
|
||
443: {Port: 443, ServiceName: "HTTPS", Description: "安全超文本传输协议", IsHTTPS: true},
|
||
8080: {Port: 8080, ServiceName: "HTTP代理", Description: "HTTP代理服务", IsHTTP: true},
|
||
8443: {Port: 8443, ServiceName: "HTTPS代理", Description: "HTTPS代理服务", IsHTTPS: true},
|
||
3000: {Port: 3000, ServiceName: "开发服务器", Description: "React/Node.js开发服务器", IsHTTP: true},
|
||
3001: {Port: 3001, ServiceName: "开发服务器", Description: "开发服务器", IsHTTP: true},
|
||
4000: {Port: 4000, ServiceName: "开发服务器", Description: "Jekyll/Hugo开发服务器", IsHTTP: true},
|
||
8000: {Port: 8000, ServiceName: "HTTP服务", Description: "Python HTTP服务器", IsHTTP: true},
|
||
9000: {Port: 9000, ServiceName: "PHP-FPM", Description: "PHP FastCGI Process Manager", IsHTTP: true},
|
||
|
||
// 数据库
|
||
3306: {Port: 3306, ServiceName: "MySQL", Description: "MySQL数据库服务器"},
|
||
5432: {Port: 5432, ServiceName: "PostgreSQL", Description: "PostgreSQL数据库服务器"},
|
||
27017: {Port: 27017, ServiceName: "MongoDB", Description: "MongoDB数据库服务器"},
|
||
6379: {Port: 6379, ServiceName: "Redis", Description: "Redis内存数据库"},
|
||
11211: {Port: 11211, ServiceName: "Memcached", Description: "Memcached缓存服务器"},
|
||
|
||
// 远程访问
|
||
22: {Port: 22, ServiceName: "SSH", Description: "安全外壳协议"},
|
||
23: {Port: 23, ServiceName: "Telnet", Description: "远程登录协议"},
|
||
3389: {Port: 3389, ServiceName: "RDP", Description: "远程桌面协议"},
|
||
5900: {Port: 5900, ServiceName: "VNC", Description: "虚拟网络计算"},
|
||
|
||
// 邮件服务
|
||
25: {Port: 25, ServiceName: "SMTP", Description: "简单邮件传输协议"},
|
||
110: {Port: 110, ServiceName: "POP3", Description: "邮局协议版本3"},
|
||
143: {Port: 143, ServiceName: "IMAP", Description: "互联网消息访问协议"},
|
||
465: {Port: 465, ServiceName: "SMTPS", Description: "安全SMTP"},
|
||
587: {Port: 587, ServiceName: "SMTP-Submission", Description: "邮件提交"},
|
||
993: {Port: 993, ServiceName: "IMAPS", Description: "安全IMAP"},
|
||
995: {Port: 995, ServiceName: "POP3S", Description: "安全POP3"},
|
||
|
||
// FTP
|
||
20: {Port: 20, ServiceName: "FTP-Data", Description: "文件传输协议数据"},
|
||
21: {Port: 21, ServiceName: "FTP", Description: "文件传输协议"},
|
||
|
||
// DNS
|
||
53: {Port: 53, ServiceName: "DNS", Description: "域名系统"},
|
||
|
||
// DHCP
|
||
67: {Port: 67, ServiceName: "DHCP-Server", Description: "动态主机配置协议服务器"},
|
||
68: {Port: 68, ServiceName: "DHCP-Client", Description: "动态主机配置协议客户端"},
|
||
|
||
// NTP
|
||
123: {Port: 123, ServiceName: "NTP", Description: "网络时间协议"},
|
||
|
||
// SNMP
|
||
161: {Port: 161, ServiceName: "SNMP", Description: "简单网络管理协议"},
|
||
162: {Port: 162, ServiceName: "SNMP-Trap", Description: "SNMP陷阱"},
|
||
|
||
// LDAP
|
||
389: {Port: 389, ServiceName: "LDAP", Description: "轻量目录访问协议"},
|
||
636: {Port: 636, ServiceName: "LDAPS", Description: "安全LDAP"},
|
||
|
||
// NAS相关服务
|
||
139: {Port: 139, ServiceName: "NetBIOS-SSN", Description: "NetBIOS会话服务"},
|
||
445: {Port: 445, ServiceName: "SMB", Description: "服务器消息块"},
|
||
548: {Port: 548, ServiceName: "AFP", Description: "Apple文件协议"},
|
||
2049: {Port: 2049, ServiceName: "NFS", Description: "网络文件系统"},
|
||
|
||
// 开发工具
|
||
9090: {Port: 9090, ServiceName: "Prometheus", Description: "Prometheus监控系统", IsHTTP: true},
|
||
5601: {Port: 5601, ServiceName: "Kibana", Description: "Kibana可视化平台", IsHTTP: true},
|
||
9200: {Port: 9200, ServiceName: "Elasticsearch", Description: "Elasticsearch搜索引擎", IsHTTP: true},
|
||
8888: {Port: 8888, ServiceName: "Jupyter", Description: "Jupyter Notebook", IsHTTP: true},
|
||
4040: {Port: 4040, ServiceName: "Spark UI", Description: "Apache Spark Web UI", IsHTTP: true},
|
||
|
||
// 消息队列
|
||
5672: {Port: 5672, ServiceName: "RabbitMQ", Description: "RabbitMQ消息队列"},
|
||
15672: {Port: 15672, ServiceName: "RabbitMQ管理", Description: "RabbitMQ管理界面", IsHTTP: true},
|
||
9092: {Port: 9092, ServiceName: "Kafka", Description: "Apache Kafka"},
|
||
|
||
// 容器和编排
|
||
2375: {Port: 2375, ServiceName: "Docker", Description: "Docker守护进程(非安全)"},
|
||
2376: {Port: 2376, ServiceName: "Docker", Description: "Docker守护进程(TLS)"},
|
||
6443: {Port: 6443, ServiceName: "Kubernetes API", Description: "Kubernetes API服务器"},
|
||
10250: {Port: 10250, ServiceName: "kubelet", Description: "Kubernetes kubelet"},
|
||
|
||
// Synology NAS特定端口
|
||
5000: {Port: 5000, ServiceName: "DSM", Description: "Synology DiskStation Manager", IsHTTP: true},
|
||
5001: {Port: 5001, ServiceName: "DSM (HTTPS)", Description: "Synology DiskStation Manager", IsHTTPS: true},
|
||
6690: {Port: 6690, ServiceName: "Cloud Station", Description: "Synology Cloud Station"},
|
||
9117: {Port: 9117, ServiceName: "Jackett", Description: "Jackett种子搜索", IsHTTP: true},
|
||
8989: {Port: 8989, ServiceName: "Sonarr", Description: "Sonarr自动下载", IsHTTP: true},
|
||
7878: {Port: 7878, ServiceName: "Radarr", Description: "Radarr电影下载", IsHTTP: true},
|
||
8686: {Port: 8686, ServiceName: "Lidarr", Description: "Lidarr音乐下载", IsHTTP: true},
|
||
}
|
||
}
|