如何处理从不执行的代码

时间:2016-02-22 06:35:39

标签: c++ linux loops networking infinite-loop

我有一些看起来像这样的代码,我不确定如何处理永远不会执行的部分,因为这段代码的一部分在等待连接时以无限循环运行,当我终止程序时,它退出只有。

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;
}

2 个答案:

答案 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语句,代码仍然会正常运行。

BTW:我在这里看到的更严重的问题是缺乏正确的错误处理。