457 lines
12 KiB
C
457 lines
12 KiB
C
/**
|
||
* @file: $RCSfile: main.c,v $
|
||
* @brief: $IEC 61850 Protocol
|
||
*
|
||
* @version: $Revision: 1.9 $
|
||
* @date: $Date: 2020/10/28 05:21:18 $
|
||
* @author: $Author: lizhongming $
|
||
* @state: $State: Exp $
|
||
*
|
||
* @latest: $Id: main.c,v 1.9 2020/10/28 05:21:18 lizhongming Exp $
|
||
*
|
||
*/
|
||
|
||
#include "rdb_client.h"
|
||
#include "db_interface.h"
|
||
#include "node.h"
|
||
|
||
/*lnk10-10 */
|
||
#include "../include/rocketmq/SimpleProducer.h"
|
||
extern G_TEST_FLAG;
|
||
extern pthread_mutex_t mtx;
|
||
|
||
extern pt61850app_t *g_pt61850app;
|
||
extern node_t *g_node;
|
||
char g_my_conf_fname[256];
|
||
//extern byte_t g_Master;
|
||
char g_onlyIP[255]; //直连某个IP，仅仅为方便测试
|
||
//extern byte_t g_protect_file; //0:不召唤保护录播文件 1:召唤保护录播文件
|
||
|
||
apr_pool_t *g_root_pool;
|
||
apr_pool_t *g_rdb_pool;
|
||
apr_pool_t *g_cfg_pool;
|
||
uint16_t g_client_id = 0;
|
||
uint32_t g_node_id = 0;
|
||
uint32_t g_min_free_size = 1024;
|
||
int g_need_password = 0;
|
||
|
||
char subdir[128] = "cfg_stat_data" ;
|
||
|
||
int usage();
|
||
int parse_param(int argc, const char **argv);
|
||
|
||
#include "ver_conf.h"
|
||
|
||
#ifdef DEBUG_SISCO
|
||
SD_CONST static ST_CHAR *SD_CONST thisFileName = __FILE__;
|
||
#endif
|
||
|
||
///////////////////////////////////////////////////////
|
||
uint32_t g_dead_lock_counter = 0;
|
||
uint32_t g_thread_blocked_times = 0;
|
||
///////////////////////////////////////////////////////////////////////////////
|
||
int three_secs_enabled = 0;
|
||
int auto_register_report_enabled = 0;
|
||
int g_front_seg_index = 0;
|
||
int g_front_seg_num = 0;
|
||
int FRONT_MP_NUM = 0;//终端连接数量
|
||
int g_front_num_count = 0;//终端连接数量过低次数
|
||
|
||
|
||
///////////////////////////////////////////////////////////////////////////////
|
||
//根据配置子目录判别全局功能的投退，设定四个分功能性能均衡程序的功能运行划分设定
|
||
void init_global_function_enable()
|
||
{
|
||
if (strcmp(subdir,"cfg_stat_data")==0) { //历史稳态
|
||
g_node_id = STAT_DATA_BASE_NODE_ID;
|
||
auto_register_report_enabled = 1;
|
||
}else if (strcmp(subdir,"cfg_3s_data")==0) { //实时
|
||
g_node_id = THREE_SECS_DATA_BASE_NODE_ID;
|
||
three_secs_enabled = 1;
|
||
}else if (strcmp(subdir,"cfg_soe_comtrade")==0) { //告警和录波和暂态
|
||
g_node_id = SOE_COMTRADE_BASE_NODE_ID;
|
||
}else if (strcmp(subdir,"cfg_his_data")==0) { //不使用
|
||
g_node_id = RECALL_ALL_DATA_BASE_NODE_ID;
|
||
g_node_id = HIS_DATA_BASE_NODE_ID;
|
||
}
|
||
else if (strcmp(subdir, "cfg_newhis_data") == 0) { //不使用
|
||
g_node_id = RECALL_ALL_DATA_BASE_NODE_ID;
|
||
g_node_id = NEW_HIS_DATA_BASE_NODE_ID;
|
||
}
|
||
else if (strcmp(subdir, "cfg_recallhis_data") == 0) { //补招
|
||
g_node_id = RECALL_HIS_DATA_BASE_NODE_ID;
|
||
}
|
||
else if (strcmp(subdir, "cfg_recallall_data") == 0) { //不使用
|
||
g_node_id = RECALL_ALL_DATA_BASE_NODE_ID;
|
||
}
|
||
}
|
||
//////////////////////////////////////////////////////////////////////////////////
|
||
|
||
#ifdef _OS_UNIX_
|
||
void init_daemon(void)
|
||
{
|
||
int pid;
|
||
int i;
|
||
|
||
if( pid = fork() )
|
||
exit(0); /** 是父进程，结束父进程 */
|
||
else if( pid < 0 )
|
||
exit(1); /** fork失败，退出 */
|
||
|
||
/** 是第一子进程，后台继续执行 */
|
||
|
||
setsid(); /** 第一子进程成为新的会话组长和进程组长并与控制终端分离 */
|
||
|
||
if( pid = fork() )
|
||
exit(0); /** 是第一子进程，结束第一子进程 */
|
||
else if( pid < 0)
|
||
exit(1); /** fork失败，退出 */
|
||
|
||
chdir("/FeProject/bin/"); //multi process running at same time
|
||
umask(0); /** 重设文件创建掩码 */
|
||
|
||
return;
|
||
}
|
||
#endif
|
||
|
||
//"--subdir, set the subdir of /CloudForward/ as the working directory, \n"
|
||
int usage()
|
||
{
|
||
// fprintf(stderr,"\n\n******** IEC61850 Protocol ********\n");
|
||
fprintf(stderr,"\nUsage : pt61850netd_pqfe -d [subdir] \n");
|
||
|
||
exit(-1);
|
||
}
|
||
|
||
|
||
int prepare_entironment_2()
|
||
{
|
||
apr_status_t rv;
|
||
|
||
/* apr library need call this first. */
|
||
apr_initialize(); //APACHE初始化
|
||
atexit(apr_terminate);
|
||
|
||
|
||
/* Create node root pool */
|
||
rv = apr_pool_create(&g_root_pool, NULL); //分配根内存
|
||
if(rv != APR_SUCCESS) {
|
||
fprintf(stderr,"%s","Create node root pool failed!\n");
|
||
return (-1);
|
||
}
|
||
|
||
#ifdef SIGPIPE
|
||
/* Ignore writes to connections that have been closed at the other end. */
|
||
apr_signal(SIGPIPE, SIG_IGN);
|
||
#endif
|
||
|
||
rv = create_log_handle(g_root_pool);
|
||
if(rv != APR_SUCCESS) {
|
||
printf("init_log failed!\n");
|
||
return (-1);
|
||
}
|
||
else {
|
||
unsigned int error = 0;
|
||
unsigned int warn = 0;
|
||
unsigned int info = 0;
|
||
log_config_t *log_handle = get_log_handle();
|
||
parse_log_switch_ini(&error,&warn,&info);
|
||
log_handle->status.debug = info;//0;//1;
|
||
log_handle->status.warn = warn;//0;//1;
|
||
log_handle->status.error = error;//0;//1;
|
||
}
|
||
echo_msg("==============================================================\n");
|
||
|
||
rv = apr_pool_create(&g_rdb_pool, g_root_pool); //分配RDB内存
|
||
if(rv != APR_SUCCESS) {
|
||
echo_errg("Create system runtime pool failed!\n");
|
||
return (-1);
|
||
}
|
||
|
||
rv = apr_pool_create(&g_cfg_pool, g_root_pool); //分配配置内存
|
||
if(rv != APR_SUCCESS) {
|
||
echo_errg("Create system config pool failed!\n");
|
||
return (-1);
|
||
}
|
||
|
||
//g_fun_pool = 0;
|
||
|
||
//rv = apr_thread_mutex_create(&g_rdb_mutex, APR_THREAD_MUTEX_NESTED,g_root_pool);//创建RDB锁
|
||
//if ( rv != APR_SUCCESS) {
|
||
// return rv;
|
||
//}
|
||
/* Initialize the register table. Call these functions first! */
|
||
echo_msg1("%-60s","Initialize system register......");
|
||
//init_default_dbparser_table(); //根据各XML文件配置RDB初始内容
|
||
//load_driver_library(); //配置各端口驱动
|
||
echo_msg("OK\n\n");
|
||
|
||
initTimezoneOffset();
|
||
|
||
return APR_SUCCESS;
|
||
}
|
||
|
||
#ifdef _OS_UNIX_
|
||
#include <stdio.h>
|
||
#include <pwd.h>
|
||
#include <unistd.h>
|
||
void printf_cur_user()
|
||
{
|
||
struct passwd *pwd;
|
||
pwd = getpwuid(getuid());
|
||
echo_warn1("Current user: %s\n", pwd->pw_name);
|
||
}
|
||
#endif
|
||
|
||
int main(int argc, const char **argv)
|
||
{
|
||
|
||
// ipcclient_t *ipcclient = NULL;
|
||
// void *cookie = NULL;
|
||
uint32_t stimer = 1;
|
||
apr_status_t rv;
|
||
// int pid;
|
||
|
||
/* Prepare the system context */
|
||
rv=prepare_entironment_2();
|
||
if (rv!=APR_SUCCESS){
|
||
return rv;
|
||
}
|
||
|
||
getVersion(argc,argv);
|
||
|
||
////////////////////
|
||
//WW 测试json
|
||
//TestJson(NULL);
|
||
//WW 2023-08-31 end
|
||
///////////////////
|
||
//TestSMSPost();//WW 2023-08-28测试公司post
|
||
//TestBodyPost();
|
||
//TestToken();//电网资源业务中台
|
||
|
||
////////////////////
|
||
//WW 测试json
|
||
//TestOSS();
|
||
//TestOBS();
|
||
//WW 2023-09-01 end
|
||
///////////////////
|
||
/* Parse the command-line parameter */
|
||
rv=parse_param(argc, argv);
|
||
if (rv!=APR_SUCCESS){
|
||
return rv;
|
||
}
|
||
|
||
#ifdef _OS_UNIX_
|
||
#ifdef QT_NO_DEBUG
|
||
if (!g_need_password)
|
||
init_daemon();
|
||
#endif
|
||
#endif
|
||
|
||
init_global_function_enable();
|
||
apr_snprintf(g_my_conf_fname,sizeof(g_my_conf_fname),"../%s/etc/pt61850netd_pqfe.xml",subdir);
|
||
|
||
#if defined (DEBUG_SISCO)
|
||
init_log_cfg ();
|
||
#endif
|
||
|
||
printf("\n\n");
|
||
#ifdef _OS_UNIX_
|
||
printf_cur_user();
|
||
#endif
|
||
|
||
if (g_need_password) {
|
||
if ( process_login_verify() == 0) {
|
||
echo_warn("User login succeeded!\n");
|
||
} else {
|
||
echo_warn("User login failed!\n");
|
||
apr_sleep(apr_time_from_sec(3));
|
||
exit(-59);
|
||
}
|
||
}
|
||
|
||
rv = init_rdb();
|
||
if (rv!=APR_SUCCESS){
|
||
return rv;
|
||
}
|
||
|
||
//lnk20241024去除数据库连接
|
||
//OTLConnect();
|
||
|
||
|
||
rv = run_protocol();
|
||
if (rv!=APR_SUCCESS){
|
||
return rv;
|
||
}
|
||
|
||
|
||
while(1) {
|
||
/* sleep 1s, just like 1s timer */
|
||
apr_sleep(apr_time_from_sec(1));
|
||
/* 每30秒钟检查一次状态 */
|
||
|
||
/*添加mq测试lnk10-10 */
|
||
//rocketmq_test_rt();
|
||
//rocketmq_test_ud();
|
||
//rocketmq_test_rc();
|
||
/*lnk20241029recall接口测试*/
|
||
//curltest();
|
||
/*202411-1lnk添加文件上传测试 */
|
||
//SOEFileWeb_test();
|
||
//添加上传测试
|
||
//qvvr_test();
|
||
//apr_sleep(apr_time_from_sec(3));
|
||
//comflag_test();
|
||
//apr_sleep(apr_time_from_sec(3));
|
||
|
||
if( !(stimer++ % 60) ) {//分钟
|
||
if (g_dead_lock_counter++ >=3) {//三分钟
|
||
g_thread_blocked_times++;
|
||
g_dead_lock_counter = 0;
|
||
}
|
||
MVL_LOG_ACSE1 ("MYLOG: current g_thread_blocked_times = %u ", g_thread_blocked_times);
|
||
|
||
if (FRONT_MP_NUM <= 1) {//监测点数
|
||
g_front_num_count++;
|
||
}
|
||
else {
|
||
g_front_num_count = 0;
|
||
}
|
||
}
|
||
|
||
//work线程死了3*13分钟，退出进程
|
||
if (g_thread_blocked_times>=13) {
|
||
MVL_LOG_ACSE0 ("MYLOG: g_thread_blocked_times>=3, so exit to restart ");
|
||
apr_sleep(apr_time_from_sec(10));
|
||
exit(-1039);
|
||
}
|
||
|
||
//lnk20241211 添加测试开关
|
||
pthread_mutex_lock(&mtx);//主线程定时任务扫描读取台账加锁
|
||
if (!G_TEST_FLAG && g_front_num_count >= 30 && g_onlyIP[0] == 0 && g_node->n_clients>10) {//30分钟连接数量过低且不是单连且台账大于十个终端
|
||
MVL_LOG_ACSE0("MYLOG: g_front_num_count>=20, so exit to restart ");
|
||
apr_sleep(apr_time_from_sec(10));
|
||
exit(-1039);
|
||
}
|
||
pthread_mutex_unlock(&mtx);
|
||
|
||
}
|
||
|
||
echo_warn1("%-60s","System shutdown now......");
|
||
apr_pool_destroy(g_root_pool);
|
||
echo_msg("OK\n");
|
||
|
||
return 0;
|
||
}
|
||
|
||
|
||
int parse_param(int argc, const char **argv)
|
||
{
|
||
apr_status_t rv;
|
||
apr_getopt_t *opt;
|
||
char *opt_arg;
|
||
char ch;
|
||
|
||
char *p;
|
||
char temp[128];
|
||
//int g_front_seg_index, g_front_seg_num;
|
||
/* Set default command-line parameter */
|
||
g_node_id = 0;
|
||
//g_protect_file = 0;
|
||
|
||
echo_warn2("================= compiled@ %s %s =================\n",__DATE__ , __TIME__ );
|
||
|
||
/* Parse the command-line parameter */
|
||
if( argc > 1 ) {
|
||
rv = apr_getopt_init(&opt, g_root_pool, argc, argv);
|
||
|
||
while (apr_getopt(opt, "q:Q:a:A:d:D:p:P:f:F:r:R:T:t:s:S", &ch, &opt_arg) == APR_SUCCESS) {
|
||
switch (ch) {
|
||
case 'q':
|
||
case 'Q':
|
||
if (opt_arg[0] >= '0' && opt_arg[0] <= '9' ) {
|
||
g_min_free_size = atoi(opt_arg);
|
||
}
|
||
else
|
||
return (usage());
|
||
break;
|
||
case 'R':
|
||
case 'r':
|
||
// if (opt_arg[0] >= '0' && opt_arg[0] <= '9' )
|
||
// {
|
||
// g_client_id = atoi(opt_arg);
|
||
// if (g_client_id>2||g_client_id<0)
|
||
// {
|
||
// printf("Do not support triple or above clients \n ");
|
||
// return (usage());
|
||
// }
|
||
////g_auto_client_id = FALSE;
|
||
// }
|
||
// else
|
||
// return (usage());
|
||
break;
|
||
case 's':
|
||
case 'S':
|
||
echo_warn1(">>>>>>>>>>>>>>>>>>>>>parse -s %s ", opt_arg);
|
||
strcpy(temp, opt_arg);
|
||
|
||
p = strtok(temp, "_"); //1_5
|
||
echo_warn1("p %s ", p);
|
||
g_front_seg_index = atoi(p);
|
||
|
||
printf("g_front_seg_index:%d",g_front_seg_index);
|
||
|
||
p = strtok(NULL, "_"); //1_5
|
||
echo_warn1("p %s ", p);
|
||
g_front_seg_num = atoi(p);
|
||
|
||
printf("g_front_seg_num:%d",g_front_seg_num);
|
||
|
||
//echo_warn2("================= compiled@ %d %d =================\n", g_front_seg_index, g_front_seg_num);
|
||
//lnk20241206不需要记录subdir，参数单独使用
|
||
//echo_warn1("subdir %s ", opt_arg + 6);
|
||
//strcpy(subdir, opt_arg + 6);
|
||
|
||
break;
|
||
break;
|
||
case 'a':
|
||
case 'A':
|
||
strcpy(g_onlyIP,opt_arg);
|
||
break;
|
||
case 'd':
|
||
case 'D':
|
||
strcpy(subdir,opt_arg);
|
||
break;
|
||
case 'T':
|
||
case 't':
|
||
break;
|
||
case 'f':
|
||
case 'F':
|
||
//if (opt_arg[strlen(opt_arg)-1] == '/'
|
||
// || opt_arg[strlen(opt_arg)-1] == '\\') {
|
||
// echo_errg( "Error: Bad or invalid file name");
|
||
// return (usage());
|
||
// }
|
||
// //g_my_conf_fname = SHR_GetPrivateFileName(opt_arg,g_root_pool);
|
||
// echo_warn1("using config file %s!\n", g_my_conf_fname);
|
||
// break;
|
||
case 'P':
|
||
case 'p':
|
||
if (opt_arg[0] >= '0' && opt_arg[0] <= '9' ) {
|
||
g_need_password = atoi(opt_arg);
|
||
}
|
||
else
|
||
return (usage());
|
||
break;
|
||
case '?':
|
||
default:
|
||
return (usage());
|
||
}
|
||
}
|
||
}
|
||
|
||
return 0;
|
||
}
|
||
|
||
|