如何使用libreadline处理Ctrl-D

时间:2014-02-28 22:52:04

标签: c linux readline handle

我在C中构建一个程序,它是一个控制台,我使用libreadline来帮助我构建这个控制台。一切正常,但当我收到ctrl-d时,我会遇到分段错误。

我可以弄清楚如何处理这个,我曾尝试使用signal.h来做到这一点,但是不行。

以下是我的一段代码。

int init_console(char * ip, int port) {
/** Variaveis que serão utilizadas */
char * str_command;
char * filename = malloc(100);
Client * cli = malloc(sizeof (Client));

/** Monta o filename do arquivo de histórico */
strcpy(filename, getenv("HOME"));
strcat(filename, HISTORY_FILE);

/** Inicializa o histórico de comandos */
using_history();

/** Inicializa o readline */
rl_initialize();

/** Seta o número de comandos a serem guardados */
stifle_history(10);

/** Habilita o auto completation */
rl_inhibit_completion = true;

/** Cria o arquivo de historico */
if (access(filename, F_OK) != 0) {
    /** Se  o arquivo não existe, então cria ele */
    create_history_file(filename);
} else {
    /** Le o arquivo de histórico */
    read_history(filename);
}

/** Conecta com o server */
strcpy(cli->ip, ip);
cli->port = port;

if (connect_server(cli) == CONNECT_FAILED) {
    /** Falha ao conectar, retorna erro */
    return -1;
}

/** Pega o primeiro comando */
while (true) {
    /** Pega o comando */
    str_command = readline(PROMPT);

    /** Adiciona no histórico */
    add_history(str_command);

    /** Verifica se o comando é o exit pra sair */
    if (strcmp(str_command, "exit") == 0 || strcmp(str_command, "quit") == 0) {
        /** Escreve o arquivo de histórico */
        write_history(filename);

        /** Fecha a conexão */
        close_server_connection(cli);

        /** Lib era memória */
        free(cli);

        /** Sai do loop */
        break;
    }

    /** Faz o encode */
    strcpy(cli->command, encode_command(str_command));

    /** Envia para o server */
    if (send_command(cli) == SEND_COMMAND_FAILED) {
        /** Falha ao enviar o comando */
        fprintf(stderr, "Não foi possível executar o comando\n");
    }

    /** Espera a resposta */
    if (recv_response(cli) == RECV_RESPONDE_FAILED) {
        /** Informa qual foi o erro */
        fprintf(stderr, "Não foi possível receber a resposta do socket\n");
    } else {
        /** Imprime qual foi o resultado */
        printf("%s\n", cli->response);
    }
}

return 0;

}

当我按下ctrl-d,并且程序尝试将命令与“exit”或“quit”进行比较时,我收到一个seg错误。

有人知道我该如何处理?

1 个答案:

答案 0 :(得分:7)

您没有检查readline返回的值。从手册页:

  

如果在读取一行时遇到EOF,并且该行为空,则返回NULL。

所以也许在使用之前检查NULL

str_command = readline(PROMPT);
if (!str_command) {
    printf("EOF"
    /* And probably break. */
}