酷代码 AI
菜单
全部AI  /  热门  /  AI写代码神器
服务商

写一个ollama服务器发现程序 语言方向:Go 系统环境:Windows

以下是使用 Go 编写的 Ollama 服务器发现程序,适用于 Windows 系统环境。该程序会尝试扫描本地网络中运行 Ollama 的服务器实例(默认端口为 `11434`)。 --- 📦 功能说明 - 扫描局域网内所有 IP 地址的 `11434` 端口。 - 检查是否返回类似 Ollama 的响应(例如 `/api/tags` 接口)。 - 输出可能的 Ollama 服务器地址。 --- ✅ 依赖库 无需额外安装第三方包,标准库即可完成任务。 --- 💻 示例代码 ```go package main import ( "fmt" "net" "time" "strings" "io/ioutil" "bufio" "os" ) const ( port = "11434" // Ollama 默认端口 timeout = 2 * time.Second // 连接超时时间 apiEndpoint = "/api/tags" // Ollama API 标志性接口 localBaseIP = "192.168.1." // 修改为你当前局域网的网段 ipRangeStart = 1 ipRangeEnd = 254 ) func isOllamaServer(ip string) bool { url := fmt.Sprintf("http://%s:%s%s", ip, port, apiEndpoint) conn, err := net.DialTimeout("tcp", fmt.Sprintf("%s:%s", ip, port), timeout) if err != nil { return false } defer conn.Close() // 发送 HTTP GET 请求 fmt.Fprintf(conn, "GET %s HTTP/1.1\r\nHost: %s\r\n\r\n", apiEndpoint, ip) // 读取响应 reader := bufio.NewReader(conn) resp, err := reader.ReadString('\n') if err != nil { return false } // 判断是否是 HTTP 响应 if strings.Contains(resp, "HTTP") { fmt.Printf("[+] Found potential Ollama server at: %s\n", url) return true } return false } func getLocalIP() (string, error) { addrs, err := net.InterfaceAddrs() if err != nil { return "", err } for _, addr := range addrs { if ipnet, ok := addr.(*net.IPNet); ok && !ipnet.IP.IsLoopback() { if ipnet.IP.To4() != nil { return ipnet.IP.String(), nil } } } return "", fmt.Errorf("no IP address found") } func scanNetwork(baseIP string) { fmt.Printf("[*] Scanning network for Ollama servers on port %s...\n", port) for i := ipRangeStart; i <= ipRangeEnd; i++ { ip := fmt.Sprintf("%s%d", baseIP, i) go func(ip string) { isOllamaServer(ip) }(ip) time.Sleep(10 * time.Millisecond) // 控制并发节奏 } } func main() { // 获取本机 IP 并推断局域网网段 localIP, err := getLocalIP() if err != nil { fmt.Println("[-] Failed to get local IP:", err) os.Exit(1) } // 提取前缀作为扫描目标 baseIP := localIP[:strings.LastIndex(localIP, ".")+1] fmt.Printf("[i] Using network base IP: %s\n", baseIP) scanNetwork(baseIP) // 防止主协程退出 fmt.Println("[*] Scan completed. Press Enter to exit.") fmt.Scanln() } ``` --- 🔍 注意事项 1. 网段适配:请根据你实际连接的局域网修改 `localBaseIP` 或自动提取逻辑。 2. 防火墙设置:目标主机需开放 `11434` 端口,否则无法探测到。 3. 权限问题:在某些环境中,需要管理员权限才能进行 TCP 扫描。 --- 如果你希望增加更多功能,比如输出 JSON、保存结果或图形界面支持,请告诉我,我可以继续扩展这个程序。 [2025-07-04 19:24:53 | AI写代码神器 | 921点数解答]

相关提问