Skip to content

Commit

Permalink
Merge pull request #2 from waitspring/dev
Browse files Browse the repository at this point in the history 
tcping
  • Loading branch information
@waitspring
waitspring authored Dec 30, 2022
2 parents dd22360 + 4ea7a2c commit 76c9370
Show file tree
Hide file tree
Showing 2 changed files with 328 additions and 12 deletions.
15 changes: 3 additions & 12 deletions my/my.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -57,24 +57,15 @@ class Logger(object):

@staticmethod
def debug(message=str):
if os.getenv('LANG') in ('en_US.UTF-8', 'zh_CN.UTF-8'):
Logger.output.debug(message)
else:
Logger.output.debug(message.decode('GBK').encode('UTF-8'))
Logger.output.debug(message)

@staticmethod
def info(message=str):
if os.getenv('LANG') in ('en_US.UTF-8', 'zh_CN.UTF-8'):
Logger.output.info(message)
else:
Logger.output.info(message.decode('GBK').encode('UTF-8'))
Logger.output.info(message)

@staticmethod
def error(message=str):
if os.getenv('LANG') in ('en_US.UTF-8', 'zh_CN.UTF-8'):
Logger.output.error(message)
else:
Logger.output.error(message.decode('GBK').encode('UTF-8'))
Logger.output.error(message)

class RaisePoint(Exception):
"""
Expand Down
325 changes: 325 additions & 0 deletions tcping/tcping.c
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,325 @@
/**********************************************************************************************************************
* *
* main.c *
* Fu Xuanming (2022-2023) *
* *
**********************************************************************************************************************
*
*
*/

#include <arpa/inet.h>
#include <errno.h>
#include <getopt.h>
#include <netdb.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <time.h>
#include <unistd.h>



/**********************************************************************************************************************
* Function *
**********************************************************************************************************************
*/
void usage(void) {
puts("Usage: tcping [option...] socket");
puts(" -c, --count=TIMES 配置命令发起 TIMES 次 TCP 连接, 默认值为 -1 (保持长 TCPING)");
puts(" -i, --interval=NUM 配置两次连接中的停顿时间为 NUM 秒钟, 默认值为 1 秒钟");
puts(" -q, --quiet 开启静默执行模式, 命令屏蔽正常输出");
puts(" -h, --help 列出命令的帮助信息");
puts(" -v, --version 列出命令的版本信息");
puts("~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~");
puts("Examp: tcping -c 1 127.0.0.1:22");
puts(" tcping -c 1 [::1]:22");
}

void version(void) {
puts("");
puts("command version: alpha 1.0.0");
puts("created by FuXuanming (waitspring)");
puts("");
}

void info(const char *func, const char *info) {
char timestamp[64];
time_t timepoint = time(NULL);
struct tm *timeinfo = localtime(&timepoint);
struct timeval tv; // 使用 sys/time.h/timeval 数据结构获取毫秒级时间戳

gettimeofday(&tv, NULL);
strftime(timestamp, sizeof timestamp, "%F %T", timeinfo);
fprintf(stdout, "%s.%03d INFO [%s] %s\n", timestamp, tv.tv_usec/1000, func, info);
}

void warn(const char *func, const char *warn) {
char timestamp[64];
time_t timepoint = time(NULL);
struct tm *timewarn = localtime(&timepoint);
struct timeval tv;

gettimeofday(&tv, NULL);
strftime(timestamp, sizeof timestamp, "%F %T", timewarn);
fprintf(stdout, "%s.%03d WARN [%s] %s\n", timestamp, tv.tv_usec/1000, func, warn);
}

void error(const char *func, const char *error) {
char timestamp[64];
time_t timepoint = time(NULL);
struct tm *timerror = localtime(&timepoint);
struct timeval tv;

gettimeofday(&tv, NULL);
strftime(timestamp, sizeof timestamp, "%F %T", timerror);
fprintf(stderr, "%s.%03d ERROR [%s] %s\n", timestamp, tv.tv_usec/1000, func, error);
}



/**********************************************************************************************************************
* Function *
**********************************************************************************************************************
*/
int judge_ip(char *ip, char **ipp, struct sockaddr_in *addr) {
/* Todo: 分切套接字得到 IPv4/IPv6 地址, 验证 IPv4/IPv6 地址是否符合书写规范
* Todo: 已经验证通过的 IPv4/IPv6 地址, 赋予 sockaddr_in.sin_addr.s_addr 数据字段
* Todo: 已经验证通过的 IPv4/IPv6 地址, 赋予 ipp 字符串指针
* Return: IPv4/IPv6 地址校验成功返回值为 1, IPv4/IPv6 地址校验失败返回值为 0
*/
struct in_addr ipv4;
struct in6_addr ipv6;
const char tag = ':';
const char sign[2] = {'[', ']'};

ip[strlen(ip) - strlen(strrchr(ip, tag))] = '\0';
if (strrchr(ip, sign[0])) {
ip = &strrchr(ip, sign[0])[1];
}
if (strrchr(ip, sign[1])) {
ip[strlen(ip) - strlen(strrchr(ip, sign[1]))] = '\0';
}
if (inet_pton(AF_INET, (char *)ip, &ipv4)) {
addr->sin_family = AF_INET;
addr->sin_addr.s_addr = inet_addr(ip);
*ipp = ip;
return 1;
}
if (inet_pton(AF_INET6, (char *)ip, &ipv6)) {
addr->sin_family = AF_INET6;
addr->sin_addr.s_addr = inet_addr(ip);
*ipp = ip;
return 1;
}

error("judge_ip", "目标地址检查失败, 目标地址不符合 IPv4/ IPv6 的书写规范");
return 0;
}

int judge_po(char *po, char **pop, struct sockaddr_in *addr) {
/* Todo: 分切套接字得到端口号, 验证端口号是否符合书写规范
* Todo: 已经验证通过的端口号, 赋予 sockaddr_in.sin_port 数据字段
* Todo: 已经验证通过的端口号, 赋予 pop 字符串指针
* Return: 端口号校验成功返回值为 1, 端口号校验失败返回值为 0
*/
const char tag = ':';

po = &strrchr(po, tag)[1];
for (int i = 0; i < strlen(po); i++) {
if (po[i] >= '0' && po[i] <='9') {
continue;
} else {
error("judge_po", "目标端口检查失败, 目标端口不符合书写规范");
return 0;
}
}
if (atoi(po) > 65535) {
error("judge_po", "目标端口检查失败, 目标端口已超过取值范围");
return 0;
}

addr->sin_port = htons(atoi(po));
*pop = po;
return 1;
}

int connect_to(struct timeval *begin, struct timeval *end, struct sockaddr_in *addr) {
/* Todo: 创建套接字, 并发起 TCP 连接 (默认 10 秒钟超时)
* Todo: 针对上述过程计时, 返回计时结果, 若连接失败则计时结果为 0
* Notice: 套接字的非阻塞工作模式无法适应内网的 TCP 连接
*/
int sock;
int connect_result;
int ms;
int timeout = 10;

gettimeofday(begin, NULL);
sock = socket(addr->sin_family, SOCK_STREAM, 0);
setsockopt(sock, SOL_SOCKET, SO_SNDTIMEO, (char *)&timeout, sizeof(int));
setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, (char *)&timeout, sizeof(int));
connect_result = connect(sock, (struct sockaddr *)addr, sizeof(*addr));
close(sock);
gettimeofday(end, NULL);
if (connect_result == 0) {
ms = (1000000 * (end->tv_sec - begin->tv_sec) + (end->tv_usec - begin->tv_usec)) / 1000;
if (ms == 0) {
/* Todo: 建立本机的 TCP 连接时, 因数据精度不够, 可能导致计时为 0 毫秒, 人工将其重置为 1 毫秒
*/
ms = 1;
}
return ms;
} else {
return 0;
}
}

int max(int *num, int size) {
int max = num[0];

for (int i = 0; i < size; i++) {
if (max < num[i] && num[i] != 0) {
max = num[i];
}
}

return max;
}

int min(int *num, int size) {
int min = num[0];

for (int i = 0; i < size; i++) {
if (min > num[i] && num[i] != 0) {
min = num[i];
}
}

return min;
}

int avg(int *num, int size, int counts) {
int total = 0;

for (int i = 0; i < size; i++) {
if (num[i] != 0) {
total += num[i];
}
}

if (counts != 0) {
return total / counts;
} else {
return 0;
}
}



/**********************************************************************************************************************
* Main Part *
**********************************************************************************************************************
*/
int main(int argc, char *argv[]) {
int opt;
int count = 86400; // count = 86400 设置连接次数, 默认为 86400 次连接 (24小时不间断)
int interval = 1; // interval = 1 设置两次连接中的停顿时间为 1 秒钟
int quiet = 0; // quiet = 0 关闭静默执行模式, quiet = 1 开启静默执行模式

char *parameter;
char ip[256];
char po[256];
char *ipp; // 指向清洗过的 IPv4/IPv6 地址
char *pop; // 指向清洗过的端口号

struct sockaddr_in addr;
memset(&addr, 0, sizeof(addr)); // 使用数值 0 填充数据结构 sockaddr_in 的每个字节

while ((opt = getopt(argc, argv, "c:i:qhv")) != -1) {
switch (opt) {
case 'c':
count = atoi(optarg);
break;
case 'i':
interval = atoi(optarg);
break;
case 'q':
quiet = 1;
break;
case 'h':
usage();
return 0;
case 'v':
version();
return 0;
case '?': // getopt 函数提供给用户捕获无效的命令选项
error("main", "选项类型检查失败, 请使用 $ tcping -h 查阅命令的帮助信息");
return 1;
}
}

if ((argc - optind) != 1) {
error("main", "参数数量检查失败, 请使用 $ tcping -h 查阅命令的帮助信息");
return 1;
} else {
parameter = argv[optind];
strcpy(ip, parameter);
strcpy(po, parameter);
}

if (judge_ip(ip, &ipp, &addr) == 0 || judge_po(po, &pop, &addr) == 0) {
/* judge_ip() 执行结束时, 字符数组变量 ip 仍然存储完整的目标套接字 (:port 部分被切割)
* judge_po() 执行结束时, 字符数组变量 po 仍然存储完整的目标套接字
*/
error("main", "参数赋值检查失败, 请使用 $ tcping -h 查阅命令的帮助信息");
return 1;
}

int ms[count];
int max_ms = 0;
int min_ms = 0;
int avg_ms = 0;
int counts = 0; // 统计成功次数
int counte = 0; // 统计失败次数

for (int i = 0; i < count; i++) {
struct timeval begin;
struct timeval end;

ms[i] = connect_to(&begin, &end, &addr);
if (ms[i] == 0) {
counte++;
if (quiet == 0) {
printf("Ping tcp://%s:%s - error\n", ipp, pop);
}
} else {
counts++;
if (quiet == 0) {
printf("Ping tcp://%s:%s - success: timepay %d(ms)\n", ipp, pop, ms[i]);
}
}
sleep(interval);
}

max_ms = max(ms, count);
min_ms = min(ms, count);
avg_ms = avg(ms, count, counts);
if (quiet == 0) {
printf("Tcping statistics tcp://%s:%s\n", ipp, pop);
printf(" %d times request success, %d times request error\n", counts, counte);
printf("Approximate trip times:\n");
printf(" Maximum = %d(ms), Minimum = %d(ms), Average = %d(ms)\n", max_ms, min_ms, avg_ms);
}

if (counte == count) {
return 2;
}
if (counte != 0) {
return 1;
}
return 0;
}


0 comments on commit 76c9370

Please sign in to comment.