rlib/rbuffer.h at main

 #ifndef RBUFFER_H
 #define RBUFFER_H
 #include "rmalloc.h"
 #include <stdlib.h>
 #include <stdio.h>
 #include <string.h>
 #include <stdbool.h>
 #include <assert.h>
 typedef struct rbuffer_t {
     unsigned char *data;
     unsigned char *_data;
     size_t size;
     size_t pos;
     bool eof;
 } rbuffer_t;
 rbuffer_t *rbuffer_new(unsigned char *data, size_t size);
 void rbuffer_free(rbuffer_t *rfb);
 void rbuffer_reset(rbuffer_t *rfb);
 void rbuffer_write(rbuffer_t *rfb, const unsigned char *data, size_t size);
 size_t rbuffer_push(rbuffer_t *rfb, unsigned char);
 unsigned char rbuffer_pop(rbuffer_t *rfb);
 unsigned char *rbuffer_expect(rbuffer_t *rfb, char *options, char *ignore);
 void rbuffer_set(rbuffer_t *rfb, const unsigned char *data, size_t size);
 void rbuffer_set(rbuffer_t *rfb, const unsigned char *data, size_t size) {
     if (rfb->_data) {
         free(rfb->_data);
         rfb->_data = NULL;
         rfb->data = NULL;
         rfb->eof = true;
     }
     if (size) {
         rfb->_data = (unsigned char *)malloc(size);
         memcpy(rfb->_data, data, size);
         rfb->data = rfb->_data;
         rfb->eof = false;
     }
     rfb->size = size;
     rfb->pos = 0;
 }
 rbuffer_t *rbuffer_new(unsigned char *data, size_t size) {
     rbuffer_t *rfb = (rbuffer_t *)malloc(sizeof(rbuffer_t));
     if (size) {
         rfb->_data = (unsigned char *)malloc(size);
         memcpy(rfb->_data, data, size);
         rfb->eof = false;
     } else {
         rfb->_data = NULL;
         rfb->eof = true;
     }
     rfb->size = size;
     rfb->pos = 0;
     rfb->data = rfb->_data;
     return rfb;
 }
 void rbuffer_free(rbuffer_t *rfb) {
     if (rfb->_data)
         free(rfb->_data);
     free(rfb);
 }
 size_t rbuffer_push(rbuffer_t *rfb, unsigned char c) {
     if (rfb->pos < rfb->size) {
         rfb->_data[rfb->pos++] = c;
         return 1;
     }
     rfb->_data = realloc(rfb->_data, rfb->size ? rfb->size + 1 : rfb->size + 2);
     rfb->_data[rfb->pos++] = c;
     rfb->size++;
     return rfb->pos;
 }
 void rbuffer_write(rbuffer_t *rfb, const unsigned char *data, size_t size) {
     unsigned char *data_ptr = (unsigned char *)data;
     for (size_t i = 0; i < size; i++) {
         rbuffer_push(rfb, data_ptr[i]);
     }
 }
 unsigned char rbuffer_peek(rbuffer_t *rfb) {
     unsigned char result = EOF;
     if (rfb->pos != rfb->size) {
         result = rfb->_data[rfb->pos];
         return result;
     }
     rfb->eof = true;
     return EOF;
 }
 unsigned char rbuffer_pop(rbuffer_t *rfb) {
     unsigned char result = EOF;
     if (rfb->pos <= rfb->size) {
         result = rfb->_data[rfb->pos];
         rfb->pos++;
         rfb->data++;
         if (rfb->pos == rfb->size) {
             rfb->eof = true;
         }
         return result;
     }
     rfb->eof = true;
     return result;
 }
 void rbuffer_reset(rbuffer_t *rfb) {
     rfb->data = rfb->_data;
     rfb->pos = 0;
 }
 unsigned char ustrncmp(const unsigned char *s1, const unsigned char *s2, size_t n) {
     return strncmp((char *)s1, (char *)s2, n);
     while (n && *s1 == *s2) {
         n--;
         s1++;
         s2++;
     }
     return *s1 != *s2;
 }
 size_t ustrlen(const unsigned char *s) { return strlen((char *)s); }
 unsigned char *rbuffer_to_string(rbuffer_t *rfb) {
     unsigned char *result = rfb->_data;
     rfb->_data = NULL;
     rfb->data = NULL;
     rbuffer_free(rfb);
     return result;
 }
 unsigned char *rbuffer_match_option(rbuffer_t *rfb, char *options) {
     char *option = NULL;
     char options_cpy[1024] = {0};
     strcpy(options_cpy, options);
     char *memory = options_cpy;
     while ((option = strtok_r(option == NULL ? memory : NULL, "|", &memory)) != NULL) {
         size_t option_length = strlen(option);
         if (option_length > rfb->size - rfb->pos) {
             continue;
         }
         if (!strcmp(option, "\\d") && *rfb->data >= '0' && *rfb->data <= '9') {
             return rfb->data;
         }
         if (rfb->size - rfb->pos >= 5 && !strcmp(option, "\\b") &&
             ((!ustrncmp(rfb->data, (unsigned char *)"true", 4) || !ustrncmp(rfb->data, (unsigned char *)"false", 5)))) {
             return rfb->data;
         }
         if (!ustrncmp(rfb->data, (unsigned char *)option, option_length)) {
             return rfb->data;
         }
     }
     return NULL;
 }
 unsigned char *rbuffer_expect(rbuffer_t *rfb, char *options, char *ignore) {
     while (rfb->pos < rfb->size) {
         if (rbuffer_match_option(rfb, options) != NULL) {
             return rfb->data;
         }
         if (rbuffer_match_option(rfb, ignore)) {
             printf("SKIP:%s\n", rfb->data);
             rbuffer_pop(rfb);
             continue;
         }
         break;
     }
     return NULL;
 }
 unsigned char *rbuffer_consume(rbuffer_t *rfb, char *options, char *ignore) {
     unsigned char *result = NULL;
     if ((result = rbuffer_expect(rfb, options, ignore)) != NULL) {
         rbuffer_pop(rfb);
     }
     return result;
 }
 #endif