106 lines
3.5 KiB
C
106 lines
3.5 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 │
|
|
│ │
|
|
│ This program is free software; you can redistribute it and/or modify │
|
|
│ it under the terms of the GNU General Public License as published by │
|
|
│ the Free Software Foundation; version 2 of the License. │
|
|
│ │
|
|
│ This program is distributed in the hope that it will be useful, but │
|
|
│ WITHOUT ANY WARRANTY; without even the implied warranty of │
|
|
│ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU │
|
|
│ General Public License for more details. │
|
|
│ │
|
|
│ You should have received a copy of the GNU General Public License │
|
|
│ along with this program; if not, write to the Free Software │
|
|
│ Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA │
|
|
│ 02110-1301 USA │
|
|
╚─────────────────────────────────────────────────────────────────────────────*/
|
|
#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;
|
|
}
|
|
}
|