Esempio n. 1
0
int nd_net_ioctl(nd_netui_handle  socket_node, int cmd, void *val, int *size) 
{
	int ret = -1 ;
	NDUINT32 inputVal ;
	switch(cmd)
	{
	case NDIOCTL_SET_SENDVBUF:
		if (*size <4){
			return -1 ;
		}
		if (socket_node->type ==NDHANDLE_TCPNODE|| socket_node->type ==NDHANDLE_UDPNODE ){
			if (!socket_node->is_session){
				inputVal = *(NDUINT32*)val ;
				ret = ndlbuf_realloc (&socket_node->send_buffer,inputVal) ;
			}
		}
		break ;
	case NDIOCTL_GET_SENDBUF:
		if (*size <4){
			return -1 ;
		}
		if (socket_node->type ==NDHANDLE_TCPNODE|| socket_node->type ==NDHANDLE_UDPNODE ){
			if (!socket_node->is_session){
				*(NDUINT32*)val = (NDUINT32)ndlbuf_capacity(&socket_node->send_buffer) ;
				*size = sizeof(NDUINT32) ;
				ret = 0;
			}
		}
		break ;
	case NDIOCTL_SET_RECVVBUF:
		if (*size <4){
			return -1 ;
		}
		if (socket_node->type ==NDHANDLE_TCPNODE|| socket_node->type ==NDHANDLE_UDPNODE ){
			if (!socket_node->is_session){
				inputVal = *(NDUINT32*)val ;
				ret = ndlbuf_realloc (&socket_node->recv_buffer,inputVal) ;
			}
		}
		break ;

	case NDIOCTL_GET_RECVBUF:
		if (*size <4){
			return -1 ;
		}
		if (socket_node->type ==NDHANDLE_TCPNODE|| socket_node->type ==NDHANDLE_UDPNODE ){
			if (!socket_node->is_session){
				*(NDUINT32*)val = (NDUINT32)ndlbuf_capacity(&socket_node->recv_buffer) ;
				*size = sizeof(NDUINT32) ;
				ret = 0;
			}
		}
		break ;
	case NDIOCTL_SET_TCP_RECV_WNDSIZE:
		if (socket_node->type ==NDHANDLE_TCPNODE|| socket_node->type ==NDHANDLE_UDPNODE ){
			if (socket_node->fd){
				ret = setsockopt(socket_node->fd, SOL_SOCKET, SO_RCVBUF, (sock_opval_t)val, *size) ;
			}		
		}
		break ;
	case NDIOCTL_GET_TCP_RECV_WNDSIZE:
		if (socket_node->type ==NDHANDLE_TCPNODE|| socket_node->type ==NDHANDLE_UDPNODE ){
			if (socket_node->fd){
				inputVal = *(NDUINT32*)val ;
				ret = getsockopt(socket_node->fd, SOL_SOCKET, SO_RCVBUF, (sock_opval_t)val, size) ;
			}		
		}
		break ;
	case NDIOCTL_SET_TCP_SEND_WNDSIZE:
		if (socket_node->type ==NDHANDLE_TCPNODE|| socket_node->type ==NDHANDLE_UDPNODE ){
			if (socket_node->fd){
				inputVal = *(NDUINT32*)val ;
				ret = setsockopt(socket_node->fd, SOL_SOCKET, SO_SNDBUF,(sock_opval_t)val, *size) ;
			}		
		}
		break ;
	case NDIOCTL_GET_TCP_SEND_WNDSIZE:
		if (socket_node->type ==NDHANDLE_TCPNODE|| socket_node->type ==NDHANDLE_UDPNODE ){
			if (socket_node->fd){
				inputVal = *(NDUINT32*)val ;
				ret = getsockopt(socket_node->fd, SOL_SOCKET, SO_SNDBUF, (sock_opval_t)val, size) ;
			}		
		}
		break ;
	case NDIOCTL_SET_TIMEOUT:
		socket_node->disconn_timeout = (*(ndtime_t*)val) *1000 ;
		ret = 0;
		break;
	case NDIOCTL_GET_TIMEOUT:
		*(ndtime_t*)val  = socket_node->disconn_timeout /1000 ;
		*size = sizeof(ndtime_t) ;
		ret = 0;
		break ;
    case NDIOCTL_GET_LAST_RECV_TIME:
		*(ndtime_t*)val  = socket_node->last_recv ;
		*size = sizeof(ndtime_t) ;
		ret = 0;
        break ;
        
    case NDIOCTL_GET_LAST_SEND_TIME:
		*(ndtime_t*)val  = socket_node->last_push ;
		*size = sizeof(ndtime_t) ;
		ret = 0;
        break ;
        
    case NDIOCTL_GET_USERDATA:
		*(nd_userdata_t*)val  = socket_node->user_data ;
		ret = 0;
        break ;
    case NDIOCTL_SET_USERDATA:
		socket_node->user_data = val;
		ret = 0;
        break ;
        
        ///
        
    case NDIOCTL_GET_PEER_IP:
		*(ndip_t*) val = nd_net_peer_getip(socket_node);
		ret = 0;
        break;
        
    case NDIOCTL_GET_PEER_PORT:
		*(ndport_t*) val = nd_net_peer_getport(socket_node);
		ret = 0;
        break;

        
    case NDIOCTL_HOOK_DATA:
		nd_hook_data(socket_node,(data_in_entry) val ) ;
		ret = 0;
        break ;
        
    case NDIOCTL_HOOK_PACKET:
		nd_hook_packet(socket_node,(net_msg_entry) val ) ;
		ret = 0;
        break ;
        
    case NDIOCTL_SET_UNREG_MSG_CLOSED: //close peers when received unregister message
		nd_net_set_unregmsg_handler(socket_node, *(int*)val);
		ret = 0;

        break ;
    case NDIOCTL_SET_UNAUTHORIZE_CLOSE:
		nd_net_set_unauthorize_handler(socket_node, *(int*)val);
		ret = 0;
        break ;
        

    case NDIOCTL_SET_CRYPT_KEY:
		nd_connector_set_crypt(socket_node,val, *size);
		ret = 0;
        break ;
    case NDIOCTL_GET_CRYPT_KEY:
		{
		int keysize = 0 ;
		void *addr = nd_connector_get_crypt(socket_node, &keysize) ;
		ret = -1 ;
		if (keysize <= *size) {
			memcpy(val, addr, keysize) ;
			*size = keysize ;
			ret = 0 ;
		}
		}
        break;
            
    case NDIOCTL_SET_LEVEL:
		socket_node->level = (NDUINT8) (*(int*)val ) ;
		ret = 0;
        break;
            
    case NDIOCTL_GET_LEVEL:
		*(int*)val = socket_node->level ;
		ret = 0;
        break;

	case NDIOCTL_LOG_SEND_MSG:
		if (*(int*)val){
			socket_node->is_log_send = 1;
		}
		else {
			socket_node->is_log_send = 0;
		}
		ret = 0;
		break;

	case NDIOCTL_LOG_RECV_MSG:
		if (*(int*)val){
			socket_node->is_log_recv = 1;
		}
		else {
			socket_node->is_log_recv = 0;
		}
		ret = 0;
		break;

	case NDIOCTL_SYS_SET_LOG_FUNC:
		nd_setlog_func((logfunc)val);
		break;

	case NDIOCTL_SYS_GET_LOG_FUNC:
	{
		logfunc oldval = nd_setlog_func((logfunc)0);
		logfunc *retval = (logfunc *) val;
		*retval = oldval;
		nd_setlog_func(oldval);
	}
		break;

	case NDIOCTL_GET_SESSIONID:
		*(int*)val = socket_node->session_id ;
		ret = 0;
		break;
	case NDIOCTL_SET_SESSIONID:
		socket_node->session_id = (NDUINT16) (*(int*)val);
		ret = 0;
		break;
		
	case NDIOCTL_GET_WRITABLE_CALLBACK:
		*(net_writable_callback*)val = socket_node->writable_callback ;
		ret = 0 ;
		break ;
	case NDIOCTL_SET_WRITABLE_CALLBACK:
		socket_node->writable_callback =(net_writable_callback) *(nd_userdata_t*)val ;
		ret = 0 ;
		break ;
		
	case NDIOCTL_GET_WRITABLE_CALLBACK_PARAM:
		*(nd_userdata_t*)val = socket_node->writable_param ;
		ret = 0 ;
		break ;
	case NDIOCTL_SET_WRITABLE_CALLBACK_PARAM:
		socket_node->writable_param = *(nd_userdata_t*)val ;
		ret = 0 ;
		break ;
		

	case NDIOCTL_SET_BLOCK:
		ret = nd_socket_nonblock(socket_node->fd, 0) ;
		break ;
	case NDIOCTL_GET_BLOCK:
		ret = nd_socket_nonblock(socket_node->fd, 1) ;
		break ;

	case NDIOCTL_GET_RECV_PACK_NUM:
		if (socket_node->type ==NDHANDLE_TCPNODE|| socket_node->type ==NDHANDLE_UDPNODE ){
			*(NDUINT32*)val = (NDUINT32) nd_atomic_read(&socket_node->recv_pack_times) ;
			*size = sizeof(NDUINT32) ;
			ret = 0;
		}
		break ;
	case NDIOCTL_SET_RECV_PACK_NUM:
		if (socket_node->type ==NDHANDLE_TCPNODE|| socket_node->type ==NDHANDLE_UDPNODE ){
			inputVal = *(NDUINT32*)val ;
			nd_atomic_set(&socket_node->recv_pack_times, inputVal );
			ret = 0;
		}
		break ;
	case NDIOCTL_GET_SEND_PACK_NUM:
		if (socket_node->type ==NDHANDLE_TCPNODE|| socket_node->type ==NDHANDLE_UDPNODE ){
			*(NDUINT32*)val = (NDUINT32) nd_atomic_read(&socket_node->send_pack_times );
			*size = sizeof(NDUINT32) ;
			ret = 0;
		}
		break ;
	case NDIOCTL_SET_SEND_PACK_NUM:
		if (socket_node->type ==NDHANDLE_TCPNODE|| socket_node->type ==NDHANDLE_UDPNODE ){
			inputVal = *(NDUINT32*)val ;
			nd_atomic_set(&socket_node->send_pack_times, inputVal );
			ret = 0;
		}
		break ;
			
			
			
	case NDIOCTL_LOG_SEND_STRAM_FILE:
			ret = -1 ;
			if (val) {
				socket_node->save_send_stream = 1 ;
				if (_s_send_stream) {
					free(_s_send_stream) ;
				}
				_s_send_stream = (char*) malloc(*size + 1) ;
				if (_s_send_stream) {
					strncpy(_s_send_stream, (const char*)val, *size +1) ;
					ret =  0 ;
				}
			}
			else {
				socket_node->save_send_stream = 0;
				if (_s_send_stream) {
					free(_s_send_stream) ;
				}
				ret = 0 ;
			}
		break ;
	case NDIOCTL_LOG_RECV_STRAM_FILE:
			ret = -1 ;
			if (val) {
				socket_node->save_recv_stream = 1 ;
				if (_s_recv_stream) {
					free(_s_recv_stream) ;
				}
				_s_recv_stream = (char*) malloc(*size + 1) ;
				if (_s_recv_stream) {
					strncpy(_s_recv_stream, (const char*)val, *size +1) ;
					ret =  0 ;
				}
			}
			else {
				socket_node->save_recv_stream = 0;
				if (_s_recv_stream) {
					free(_s_recv_stream) ;
				}
				ret = 0 ;
			}
		break ;
	default :
		ret =-1 ;
		break ;
	}
	return ret ;
}
Esempio n. 2
0
int NDInstanceBase::Create(int argc,const char *argv[])
{
	int i ;
	const char *logfileName = NULL ;
	if (m_bCreated)	{
		return 0;
	}

    system_signals_init() ;
    
	nd_arg(argc, argv);	
	
	//get config file 	
	for (i=1; i<argc; i++){
		if(0 == strcmp(argv[i],"-f" ) && i< argc-1) {
			NDInstanceBase::config_file = argv[++i] ;
		}
		else if(0== strcmp(argv[i], "-c") && i< argc-1) {
			m_config_name = argv[++i] ;
		}
		else if(0==strcmp(argv[i],"-nodev")) {
			m_un_develop = 1;
		}
		
		else if(0==strcmp(argv[i],"-log") && i<argc -1) {
			logfileName = argv[++i];
		}
		else if(0==strcmp(argv[i],"-workdir") &&i<argc -1 ) {
			nd_chdir(argv[++i]) ;
		}
		else if(0==strcmp(argv[i],"-v") || 0==strcmp(argv[i],"--version") ) {
			fprintf(stdout, "version : %s \n" , __g_version_desc) ;
			exit(0) ;
		}
		else if(0==strcmp(argv[i],"-l") || 0==strcmp(argv[i],"--rlimit") ) {
			char buf[1024] ;
			fprintf(stdout, "rlimit: \n %s\n" , get_rlimit_info(buf, sizeof(buf)) ) ;
			exit(0) ;
		}

        else if(0==strcmp(argv[i],"-pid") && i<argc -1){
            FILE *pf = fopen(argv[i+1], "w") ;
            if (pf) {
                fprintf(pf, "%d", nd_processid()) ;
                fclose(pf) ;
            }
            else {
                fprintf(stderr,"write pid error %s file not exist\n", argv[i+1]) ;
                exit(1) ;
            }
        }
	}

	if(!config_file|| !m_config_name) {
		printf("usage: -f config-file -c config-name\n press ANY key to continue\n") ;
		//getch() ;
		exit(1) ;
		//return -1 ;
	}

	if(-1==ReadConfig(m_config_name) ) {
		printf("Read config %s file error \n press ANY key to continue\n", m_config_name) ;
		//getch() ;
		exit(1) ;
	}
	
	nd_set_exit_callback(applib_exit_callback) ;

	if (m_config.i_cfg.open_dump){
#ifndef ND_UNIX
		::SetUnhandledExceptionFilter((LPTOP_LEVEL_EXCEPTION_FILTER)nd_unhandler_except);
#else
		enable_core_dump() ;
#endif
	}
	size_t max_connector = m_config.l_cfg.max_connect ;
	if ( max_connector > get_maxopen_fd() ) {
		size_t new_fd_num = set_maxopen_fd(max_connector) ;
		nd_logmsg("set max connect %d real max connect =%d\n",m_config.l_cfg.max_connect, (int) new_fd_num);
	}
	
	if(logfileName && logfileName[0]) {
		strncpy(m_config.i_cfg.log_file, logfileName, sizeof(m_config.i_cfg.log_file)) ;
	}
	if (m_config.i_cfg.log_file[0]){
		set_log_file(m_config.i_cfg.log_file) ;
		if (m_config.i_cfg.log_file_size > 0) {
			nd_setlog_maxsize(m_config.i_cfg.log_file_size) ;
		}
		nd_setlog_func(nd_default_filelog) ;
	}

	if (m_config.i_cfg.callstack_file[0]){
		if(CALLSTACK_INIT(m_config.i_cfg.callstack_file)==-1) {
			nd_logfatal("create map file %s error!\n" AND m_config.i_cfg.callstack_file) ;
			return -1 ;
		}
	}

	nd_net_set_crypt((nd_netcrypt)nd_TEAencrypt, (nd_netcrypt)nd_TEAdecrypt, sizeof(tea_v)) ;
	
// 	if (g_base_inst==NULL){
// 		g_base_inst = this ;
// 	}

	OnCreate();

	m_bCreated = 1;
	return 0 ;

}
Esempio n. 3
0
BOOL CapolloParseEditorApp::InitInstance()
{
	// 如果一个运行在 Windows XP 上的应用程序清单指定要
	// 使用 ComCtl32.dll 版本 6 或更高版本来启用可视化方式,
	//则需要 InitCommonControlsEx()。  否则,将无法创建窗口。
	INITCOMMONCONTROLSEX InitCtrls;
	InitCtrls.dwSize = sizeof(InitCtrls);
	// 将它设置为包括所有要在应用程序中使用的
	// 公共控件类。
	InitCtrls.dwICC = ICC_WIN95_CLASSES;
	InitCommonControlsEx(&InitCtrls);

	CWinApp::InitInstance();

	nd_common_init();
	nd_setxml_log(_xml_error_output);
	nd_log_no_file(1);
	nd_log_no_date(1);
	ndstr_set_code(E_SRC_CODE_GBK);
	// 创建 shell 管理器,以防对话框包含
	// 任何 shell 树视图控件或 shell 列表视图控件。
	CShellManager *pShellManager = new CShellManager;

	// 激活“Windows Native”视觉管理器,以便在 MFC 控件中启用主题
	CMFCVisualManager::SetDefaultManager(RUNTIME_CLASS(CMFCVisualManagerWindows));

	// 标准初始化
	// 如果未使用这些功能并希望减小
	// 最终可执行文件的大小,则应移除下列
	// 不需要的特定初始化例程
	// 更改用于存储设置的注册表项
	// TODO:  应适当修改该字符串,
	// 例如修改为公司或组织名
	SetRegistryKey(_T("apolloEditor"));

	CapolloParseEditorDlg dlg;
	m_pMainWnd = &dlg;

	nd_setlog_func(out_log);
	nd_register_error_convert(apollo_error);

	INT_PTR nResponse = dlg.DoModal();
	if (nResponse == IDOK)
	{
		// TODO:  在此放置处理何时用
		//  “确定”来关闭对话框的代码
	}
	else if (nResponse == IDCANCEL)
	{
		// TODO:  在此放置处理何时用
		//  “取消”来关闭对话框的代码
	}
	else if (nResponse == -1)
	{
		TRACE(traceAppMsg, 0, "警告: 对话框创建失败,应用程序将意外终止。\n");
		TRACE(traceAppMsg, 0, "警告: 如果您在对话框上使用 MFC 控件,则无法 #define _AFX_NO_MFC_CONTROLS_IN_DIALOGS。\n");
	}

	// 删除上面创建的 shell 管理器。
	if (pShellManager != NULL)
	{
		delete pShellManager;
	}

	// 由于对话框已关闭,所以将返回 FALSE 以便退出应用程序,
	//  而不是启动应用程序的消息泵。

	nd_common_release();
	return FALSE;
}