Получение SMTP-сообщения - PullRequest
Новая
       6

Получение SMTP-сообщения

2 голосов
/ 17 июня 2019

Я пытаюсь реализовать SMTP-сервер.
Проблема в том, что после моего сокета accept s новое соединение не получает байтов (пустая строка в примере вывода), и поэтому мой цикл read завершается.

Я пытался отправить 220 и 250 команды после начального accept, но все еще безуспешно.

Мой вывод: 

Socket created
Waiting for incoming connections...
Connection accepted
250 mail.domain.com says hello

Finished reading current message

Мой код: 

#include <stdio.h>
#include <string.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>

#define MAX_CONNECTIONS 10
#define BUF_SIZE        2000
#define STREQU(a,b)     (strcmp(a, b) == 0)

void init_local_sockaddr_in(struct sockaddr_in *sock, int portnum) {
    sock->sin_family = AF_INET;
    sock->sin_addr.s_addr = inet_addr("127.0.0.1");
    sock->sin_port = htons(portnum);
}

int main(int argc, char *argv[])
{
    int listen_sockfd, input_sockfd, c, read_size, n;
    struct sockaddr_in server, client;
    char client_message[2000] = {0};
    char bufferout[2000] = {0};

    listen_sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (listen_sockfd == -1) {
        printf("Could not create sockets");
    }
    puts("Socket created");

    // Just for better debugging
    int optval = 1;
    setsockopt(listen_sockfd, SOL_SOCKET, SO_REUSEADDR, (const void *)&optval , sizeof(int));

    init_local_sockaddr_in(&server, 10025);
    n = bind(listen_sockfd, (struct sockaddr*) &server, sizeof(server));
    if (n < 0) {
        perror("bind failed. Error");
        return 1;
    }

    listen(listen_sockfd , MAX_CONNECTIONS);

    puts("Waiting for incoming connections...");
    c = sizeof(struct sockaddr_in);

    while(1) {
        input_sockfd = accept(listen_sockfd, (struct sockaddr*) &client, (socklen_t*) &c);
        if (input_sockfd < 0) {
            perror("accept failed");
            return 1;
        }
        puts("Connection accepted");

        sprintf(bufferout, "250 mail.domain.com says hello\r\n"); // I tried different replies here, including 220
        send(input_sockfd, bufferout, strlen(bufferout), 0);
        puts(bufferout);

        int keep_running = 1;

        while (keep_running) {
            n = read(listen_sockfd, client_message, sizeof(client_message));
            if (STREQU(client_message, ".") || n < 0) {
                keep_running = 0;
                break;
            }
            printf("Read %lu bytes: %s\n", strlen(client_message), client_message);
            memset(client_message, 0, sizeof(client_message));
        }
        close(input_sockfd);
        puts("Finished reading current message");
    }
    close(listen_sockfd);

    return 0;
}

1 Ответ

2 голосов
/ 17 июня 2019

Вы читаете не из того сокета. accept возвращает сокет, из которого вы должны прочитать, то есть input_sockfd. Вы читаете из сокета, который вы передали accept, который не подключен к клиенту, поэтому read правильно возвращает 0 байтов.

У вас также есть чек на счету n < 0. read возвращает 0, когда соединение закрыто, так что вы действительно должны проверить n <= 0, иначе цикл будет продолжаться вечно, всегда читая 0 байтов.

...