Управлять протоколом связи локальным сокетом c

Question

Я пытаюсь реализовать свое первое клиент-серверное приложение с сокетом AF_UNIX в C.

Я должен соблюдать определенный протокол связи для запросов, отправленных клиентом.Доступны три типа запросов:

1) <command_name> <object_name> <obj_len> \n <obj_data>

2) <command_name> <object_name> \n

3) <command_name> \n

Обратите внимание, что каждый запрос, кроме первого, заканчивается специальным символом \ n, а длина каждого сообщения неизвестна.

У меня нет проблемчитать 2) и 3) потому что я знаю, что сообщение будет заканчиваться на \ n, поэтому мне просто нужно проверить, является ли последний байт, прочитанный в буфере, \ n, если нет, я продолжу читать из сокета и добавлю последующиевызовы в буфер.

Вместо этого для первой операции вышеуказанный метод не работает, потому что символ \ n находится не в конце сообщения, а в произвольной точке буфера.Что я хотел бы сделать, это прочитать (в этом случае), пока я не достигну \ n, затем проанализировать токен obj_len из буфера и выполнить последнее чтение байтов obj_len из сокета, содержащеговсе содержимое obj_data .

Вот код моей функции readLine, которая работает только с 1) и 2) типами сообщений:

char * readLine(int fd){
    char * buf=NULL; //here i store the content 
    char * tmp=calloc(CHUNK,sizeof(char));
    if(!tmp)
        return NULL;
    if(!buf)
        return NULL;
    buf[0]='\0';
    size_t byte_read=-1; 
    int len=0;
    do{ 
        bzero(tmp,CHUNK); //reset tmp for read
        byte_read=read(fd,(void *)tmp,CHUNK);
        if(byte_read==-1){
            perror("read");
            if(tmp)free(tmp);
            if(buf)free(buf);
            return NULL;
        }
        len=len+byte_read; //update the len of message
        buf=realloc(buf,len+1);
        if(!buf){
            perror("realloc");
            if(tmp)free(tmp);
            return NULL;
        }
        buf=strncat(buf,tmp,byte_read);//append each call of read
        if(byte_read>0 && buf[byte_read-1]=='\n')
        //the last byte read is  the special character
            break;
    }   
    while(byte_read!=0);
    if(byte_read==0)
        //read error SIGPIPE
    if(tmp)free(tmp);
    return buf;
    }

Что мне делатьКак вы думаете, для реализации функции, которая читает сообщения, отформатированные как 2), 3) и 1) из сокета?

Answer 1

Я предлагаю другой подход.Сначала прочитайте строку, затем определите, что это за сообщение, и только после этого решите, является ли это концом, или вам нужно прочитать больше байтов.

Итак, начнем с чтения строки:

int socket_readline(int fd, char *buf, size_t *inout_size)
{   
    char c = 0;
    size_t capacity = *inout_size;
    *inout_size = 0;

    while(c != '\n') {
        int rc = read(fd, &c, 1);

        if (rc != 1) return -1;

        buf[*inout_size] = c;
        *inout_size += 1;

        if (capacity - *inout_size == 0) return -1;
    }

    buf[*inout_size] = '\0';

    return 0;
}

Эта процедура возвращает 0 в случае успеха и -1 в противном случае.

В успешном случае буфер определяется NUL, а inout_size будет длиной строки в буфере (исключая байт, определяющий NUL).В случае ошибки вы должны проверить значение inout_size.в случае переполнения (слишком длинная строка) он будет равен размеру заданного буфера (входной буфер не будет определен NUL!), а для ошибки чтения он будет содержать количество байтов, прочитанных допроизошла ошибка.

Теперь, когда у нас есть подпрограмма чтения строк, мы можем использовать ее для анализа входящих строк:

int rc;
size_t size;
char buf[MAX_INPUT_LENGTH];

int object_len;
char command_name[MAX_INPUT_LENGTH], object_name[MAX_INPUT_LENGTH];

do {
    size = sizeof(buf);
    rc = socket_readline(fd, buf, &size);
    if (rc == 0) {
        int tokens_matched = sscanf(buf, "%s %s %d \n", command_name, object_name, &object_len);

        switch(tokens_matched) {
            case 1:
                printf("This is a command: %s\n", command_name);
                break;
            case 2:
                printf("This is a command: %s with object name: %s\n", command_name, object_name);
                break;
            case 3:
                printf("This is a command: %s with object name: %s of length %d\n", command_name, object_name, object_len);
                printf("TODO read up remaining: %d bytes\n", object_len);
                break;
            default:
                printf("Error unable to match command format\n");
                break;
        }

    } else if (size == sizeof(buf)) {
        printf("Overflow! Bytes read: %lu. \n", size);
        // TODO How to handle overflows?
    } else {
        printf("Read error must have occured. Read %lu bytes.\n", size);
    }

} while (rc == 0 || size == sizeof(buf));