kmx git

/*
 * kv - key value text file format
 *
 * Copyright 2022 Thomas de Grivel
*/

#include <assert.h>
#include <stdlib.h>
#include <string.h>
#include "kv.h"
#include "rope.h"

const char *g_delims[] = {
  "END",
  "---",
  "___",
  "===",
  "***",
  "+++",
  "!!!",
  "###",
  "-END-",
  "_END_",
  "=END=",
  "*END*",
  "+END+",
  "!END!",
  "#END#",
  NULL
};

int kv_check_delimiter (const char *delim)
{
  assert(delim);
  return kv_contains_reserved_char(delim, strlen(delim));
}

int kv_contains_delimiter (const char *str, size_t str_size,
                           const char *delim)
{
  size_t dsz;
  assert(str);
  assert(delim);
  dsz = strlen(delim);
  while (str_size >= 4) {
    if (str[0] == '\n' && str[1] == '"' && str[2] == '"' &&
        str[3] == '"')
      return 1;
    str++;
    str_size--;
  }
  return 0;
}

int kv_contains_reserved_char (const char *str, size_t str_size)
{
  while (str_size) {
    if (kv_is_reserved_char(str[0]))
      return 1;
    str++;
    str_size--;
  }
  return 0;
}

int kv_contains_triple_double_quotes (const char *str, size_t str_size)
{
  while (str_size >= 4) {
    if (str[0] == '\n' && str[1] == '"' && str[2] == '"' &&
        str[3] == '"')
      return 1;
    str++;
    str_size--;
  }
  return 0;
}

int kv_needs_quoting (const char *str, size_t str_size)
{
  if (str_size == 0 ||
      (str[0] != '<' && !kv_contains_reserved_char(str, str_size)))
    return 0;
  return 1;
}

int kv_quote (char **str, size_t *str_size)
{
  const char **delim;
  int          r;
  s_rope      *rope = NULL;
  char        *q;
  size_t       q_size;
  assert(str);
  assert(*str);
  assert(str_size);
  if (!kv_needs_quoting(*str, *str_size))
    return 0;
  q = *str;
  q_size = *str_size;
  r = kv_quote_double_quotes(&q, &q_size);
  if (r < 0)
    return r;
  if (r == 0 && rope_push(&rope, q, q_size))
    return -3;
  q = *str;
  q_size = *str_size;
  r = kv_quote_triple_double_quotes(&q, &q_size);
  if (r < 0)
    return r;
  if (r == 0 && rope_push(&rope, q, q_size)) {
    rope_delete_all_free(rope);
    return -3;
  }
  q = *str;
  q_size = *str_size;
  delim = &g_delims[0];
  r = kv_quote_delim(&q, &q_size, *delim);
  delim++;
  while (*delim && r > 0) {
    r = kv_quote_delim(&q, &q_size, *delim);
    delim++;
  }
  if (r < 0)
    return r;
  if (r == 0 && rope_push(&rope, q, q_size)) {
    rope_delete_all_free(rope);
    return -3;
  }
  if (!rope)
    return 1;
  rope_sort_by_size(rope);
  *str = rope->str;
  *str_size = rope->size;
  rope = rope->next;
  rope_delete_all_free(rope);
  return 0;
}

size_t kv_count_reserved_chars (const char *str, size_t str_size)
{
  size_t c = 0;
  while (str_size--)
    if (kv_is_reserved_char(*str++))
      c++;
  return c;
}

int kv_quote_double_quotes (char **str, size_t *str_size)
{
  size_t i = 0;
  size_t o = 0;
  size_t res;
  char  *s;
  assert(str);
  assert(str_size);
  res = kv_count_reserved_chars(*str, *str_size);
  s = calloc(*str_size + res + 3, 1);
  if (!s)
    return -1;
  s[0] = '"';
  while (i < *str_size) {
    int c = (*str)[i];
    if (kv_is_reserved_char(c))
      s[o++] = '\\';
    if (c == 0)
      s[o++] = '0';
    else
      s[o++] = c;
  }
  assert(o == *str_size + res + 2);
  s[o] = 0;
  *str = s;
  *str_size = o;
  return 0;
}

int kv_quote_triple_double_quotes (char **str, size_t *str_size)
{
  char *e;
  char *s;
  if (kv_contains_triple_double_quotes(*str, *str_size))
    return 1;
  s = calloc(*str_size + 9, 1);
  if (!s)
    return -1;
  s[0] = '"'; s[1] = '"'; s[2] = '"'; s[3] = '\n';
  memcpy(s + 4, *str, *str_size);
  e = s + 4 + *str_size;
  e[0] = '\n'; e[1] = '"'; e[2] = '"'; e[3] = '"'; e[4] = 0;
  *str = e;
  *str_size = *str_size + 8;
  return 0;
}
kmx.io/kv/kv_quote.c

Commit

kv_quote.c
