我有一些看起来像这样的代码,我不确定如何处理永远不会执行的部分,因为这段代码的一部分在等待连接时以无限循环运行,当我终止程序时,它退出只有。
main(){
// do some stuff....
while(1) {
int newFD =
accept(sockFD, (struct sockaddr *)&client_addr, &client_addr_size);
if(newFD == -1) {
std::cerr << "Error while Accepting on socket" << std::endl;
continue;
}
if(!fork()) {
close(sockFD); // close child's sockfd - not needed here
// lalala do stuff send message here
close(newFD); // finally close its newFD - message sent, no use
return 0;
}
close(newFD); // close parent's newFD - no use here
}
// now execution never reaches here
close(sockFD); // so how to handle this?
freeaddrinfo(res); // and this?
return 0;
}
答案 0 :(得分:15)
如果您的代码要由其他人使用,或者您自己只是希望它更清洁,您可以,并且可能应该添加退出处理程序。在退出处理程序中,您可以切换使while()
循环终止的标志。以下代码对于此用例将100%正常工作,并且可靠且跨平台,但如果您想要执行更复杂的操作,则应使用正确的线程安全操作系统特定功能或类似Boost或C ++ 11
首先声明两个全局变量,使它们变为volatile,这样编译器总会强制我们读取或写入它的实际内存值。如果我们没有声明它是volatile,那么编译器可能会将其值放在一个寄存器中,这会使它不起作用。使用volatile set,它将读取每个循环上的内存位置并正常工作,即使有多个线程。
volatile bool bRunning=true;
volatile bool bFinished=false;
而不是您的while(1) {}
循环,请将其更改为此
while(bRunning)
{
dostuff
}
bFinished=true;
在退出处理程序中,只需设置bRunning=false;
void ExitHandler()
{
bRunning=false;
while(bFinished==false) { Sleep(1); }
}
您没有指定操作系统,但看起来您是基于Linux的,要在Linux上设置您需要的处理程序。
void ExitHandler(int s)
{
bRunning=false;
}
int main()
{
struct sigaction sigIntHandler;
sigIntHandler.sa_handler = ExitHandler;
sigemptyset(&sigIntHandler.sa_mask);
sigIntHandler.sa_flags = 0;
sigaction(SIGINT, &sigIntHandler, NULL);
while(bRunning)
{
dostuff
}
...error_handling...
}
在Windows上,当您是控制台应用程序时,请执行以下操作。
BOOL WINAPI ConsoleHandler(DWORD CEvent)
{
switch (CEvent)
{
case CTRL_C_EVENT:
case CTRL_BREAK_EVENT:
case CTRL_CLOSE_EVENT:
case CTRL_LOGOFF_EVENT:
case CTRL_SHUTDOWN_EVENT:
bRunning = false;
while (bFinished == false) Sleep(1);
break;
}
return TRUE;
}
int main()
{
SetConsoleCtrlHandler(ConsoleHandler, TRUE);
while(bRunning()
{
dostuff
}
...error_handling...
}
请注意,需要在此处测试并等待bFinished
。如果您不在Windows上执行此操作,则您的应用程序可能没有足够的时间来关闭,因为退出处理程序由单独的OS特定线程调用。在Linux上,这不是必需的,您需要退出处理程序以使主线程继续。
另一件需要注意的事情是默认情况下Windows只会在你终止之前关闭约5秒钟。在许多情况下这是不幸的,如果需要更多时间,您将需要更改注册表设置(错误的想法)或实现一个更好地挂钩这类事情的服务。对于你的简单情况,它会没事的。
答案 1 :(得分:10)
对于这些事情,操作系统将负责在关机时正确释放资源。但是,更一般地说,您仍然需要确保分配的资源在程序执行期间不会堆积,即使它们被操作系统自动回收,因为这样的资源泄漏仍会影响程序的行为和性能。
现在,关于手头的资源,没有理由不像C ++中的所有资源一样对待它们。接受的规则是将它们绑定到将在析构函数中释放它们的对象,另请参阅RAII习语。这样,即使在稍后阶段有人添加了break
语句,代码仍然会正常运行。