gfwleak/yzc-http-check
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

代码调整

Browse Source
This commit is contained in:
yzc
2019-12-31 11:19:08 +08:00
parent 0f5c209486
commit 2540b74324
1 changed files with 127 additions and 102 deletions
Download Patch File Download Diff File Expand all files Collapse all files

211
http_check/src/http_check.c
View File

@@ -25,20 +25,23 @@
#include "MESA/MESA_handle_logger.h" #include "MESA/MESA_handle_logger.h"
#include "MESA/MESA_htable.h" #include "MESA/MESA_htable.h"
#include "MESA/MESA_list.h" #include "MESA/MESA_list.h"
#include <pthread.h>
#define BUFSIZE 8192 #define BUFSIZE 8192
#define MAX_LOG_PATH_LEN 256 #define MAX_STR_LEN 256
#define CPU_START_POS 14 /* stat文件的有效起始行数 */ #define MAX_BUF_LEN 1024
int g_http_port; #define MAX_NAME_LEN 32
char g_http_address[MAX_LOG_PATH_LEN]=""; #define CPU_START_POS 14//stat文件的有效起始行数
//const char *http_check_conf_file = "./conf/http_check/http_check.conf"; #define HTTP_TIMEOUT 100
const char *http_check_conf_file = "/home/mesasoft/sapp/plug/business/http_check/http_check.conf"; uint16_t g_http_port;
char g_http_address[MAX_STR_LEN]="";
const char *http_check_conf_file = "./plug/business/http_check/http_check.conf";
struct route_info struct route_info
{ {
u_int dstAddr; uint32_t dstAddr;
u_int srcAddr; uint32_t srcAddr;
u_int gateWay; uint32_t gateWay;
char ifName[IF_NAMESIZE]; char ifName[IF_NAMESIZE];
}; };
@@ -51,7 +54,7 @@ struct route_info
//在字符串中寻找第N次空格出现的地方 //在字符串中寻找第N次空格出现的地方
char *get_items_by_pos(char *buff, unsigned int numb) char *get_items_by_pos(char *buff, unsigned int numb)
{ {
char *crpos; char *crpos = NULL;
int i, ttlen, count; int i, ttlen, count;
crpos = buff; crpos = buff;
@@ -60,11 +63,11 @@ char *get_items_by_pos(char *buff, unsigned int numb)
for (i = 0; i < ttlen; i++) for (i = 0; i < ttlen; i++)
{ {
if (' ' == *crpos) if (' ' == *crpos)//以空格为标记符进行识别
{ /* 以空格为标记符进行识别 */ {
count++; count++;
if (count == (numb - 1)) if (count == (numb - 1))//全部个数都找完了
{ /* 全部个数都找完了 */ {
crpos++; crpos++;
break; break;
} }
@@ -78,20 +81,24 @@ char *get_items_by_pos(char *buff, unsigned int numb)
//获取当前进程的CPU时间 //获取当前进程的CPU时间
long get_pro_cpu_time(unsigned int pid) long get_pro_cpu_time(unsigned int pid)
{ {
FILE *fd; FILE *fp = NULL;
char *vpos, buff[1024]; char *vpos, buff[MAX_BUF_LEN] = "";
long utime, stime, cutime, cstime; long utime, stime, cutime, cstime;
sprintf(buff, "/proc/%d/stat", pid); sprintf(buff, "/proc/%d/stat", pid);
fd = fopen(buff, "r"); fp = fopen(buff, "r");
assert(fd != NULL); if(fp == NULL)
assert(fgets(buff, sizeof(buff), fd) != NULL); {
printf("file open /proc/self/stat error!");
return -1;
}
fgets(buff, sizeof(buff), fp);
vpos = get_items_by_pos(buff, CPU_START_POS); vpos = get_items_by_pos(buff, CPU_START_POS);
sscanf(vpos, "%ld %ld %ld %ld", &utime, &stime, &cutime, &cstime); sscanf(vpos, "%ld %ld %ld %ld", &utime, &stime, &cutime, &cstime);
//printf("get_data2:%ld %ld %ld %ld\n", utime, stime, cutime, cstime); //printf("get_data_process:%ld %ld %ld %ld\n", utime, stime, cutime, cstime);
fclose(fd); fclose(fp);
return (utime + stime + cutime + cstime); return (utime + stime + cutime + cstime);
//return (utime + stime); //return (utime + stime);
@@ -100,17 +107,20 @@ long get_pro_cpu_time(unsigned int pid)
//获取整个系统的CPU时间 //获取整个系统的CPU时间
long get_sys_cpu_time(void) long get_sys_cpu_time(void)
{ {
FILE *fd; FILE *fp;
char name[32], buff[1024]; char name[MAX_NAME_LEN] = "", buff[MAX_BUF_LEN] = "";
long user, nice, syst, idle; long user, nice, syst, idle;
fd = fopen("/proc/stat", "r"); fp = fopen("/proc/stat", "r");
assert(fd != NULL); if(fp == NULL)
assert(fgets(buff, sizeof(buff), fd) != NULL); {
printf("file open /proc/stat error!");
return -1;
}
fgets(buff, sizeof(buff), fp);
sscanf(buff, "%s %ld %ld %ld %ld", name, &user, &nice, &syst, &idle); sscanf(buff, "%s %ld %ld %ld %ld", name, &user, &nice, &syst, &idle);
//printf("get_data1:%s %ld %ld %ld %ld\n", name, user, nice, syst, idle); //printf("get_data_system:%s %ld %ld %ld %ld\n", name, user, nice, syst, idle);
fclose(fd); fclose(fp);
return (user + nice + syst + idle); return (user + nice + syst + idle);
//return (user + syst); //return (user + syst);
@@ -119,19 +129,25 @@ long get_sys_cpu_time(void)
//获取进程的CPU使用率 //获取进程的CPU使用率
float get_cpu_stat(unsigned int pid) float get_cpu_stat(unsigned int pid)
{ {
float ratio; float ratio = 0;
long s_cur_pro_cpu, s_pre_pro_cpu; /* 指定程序的本轮/前轮CPU时间 */ long s_cur_pro_cpu, s_pre_pro_cpu;//指定程序的本轮/前轮CPU时间
long s_cur_sys_cpu, s_pre_sys_cpu; /* 整个系统的本轮/前轮CPU时间 */ long s_cur_sys_cpu, s_pre_sys_cpu;//整个系统的本轮/前轮CPU时间
s_pre_pro_cpu = get_pro_cpu_time(pid); s_pre_pro_cpu = get_pro_cpu_time(pid);
s_pre_sys_cpu = get_sys_cpu_time(); s_pre_sys_cpu = get_sys_cpu_time();
if(s_pre_pro_cpu < 0 || s_pre_sys_cpu < 0)
return 0;
sleep(1); sleep(1);
s_cur_pro_cpu = get_pro_cpu_time(pid); s_cur_pro_cpu = get_pro_cpu_time(pid);
s_cur_sys_cpu = get_sys_cpu_time(); s_cur_sys_cpu = get_sys_cpu_time();
if(s_cur_pro_cpu < 0 || s_cur_sys_cpu < 0)
return 0;
if ((s_cur_pro_cpu == s_pre_pro_cpu) || (s_cur_sys_cpu == s_pre_sys_cpu) || (s_cur_pro_cpu == 0) || (s_cur_sys_cpu == 0)) { if ((s_cur_pro_cpu == s_pre_pro_cpu) || (s_cur_sys_cpu == s_pre_sys_cpu) || (s_cur_pro_cpu == 0) || (s_cur_sys_cpu == 0)) {
ratio = 0; ratio = 0;
} else { }
else
{
if((s_cur_sys_cpu - s_pre_sys_cpu) != 0)
ratio = (100.0 * (s_cur_pro_cpu - s_pre_pro_cpu)) / (s_cur_sys_cpu - s_pre_sys_cpu); ratio = (100.0 * (s_cur_pro_cpu - s_pre_pro_cpu)) / (s_cur_sys_cpu - s_pre_sys_cpu);
} }
@@ -141,7 +157,7 @@ float get_cpu_stat(unsigned int pid)
//获取默认网关地址 //获取默认网关地址
int readNlSock(int sockFd, char *bufPtr, int seqNum, int pId) int readNlSock(int sockFd, char *bufPtr, int seqNum, int pId)
{ {
struct nlmsghdr *nlHdr; struct nlmsghdr *nlHdr = NULL;
int readLen = 0, msgLen = 0; int readLen = 0, msgLen = 0;
do{ do{
//收到内核的应答 //收到内核的应答
@@ -184,11 +200,9 @@ void parseRoutes(struct nlmsghdr *nlHdr, struct route_info *rtInfo,char *gateway
struct rtmsg *rtMsg; struct rtmsg *rtMsg;
struct rtattr *rtAttr; struct rtattr *rtAttr;
int rtLen; int rtLen;
char *tempBuf = NULL;
struct in_addr dst; struct in_addr dst;
struct in_addr gate; struct in_addr gate;
unsigned char *bytes = NULL;
tempBuf = (char *)malloc(100);
rtMsg = (struct rtmsg *)NLMSG_DATA(nlHdr); rtMsg = (struct rtmsg *)NLMSG_DATA(nlHdr);
// If the route is not for AF_INET or does not belong to main routing table // If the route is not for AF_INET or does not belong to main routing table
//then return. //then return.
@@ -217,18 +231,20 @@ void parseRoutes(struct nlmsghdr *nlHdr, struct route_info *rtInfo,char *gateway
} }
} }
dst.s_addr = rtInfo->dstAddr; dst.s_addr = rtInfo->dstAddr;
if (strstr((char *)inet_ntoa(dst), "0.0.0.0")) //if (strstr((char *)inet_ntoa(dst), "0.0.0.0"))
if (dst.s_addr == 0)
{ {
//printf("oif:%s",rtInfo->ifName); //printf("oif:%s",rtInfo->ifName);
gate.s_addr = rtInfo->gateWay; gate.s_addr = rtInfo->gateWay;
sprintf(gateway, (char *)inet_ntoa(gate)); //sprintf(gateway, (char *)inet_ntoa(gate));
bytes = (unsigned char *) &gate.s_addr;
sprintf(gateway, "%d.%d.%d.%d", bytes[0], bytes[1], bytes[2], bytes[3] );
//printf("%s\n",gateway); //printf("%s\n",gateway);
gate.s_addr = rtInfo->srcAddr; gate.s_addr = rtInfo->srcAddr;
//printf("src:%s\n",(char *)inet_ntoa(gate)); //printf("src:%s\n",(char *)inet_ntoa(gate));
gate.s_addr = rtInfo->dstAddr; gate.s_addr = rtInfo->dstAddr;
//printf("dst:%s\n",(char *)inet_ntoa(gate)); //printf("dst:%s\n",(char *)inet_ntoa(gate));
} }
free(tempBuf);
return; return;
} }
@@ -236,7 +252,6 @@ void parseRoutes(struct nlmsghdr *nlHdr, struct route_info *rtInfo,char *gateway
int get_gateway(char *gateway) int get_gateway(char *gateway)
{ {
struct nlmsghdr *nlMsg; struct nlmsghdr *nlMsg;
//struct rtmsg *rtMsg;
struct route_info *rtInfo; struct route_info *rtInfo;
char msgBuf[BUFSIZE]; char msgBuf[BUFSIZE];
int sock, len, msgSeq = 0; int sock, len, msgSeq = 0;
@@ -249,12 +264,11 @@ int get_gateway(char *gateway)
memset(msgBuf, 0, BUFSIZE); memset(msgBuf, 0, BUFSIZE);
nlMsg = (struct nlmsghdr *)msgBuf; nlMsg = (struct nlmsghdr *)msgBuf;
//rtMsg = (struct rtmsg *)NLMSG_DATA(nlMsg);
nlMsg->nlmsg_len = NLMSG_LENGTH(sizeof(struct rtmsg)); // Length of message. nlMsg->nlmsg_len = NLMSG_LENGTH(sizeof(struct rtmsg)); // Length of message.
nlMsg->nlmsg_type = RTM_GETROUTE; // Get the routes from kernel routing table . nlMsg->nlmsg_type = RTM_GETROUTE; // Get the routes from kernel routing table
nlMsg->nlmsg_flags = NLM_F_DUMP | NLM_F_REQUEST; // The message is a request for dump. nlMsg->nlmsg_flags = NLM_F_DUMP | NLM_F_REQUEST; // The message is a request for dump
nlMsg->nlmsg_seq = msgSeq++; // Sequence of the message packet. nlMsg->nlmsg_seq = msgSeq++; // Sequence of the message packet
nlMsg->nlmsg_pid = getpid(); // PID of process sending the request. nlMsg->nlmsg_pid = getpid(); // PID of process sending the request
if(send(sock, nlMsg, nlMsg->nlmsg_len, 0) < 0) if(send(sock, nlMsg, nlMsg->nlmsg_len, 0) < 0)
{ {
@@ -293,19 +307,12 @@ void http_handler_status_msg(struct evhttp_request *req,void *arg)
//启动时间,启动时长 //启动时间,启动时长
if (sysinfo(&info)) if (sysinfo(&info))
{ {
fprintf(stderr, "Failed to get sysinfo, errno:%u, reason:%s\n", printf("====line:%d,%s\n",__LINE__,"Failed to get sysinfo!");
errno, strerror(errno));
return ; return ;
} }
time(&cur_time); time(&cur_time);
if (cur_time > info.uptime)
{
boot_time = cur_time - info.uptime; boot_time = cur_time - info.uptime;
}
else
{
boot_time = info.uptime - cur_time;
}
ptm = localtime(&boot_time); ptm = localtime(&boot_time);
/*printf("System boot time:%d-%-d-%d %d:%d:%d\n", ptm->tm_year + 1900, /*printf("System boot time:%d-%-d-%d %d:%d:%d\n", ptm->tm_year + 1900,
ptm->tm_mon + 1, ptm->tm_mday, ptm->tm_hour, ptm->tm_min, ptm->tm_sec);*/ ptm->tm_mon + 1, ptm->tm_mday, ptm->tm_hour, ptm->tm_min, ptm->tm_sec);*/
@@ -321,35 +328,34 @@ void http_handler_status_msg(struct evhttp_request *req,void *arg)
//printf("process id:%d\n", process_id); //printf("process id:%d\n", process_id);
//默认网关地址 //默认网关地址
char buff[256]; char default_gateway_tmp[MAX_STR_LEN]="";
get_gateway(buff); get_gateway(default_gateway_tmp);
//printf("gateway:%s\n",buff); //printf("gateway:%s\n",buff);
//内存(虚存，实存) //内存(虚存，实存)
unsigned int n_rss, n_resident, n_share, n_text, n_lib, n_data, n_dt; unsigned int n_rss, n_resident, n_share, n_text, n_lib, n_data, n_dt;
char buf[256]; char mem_buf_tmp[MAX_STR_LEN];
char buf1[256]; char boot_time_tmp[MAX_STR_LEN];
char buf2[256]; char running_time_tmp[MAX_STR_LEN];
char buf3[256]; char cpu_usage_tmp[MAX_STR_LEN];
FILE *f = fopen("/proc/self/statm","r"); FILE *fp = fopen("/proc/self/statm","r");
if(f) if(fp)
{ {
fscanf(f,"%u%u%u%u%u%u%u",&n_rss,&n_resident,&n_share,&n_text,&n_lib,&n_data,&n_dt); fscanf(fp,"%u%u%u%u%u%u%u",&n_rss,&n_resident,&n_share,&n_text,&n_lib,&n_data,&n_dt);
fclose(f); fclose(fp);
} }
else else
{ {
printf("open /proc/self/statm failed!"); printf("====line:%d,%s\n",__LINE__,"open /proc/self/statm failed!");
return; return;
} }
n_rss = n_rss * 4; n_rss = n_rss * (getpagesize()/1024);
n_resident = n_resident *4; n_resident = n_resident * (getpagesize()/1024);
sprintf(mem_buf_tmp, "VmSize=%ludkB, VmRss=%lukB\n", n_rss, n_resident);
sprintf(buf, "VmSize=%ukB, VmRss=%ukB", n_rss, n_resident);
//printf("memory info:%s\n",buf); //printf("memory info:%s\n",buf);
//cpu //当前进程cpu占用
float cpu_rate; float cpu_rate;
unsigned int cpu_num; unsigned int cpu_num;
cpu_rate = get_cpu_stat(process_id); cpu_rate = get_cpu_stat(process_id);
@@ -363,20 +369,23 @@ void http_handler_status_msg(struct evhttp_request *req,void *arg)
printf("====line:%d,%s\n",__LINE__,"retbuff is null."); printf("====line:%d,%s\n",__LINE__,"retbuff is null.");
return; return;
} }
sprintf(buf1, "%d-%-d-%d %d:%d:%d", ptm->tm_year + 1900,ptm->tm_mon + 1, ptm->tm_mday, ptm->tm_hour, ptm->tm_min, ptm->tm_sec); sprintf(boot_time_tmp, "%d-%-d-%d %d:%d:%d", ptm->tm_year + 1900,ptm->tm_mon + 1, ptm->tm_mday, ptm->tm_hour, ptm->tm_min, ptm->tm_sec);
sprintf(buf2, "%ddays %dhours %dminutes %dseconds", run_days,run_hour,run_minute,run_second); sprintf(running_time_tmp, "%ddays %dhours %dminutes %dseconds", run_days,run_hour,run_minute,run_second);
sprintf(buf3, "%f", cpu_rate*cpu_num); sprintf(cpu_usage_tmp, "%f", cpu_rate*cpu_num);
cJSON * root = cJSON_CreateObject(); cJSON * root = cJSON_CreateObject();
if(!root)
{
printf("====line:%d,%s\n",__LINE__,"cJSON_CreateObject error!");
return;
}
//根节点下添加 //根节点下添加
cJSON_AddItemToObject(root, "boot time", cJSON_CreateString(buf1)); cJSON_AddItemToObject(root, "boot time", cJSON_CreateString(boot_time_tmp));
cJSON_AddItemToObject(root, "running time", cJSON_CreateString(buf2)); cJSON_AddItemToObject(root, "running time", cJSON_CreateString(running_time_tmp));
cJSON_AddItemToObject(root, "process id", cJSON_CreateNumber(process_id)); cJSON_AddItemToObject(root, "process id", cJSON_CreateNumber(process_id));
cJSON_AddItemToObject(root, "server ip", cJSON_CreateString(g_http_address)); cJSON_AddItemToObject(root, "server ip", cJSON_CreateString(g_http_address));
cJSON_AddItemToObject(root, "default gateway", cJSON_CreateString(buff)); cJSON_AddItemToObject(root, "default gateway", cJSON_CreateString(default_gateway_tmp));
cJSON_AddItemToObject(root, "memory info", cJSON_CreateString(buf)); cJSON_AddItemToObject(root, "memory info", cJSON_CreateString(mem_buf_tmp));
cJSON_AddItemToObject(root, "cpu usage", cJSON_CreateString(buf3)); cJSON_AddItemToObject(root, "cpu usage", cJSON_CreateString(cpu_usage_tmp));
printf("cjson_status_print:%s\n", cJSON_Print(root)); printf("cjson_status_print:%s\n", cJSON_Print(root));
//evbuffer_add_printf(retbuff,"System boot time: %d-%-d-%d %d:%d:%d\r\nsystem running time:%dday %dhour %dminute %dsecond\r\nprocess id:%d\r\ngateway:%s\r\nmemory info:%s\r\ncpu% = %f\r\n", //evbuffer_add_printf(retbuff,"System boot time: %d-%-d-%d %d:%d:%d\r\nsystem running time:%dday %dhour %dminute %dsecond\r\nprocess id:%d\r\ngateway:%s\r\nmemory info:%s\r\ncpu% = %f\r\n",
@@ -395,7 +404,7 @@ void http_handler_status_msg(struct evhttp_request *req,void *arg)
//keepalive回响应 //keepalive回响应
void http_handler_keepalive_msg(struct evhttp_request *req,void *arg) void http_handler_keepalive_msg(struct evhttp_request *req,void *arg)
{ {
struct tm *local; struct tm *local = NULL;
time_t tt; time_t tt;
tt=time(NULL); tt=time(NULL);
local=localtime(&tt); local=localtime(&tt);
@@ -409,6 +418,11 @@ void http_handler_keepalive_msg(struct evhttp_request *req,void *arg)
} }
cJSON * root = cJSON_CreateObject(); cJSON * root = cJSON_CreateObject();
if(!root)
{
printf("====line:%d,%s\n",__LINE__,"cJSON_CreateObject error!");
return;
}
cJSON_AddItemToObject(root, "current time", cJSON_CreateString(asctime(local))); cJSON_AddItemToObject(root, "current time", cJSON_CreateString(asctime(local)));
printf("cjson_keepalive_print:%s\n", cJSON_Print(root)); printf("cjson_keepalive_print:%s\n", cJSON_Print(root));
@@ -423,16 +437,15 @@ void http_handler_keepalive_msg(struct evhttp_request *req,void *arg)
return; return;
} }
//int main() void *pthread(void *arg)
int http_check_init()
{ {
struct evhttp *http_server = NULL; struct evhttp *http_server = NULL;
//short http_port = 10000; //short http_port = 10000;
//char *http_addr = "0.0.0.0"; //char *http_addr = "0.0.0.0";
MESA_load_profile_int_def(http_check_conf_file, "HTTP", "SERVER_PORT", &g_http_port, 0); MESA_load_profile_int_def(http_check_conf_file, "HTTP", "SERVER_PORT", (int *)(&g_http_port), 0);
MESA_load_profile_string_def(http_check_conf_file, "HTTP", "SERVER_IP", g_http_address, MAX_LOG_PATH_LEN, NULL); MESA_load_profile_string_def(http_check_conf_file, "HTTP", "SERVER_IP", g_http_address, MAX_STR_LEN, NULL);
printf("*v1*******read_ip:%s,read_port:%d\n",g_http_address, g_http_port); printf("*v2*******read_ip:%s,read_port:%d\n",g_http_address, g_http_port);
//初始化 //初始化
event_init(); event_init();
//启动http服务端 //启动http服务端
@@ -440,19 +453,31 @@ int http_check_init()
if(http_server == NULL) if(http_server == NULL)
{ {
printf("====line:%d,%s\n",__LINE__,"http server start failed."); printf("====line:%d,%s\n",__LINE__,"http server start failed.");
return -1; return NULL;
} }
//设置请求超时时间(s) //设置请求超时时间(s)
evhttp_set_timeout(http_server,100); evhttp_set_timeout(http_server,HTTP_TIMEOUT);
//设置事件处理函数，evhttp_set_cb针对每一个事件(请求)注册一个处理函数， //设置事件处理函数，针对每一个事件(请求)注册一个处理函数，为特定的URI指定callback
//区别于evhttp_set_gencb函数，是对所有请求设置一个统一的处理函数 evhttp_set_cb(http_server,"/status",http_handler_status_msg,NULL);
evhttp_set_cb(http_server,"/status",http_handler_status_msg,NULL);//为特定的URI指定callback
evhttp_set_cb(http_server,"/keepalive",http_handler_keepalive_msg,NULL); evhttp_set_cb(http_server,"/keepalive",http_handler_keepalive_msg,NULL);
//循环监听 //循环监听
event_dispatch(); event_dispatch();
evhttp_free(http_server); evhttp_free(http_server);
return NULL;
}
int http_check_init()
{
pthread_t tidp;
if ((pthread_create(&tidp, NULL, pthread, NULL)) == -1)
{
printf("====line:%d,%s\n",__LINE__,"http_check thread create error!");
return 0;
}
return 0; return 0;
} }