snac2/xs_io.h

91 lines
1.5 KiB
C

/* copyright (c) 2022 - 2024 grunfink et al. / MIT license */
#ifndef _XS_IO_H
#define _XS_IO_H
xs_str *xs_readline(FILE *f);
xs_val *xs_read(FILE *f, int *size);
xs_val *xs_readall(FILE *f);
#ifdef XS_IMPLEMENTATION
xs_str *xs_readline(FILE *f)
/* reads a line from a file */
{
xs_str *s = NULL;
errno = 0;
/* don't even try on eof */
if (!feof(f)) {
int c;
s = xs_str_new(NULL);
while ((c = fgetc(f)) != EOF) {
unsigned char rc = c;
s = xs_append_m(s, (char *)&rc, 1);
if (c == '\n')
break;
}
}
return s;
}
xs_val *xs_read(FILE *f, int *sz)
/* reads up to size bytes from f */
{
xs_val *s = NULL;
int size = *sz;
int rdsz = 0;
errno = 0;
while (size > 0 && !feof(f)) {
char tmp[4096];
int n, r;
if ((n = sizeof(tmp)) > size)
n = size;
r = fread(tmp, 1, n, f);
/* open room */
s = xs_realloc(s, rdsz + r);
/* copy read data */
memcpy(s + rdsz, tmp, r);
rdsz += r;
size -= r;
}
/* null terminate, just in case it's treated as a string */
s = xs_realloc(s, _xs_blk_size(rdsz + 1));
s[rdsz] = '\0';
*sz = rdsz;
return s;
}
xs_val *xs_readall(FILE *f)
/* reads the rest of the file into a string */
{
int size = XS_ALL;
return xs_read(f, &size);
}
#endif /* XS_IMPLEMENTATION */
#endif /* _XS_IO_H */