用Go语言重写Linux系统命令 -- nc简化版

ops/2024/12/12 21:21:58/

用Go语言重写Linux系统命令 – nc简化版

1. 引言

netcat，简称 nc，被誉为网络工具中的“瑞士军刀”，是网络调试与分析的利器。它的功能十分强大，然而平时我们经常使用的就是他的连通性测试功能，但是nc是被设计用来测试TCP和UDP的，对于HTTP支持不是太好，于是，我决定用 Go 语言重新实现一个简化版的 nc，专注于联通性测试功能，并扩展支持测试HTTP和Ping的连通性测试。通过这个项目，你不仅能深入学习网络编程，还能体验用 Go 语言开发系统工具的乐趣！

小目标：让我们的工具小而美，轻松搞定网络连通性调试！

2. 需求分析

在开始编写代码之前，我们需要明确工具的功能和设计目标：

  • 核心功能：

    1. TCP 测试：检查指定 IP 和端口是否可建立连接，支持数据发送与接收。
    2. UDP 测试：发送 UDP 数据包并接收响应。
    3. HTTP 测试：测试指定 URL 是否可访问，支持自定义请求方法（如 GET/POST）。
    4. Ping 测试：检查目标 IP 是否可达，并测量响应时间。

  • 用户体验设计：

    • 使用命令行参数指定测试类型和选项（如目标 IP、端口、超时）。
    • 错误提示清晰直观，方便排查问题。

  • 性能目标：快速响应，保持简洁。

3. 项目初始化

3.1 安装开发环境

确保已安装 Go 编译器（推荐版本 ≥ 1.20）。执行以下命令检查 Go 是否可用：

go version

3.2 初始化项目

创建项目目录并初始化 Go 模块：

mkdir gonc
cd gonc
go mod init gonc

项目结构：

gonc/
├── main.go        # 入口文件
├── tcp.go         # TCP 测试模块
├── udp.go         # UDP 测试模块
├── http.go        # HTTP 测试模块
├── ping.go        # Ping 测试模块
└── go.mod         # 依赖管理文件

4. 模块实现

4.1 实现 TCP 测试功能

在 tcp.go 中编写代码，建立 TCP 连接，发送和接收数据：

package myncimport ("fmt""log""net""time"
)// TestTCP 测试TCP端口是否连通，并支持发送和接收数据
func FireTCP(ip string, port, timeout int, data string) error {address := fmt.Sprintf("%s:%d", ip, port)// 设置连接超时时间conn, err := net.DialTimeout("tcp", address, time.Duration(timeout)*time.Second)if err != nil {log.Printf("Failed to connect to %s: %v\n", address, err)return err}defer conn.Close()fmt.Printf("Successfully connected to %s\n", address)// 发送数据if data != "" {fmt.Printf("Sending data: %s\n", data)_, err = conn.Write([]byte(data))if err != nil {log.Printf("Failed to send data: %v\n", err)} else {fmt.Println("Data sent successfully")}}// 接收响应数据buffer := make([]byte, 1024)conn.SetReadDeadline(time.Now().Add(time.Duration(timeout) * time.Second))// 读取响应数据// 如果超时则提示用户, 服务器没有发送数据n, err := conn.Read(buffer)if err != nil {if netErr, ok := err.(net.Error); ok && netErr.Timeout() {fmt.Println("No response from server")} else {log.Printf("Failed to read response: %v\n", err)}} else {fmt.Printf("Received: %s\n", string(buffer[:n]))}return err
}

4.2 实现 UDP 测试功能

在 udp.go 中实现 UDP 数据包的发送与接收：

package myncimport ("fmt""log""net""time"
)// TestUDP 测试UDP端口是否连通，并支持发送和接收数据
func FireUDP(ip string, port, timeout int, data string) error {address := fmt.Sprintf("%s:%d", ip, port)conn, err := net.Dial("udp", address)if err != nil {log.Printf("Failed to connect to %s: %v\n", address, err)return err}defer conn.Close()fmt.Printf("Successfully connected to %s\n", address)// 设置写超时conn.SetWriteDeadline(time.Now().Add(time.Duration(timeout) * time.Second))_, err = conn.Write([]byte(data))if err != nil {log.Printf("Failed to send data: %v\n", err)return err}fmt.Println("Data sent successfully")// 设置读超时conn.SetReadDeadline(time.Now().Add(time.Duration(timeout) * time.Second))buffer := make([]byte, 1024)n, err := conn.Read(buffer)if err != nil {log.Printf("No response received: %v\n", err)} else {fmt.Printf("Received: %s\n", string(buffer[:n]))}return err
}

4.3 实现 HTTP 测试功能

在 http.go 中构建 HTTP 请求：

package myncimport ("fmt""io""log""net/http""time"
)// TestHTTP 测试HTTP/HTTPS连接
func FireHTTP(url, method string, headers map[string]string, timeout int) error {client := &http.Client{Timeout: time.Duration(timeout) * time.Second,}// 创建HTTP请求req, err := http.NewRequest(method, url, nil)if err != nil {log.Printf("Failed to create HTTP request: %v\n", err)return err}// 设置请求头for key, value := range headers {req.Header.Set(key, value)}// 发送HTTP请求resp, err := client.Do(req)if err != nil {log.Printf("HTTP request failed: %v\n", err)return err}defer resp.Body.Close()// 读取响应内容body, err := io.ReadAll(resp.Body)if err != nil {log.Printf("Failed to read response body: %v\n", err)return err}fmt.Printf("HTTP Status: %s\n", resp.Status)fmt.Printf("Response Body:\n%s\n", string(body))return err
}

4.4 实现 Ping 测试功能

在 ping.go 中：


import ("fmt""log""net""os""time"
)const icmpEchoRequest = 8// Ping 测试目标IP是否可达
func Ping(ip string, timeout int) error {conn, err := net.Dial("ip4:icmp", ip)if err != nil {log.Printf("Failed to connect to %s: %v\n", ip, err)return err}defer conn.Close()// 构造ICMP请求包id := os.Getpid() & 0xffffseq := 1msg := []byte{icmpEchoRequest, 0, 0, 0, // ICMP Type, Code, and Checksum placeholderbyte(id >> 8), byte(id & 0xff), // Identifierbyte(seq >> 8), byte(seq & 0xff), // Sequence number}checksum := calculateChecksum(msg)msg[2] = byte(checksum >> 8)msg[3] = byte(checksum & 0xff)// 设置写超时conn.SetDeadline(time.Now().Add(time.Duration(timeout) * time.Second))start := time.Now()_, err = conn.Write(msg)if err != nil {log.Printf("Failed to send ICMP request: %v\n", err)return err}// 接收ICMP响应buffer := make([]byte, 1024)_, err = conn.Read(buffer)if err != nil {log.Printf("Failed to receive ICMP response: %v\n", err)return err}elapsed := time.Since(start)fmt.Printf("Ping %s successful! Time=%v\n", ip, elapsed)return err
}// calculateChecksum 计算ICMP校验和
func calculateChecksum(data []byte) uint16 {var sum intfor i := 0; i < len(data)-1; i += 2 {sum += int(data[i])<<8 | int(data[i+1])}if len(data)%2 == 1 {sum += int(data[len(data)-1]) << 8}for (sum >> 16) > 0 {sum = (sum >> 16) + (sum & 0xffff)}return uint16(^sum)
}

4.5 命令行解析与功能整合

在 main.go 中，将上述模块整合，并解析命令行参数：

package mainimport ("flag""fmt""net/url""os""strconv""strings""mync"
)func main() {// 定义命令行参数var method, data stringvar httpHeaders httpHeaderMapvar timeout intflag.StringVar(&method, "method", "GET", "HTTP method to use (default: GET)")flag.StringVar(&data, "data", "", "Data to send (for TCP or UDP)")flag.Var(&httpHeaders, "header", "Custom HTTP header (key=value pairs, can use multiple times)")flag.IntVar(&timeout, "timeout", 5, "Timeout in seconds (default: 5)")// 自定义 Usageflag.Usage = func() {fmt.Fprintf(os.Stderr, "Usage: %s <protocol>://<target> [options]\n\n", os.Args[0])fmt.Fprintf(os.Stderr, "options:\n")fmt.Fprintf(os.Stderr, "  -method <method>     HTTP method to use (default: GET)\n")fmt.Fprintf(os.Stderr, "  -data <data>         Data to send (for TCP or UDP)\n")fmt.Fprintf(os.Stderr, "  -header <header>   Custom HTTP header (key=value pairs, can use multiple times)\n")fmt.Fprintf(os.Stderr, "  -timeout <seconds>   Timeout in seconds (default: 5)\n")fmt.Fprintf(os.Stderr, "Examples:\n")fmt.Fprintf(os.Stderr, "  Test TCP connection:\n")fmt.Fprintf(os.Stderr, "    %s tcp://192.168.1.1:80 -data \"Hello, TCP!\" -timeout 10\n", os.Args[0])fmt.Fprintf(os.Stderr, "  Test UDP connection:\n")fmt.Fprintf(os.Stderr, "    %s udp://192.168.1.1:53 -data \"Hello, UDP!\" -timeout 5\n", os.Args[0])fmt.Fprintf(os.Stderr, "  Test HTTP connection:\n")fmt.Fprintf(os.Stderr, "    %s http://example.com -method GET -timeout 5\n", os.Args[0])fmt.Fprintf(os.Stderr, "  Ping a target:\n")fmt.Fprintf(os.Stderr, "    %s ping://192.168.1.1 -timeout 3\n", os.Args[0])}// 解析 flag 参数flag.Parse()// 必须至少有一个非 flag 参数args := flag.Args()if len(args) < 1 {fmt.Println("Error: Missing protocol and target. Example usage: tcp://192.168.1.1:80")flag.Usage()os.Exit(1)}// 解析第一个参数为协议和目标targetURL := args[0]parsedURL, err := url.Parse(targetURL)if err != nil || parsedURL.Scheme == "" || parsedURL.Host == "" {fmt.Printf("Error: Invalid protocol or target format: %s\n", targetURL)flag.Usage()os.Exit(1)}protocol := parsedURL.Schemetarget := parsedURL.Host// 根据协议执行对应的功能switch protocol {case "tcp":ip, port, err := parseTarget(target)if err != nil {fmt.Println("Error:", err)os.Exit(1)}if err := mync.FireTCP(ip, port, timeout, data); err != nil {fmt.Println("TCP Test Error:", err)}case "udp":ip, port, err := parseTarget(target)if err != nil {fmt.Println("Error:", err)os.Exit(1)}if err := mync.FireUDP(ip, port, timeout, data); err != nil {fmt.Println("UDP Test Error:", err)}case "http", "https":headerMap := httpHeaders.ToMap()if err := mync.FireHTTP(targetURL, method, headerMap, timeout); err != nil {fmt.Println("HTTP Test Error:", err)}case "ping":if err := mync.Ping(target, timeout); err != nil {fmt.Println("Ping Test Error:", err)}default:fmt.Printf("Error: Unsupported protocol: %s\n", protocol)flag.Usage()os.Exit(1)}
}// parseTarget 解析 <ip>:<port> 格式
func parseTarget(target string) (string, int, error) {parts := strings.Split(target, ":")if len(parts) != 2 {return "", 0, fmt.Errorf("invalid target format: %s, expected format <ip>:<port>", target)}ip := parts[0]port, err := strconv.Atoi(parts[1])if err != nil {return "", 0, fmt.Errorf("invalid port number: %s", parts[1])}return ip, port, nil
}// httpHeaderMap 用于保存用户指定的多个 -header 参数
type httpHeaderMap []stringfunc (h *httpHeaderMap) String() string {return strings.Join(*h, ", ")
}func (h *httpHeaderMap) Set(value string) error {*h = append(*h, value)return nil
}// ToMap 将 httpHeaderMap 转换为 map
func (h *httpHeaderMap) ToMap() map[string]string {headerMap := make(map[string]string)for _, header := range *h {parts := strings.SplitN(header, "=", 2)if len(parts) == 2 {headerMap[parts[0]] = parts[1]}}return headerMap
}

5 测试

#!/bin/bashset -eip=192.168.100.10
port=3333timeout=2# 测试tcp端口
echo ">>> Testing tcp port $port..."
./gonc tcp://$ip:$port -data "Hello" -timeout $timeout# 测试udp端口
echo ">>> Testing udp port $port..."
./gonc udp://$ip:$port -data "Hello" -timeout $timeout# 测试http服务
echo ">>> Testing http service..."
./gonc http://example.com -method GET -timeout $timeout# 测试ping
echo ">>> Testing ping..."
sudo ./gonc ping://$ip -timeout $timeout

由于测试ping时使用了原始套接字,所以需要root权限


http://www.ppmy.cn/ops/141343.html

相关文章

【数学建模】线性规划问题及Matlab求解

【数学建模】线性规划问题及Matlab求解

问题一 题目&#xff1a; 求解下列线性规划问题 解答&#xff1a; 先将题目中求最大值转化为求最小值&#xff0c;则有 我们就可以得到系数列向量: 我们对问题中所给出的不等式约束进行标准化则得到了 就有不等式约束条件下的变系数矩阵和常系数矩阵分别为&#xff1a; 等式…
阅读更多...
wordpress网站使用Linux宝塔面板和SQL命令行导入导出超过50M限制的数据库

wordpress网站使用Linux宝塔面板和SQL命令行导入导出超过50M限制的数据库

wordpress网站使用Linux宝塔面板添加mysql数据库&#xff0c;使用phpMyAdmin工具导入sql数据库文件&#xff0c;会有最大限制50M。即使把sql数据库文件压缩为gzip或zip压缩包&#xff0c;压缩包也超过50M&#xff0c;该怎么办&#xff1f;怎样导入超过50M数据库呢&#xff1f; …
阅读更多...
selenium爬虫

selenium爬虫

跟着大佬学习 大佬的博客获取一页数据没有问题。多页请求就会触发猫眼的验证中心&#xff0c;没有数据&#xff0c;所以优化加强了一下&#xff0c;一起进步学习。 ip代理网站&#xff1a;https://www.zdaye.com/free/4/ip&#xff08;用了selenium这一步骤ip代理可以忽略&…
阅读更多...
清理 zabbix 历史数据, 缩减 postgresql 空间

清理 zabbix 历史数据, 缩减 postgresql 空间

在 Zabbix 中使用 PostgreSQL 作为数据库后端时&#xff0c;随着监控数据的不断积累&#xff0c;数据库可能会变得非常大&#xff0c;从而导致存储空间紧张。为了清理 Zabbix 历史数据并缩减 PostgreSQL 空间&#xff0c;您可以按照以下步骤进行操作&#xff1a; 一、分析数据…
阅读更多...
k8s-Informer概要解析(2)

k8s-Informer概要解析(2)

Client-go 主要用在 k8s 控制器中 什么是 k8s Informer Informer 负责与 kubernetes APIServer 进行 Watch 操作&#xff0c;Watch 的资源&#xff0c;可以是 kubernetes 内置资源对象&#xff0c;也可以 CRD。 Informer 是一个带有本地缓存以及索引机制的核心工具包&#x…
阅读更多...
使用 Temporal 管理和调度 Couchbase SQL 脚本的实际例子

使用 Temporal 管理和调度 Couchbase SQL 脚本的实际例子

场景描述 目标&#xff1a;使用 Temporal 管理和调度一组 Couchbase SQL 脚本来完成以下任务&#xff1a; 同步数据&#xff1a;从其他数据源同步数据到 Couchbase。 执行数据聚合&#xff1a;运行统计 SQL 查询。 清理过期数据&#xff1a;定期清理 Couchbase 中过期或无效…
阅读更多...
原生html+css+ajax实现二级下拉选择的增删改及树形结构列出

原生html+css+ajax实现二级下拉选择的增删改及树形结构列出

<?php $db_host localhost; $db_user info_chalide; $db_pass j8c2rRr2RnA; $db_name info_chalide; /* 数据库结构SQL CREATE TABLE categories ( id INT AUTO_INCREMENT PRIMARY KEY, name VARCHAR(255) NOT NULL, parent_id INT DEFAULT 0 ); */ try { $pdo new PD…
阅读更多...
[笔记] 持续一个月的电路调试工作

[笔记] 持续一个月的电路调试工作

11月下旬&#xff0c;入职一家负责系统集成&#xff0c;但是触手已经触及大量的电力电子相关的模块设计的公司。入职时谈及的工作方向有三个&#xff0c;一个是一款手头正在做的开关电源的调优&#xff1b;一个是一款新型PWM移相开关电源的工作&#xff1b;最后一项是我非常期望…
阅读更多...
最新文章

Copyright @ 2022~2023