summaryrefslogtreecommitdiff
path: root/src/libstddjb/netstring_put.c
blob: 692d2fcdaab5c24e7f17c27feaf879ee28fc0bf4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
/* ISC license. */

#include <sys/types.h>
#include <errno.h>
#include <skalibs/uint64.h>
#include <skalibs/buffer.h>
#include <skalibs/netstring.h>

int netstring_put (buffer *b, char const *s, size_t len, size_t *written)
{
  char fmt[UINT64_FMT] ;
  size_t n = uint64_fmt(fmt, len) ;
  if (*written > len + n + 2) return (errno = EINVAL, 0) ;
  fmt[n] = ':' ;
  if (*written < n + 1)
  {
    size_t w = *written ;
    ssize_t r = buffer_putall(b, fmt, n+1, &w) ;
    if (r < 0) return (*written = w, 0) ;
    *written = n+1 ;
  }
  if (*written < n+1 + len)
  {
    size_t w = *written - (n+1) ;
    ssize_t r = buffer_putall(b, s, len, &w) ;
    *written = w + (n+1) ;
    if (r < 0) return (*written = n+1 + w, 0) ;
    *written = n+1 + len ;
  }
  {
    size_t w = 0 ;
    ssize_t r = buffer_putall(b, ",", 1, &w) ;
    if (r < 0) return 0 ;
  }
  *written = 0 ;
  return 1 ;
}