105 lines
3.4 KiB
C
105 lines
3.4 KiB
C
/*-*- mode:c;indent-tabs-mode:nil;c-basic-offset:2;tab-width:8;coding:utf-8 -*-│
|
|
│vi: set net ft=c ts=2 sts=2 sw=2 fenc=utf-8 :vi│
|
|
╞══════════════════════════════════════════════════════════════════════════════╡
|
|
│ Copyright 2020 Justine Alexandra Roberts Tunney │
|
|
│ │
|
|
│ Permission to use, copy, modify, and/or distribute this software for │
|
|
│ any purpose with or without fee is hereby granted, provided that the │
|
|
│ above copyright notice and this permission notice appear in all copies. │
|
|
│ │
|
|
│ THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL │
|
|
│ WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED │
|
|
│ WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE │
|
|
│ AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL │
|
|
│ DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR │
|
|
│ PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER │
|
|
│ TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR │
|
|
│ PERFORMANCE OF THIS SOFTWARE. │
|
|
╚─────────────────────────────────────────────────────────────────────────────*/
|
|
#include "libc/alg/arraylist2.internal.h"
|
|
#include "libc/calls/calls.h"
|
|
#include "libc/errno.h"
|
|
#include "libc/fmt/fmt.h"
|
|
#include "libc/macros.h"
|
|
#include "libc/mem/mem.h"
|
|
#include "libc/str/str.h"
|
|
#include "libc/str/tpenc.h"
|
|
#include "tool/build/lib/buffer.h"
|
|
|
|
void AppendData(struct Buffer *b, char *data, unsigned len) {
|
|
char *p;
|
|
unsigned n;
|
|
if (b->i + len + 1 > b->n) {
|
|
n = MAX(b->i + len + 1, MAX(16, b->n + (b->n >> 1)));
|
|
if (!(p = realloc(b->p, n))) return;
|
|
b->p = p;
|
|
b->n = n;
|
|
}
|
|
memcpy(b->p + b->i, data, len);
|
|
b->p[b->i += len] = 0;
|
|
}
|
|
|
|
void AppendChar(struct Buffer *b, char c) {
|
|
AppendData(b, &c, 1);
|
|
}
|
|
|
|
void AppendStr(struct Buffer *b, const char *s) {
|
|
AppendData(b, s, strlen(s));
|
|
}
|
|
|
|
void AppendWide(struct Buffer *b, wint_t wc) {
|
|
unsigned i;
|
|
uint64_t wb;
|
|
char buf[8];
|
|
i = 0;
|
|
wb = tpenc(wc);
|
|
do {
|
|
buf[i++] = wb & 0xFF;
|
|
wb >>= 8;
|
|
} while (wb);
|
|
AppendData(b, buf, i);
|
|
}
|
|
|
|
int AppendFmt(struct Buffer *b, const char *fmt, ...) {
|
|
int bytes;
|
|
char *tmp;
|
|
va_list va;
|
|
tmp = NULL;
|
|
va_start(va, fmt);
|
|
bytes = vasprintf(&tmp, fmt, va);
|
|
va_end(va);
|
|
if (bytes != -1) AppendData(b, tmp, bytes);
|
|
free(tmp);
|
|
return bytes;
|
|
}
|
|
|
|
/**
|
|
* Writes buffer until completion, or error occurs.
|
|
*/
|
|
ssize_t WriteBuffer(struct Buffer *b, int fd) {
|
|
bool t;
|
|
char *p;
|
|
ssize_t rc;
|
|
size_t wrote, n;
|
|
p = b->p;
|
|
n = b->i;
|
|
t = false;
|
|
do {
|
|
if ((rc = write(fd, p, n)) != -1) {
|
|
wrote = rc;
|
|
p += wrote;
|
|
n -= wrote;
|
|
} else if (errno == EINTR) {
|
|
t = true;
|
|
} else {
|
|
return -1;
|
|
}
|
|
} while (n);
|
|
if (!t) {
|
|
return 0;
|
|
} else {
|
|
errno = EINTR;
|
|
return -1;
|
|
}
|
|
}
|