mirror of
https://github.com/wesnoth/wesnoth
synced 2024-09-21 19:25:32 +00:00
136 lines
2.5 KiB
C
136 lines
2.5 KiB
C
/* Random util functions, mainly taken from stdrusty.h */
|
|
#define _GNU_SOURCE
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <limits.h>
|
|
#include <stdint.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <errno.h>
|
|
#include <assert.h>
|
|
#include "utils.h"
|
|
|
|
void *_realloc_array(void *ptr, size_t size, size_t num)
|
|
{
|
|
if (num >= SIZE_MAX/size)
|
|
return NULL;
|
|
return realloc_nofail(ptr, size * num);
|
|
}
|
|
|
|
void *realloc_nofail(void *ptr, size_t size)
|
|
{
|
|
ptr = realloc(ptr, size);
|
|
if (ptr)
|
|
return ptr;
|
|
barf("realloc of %zu failed", size);
|
|
}
|
|
|
|
void *malloc_nofail(size_t size)
|
|
{
|
|
void *ptr = malloc(size);
|
|
if (ptr)
|
|
return ptr;
|
|
barf("malloc of %zu failed", size);
|
|
}
|
|
|
|
/* Avoid stupid sprintf-style non-value return. */
|
|
char *aprintf(const char *fmt, ...)
|
|
{
|
|
char *ret;
|
|
va_list arglist;
|
|
|
|
va_start(arglist, fmt);
|
|
vasprintf(&ret, fmt, arglist);
|
|
va_end(arglist);
|
|
return ret;
|
|
}
|
|
|
|
char *aprintf_add(char *s, const char *fmt, ...)
|
|
{
|
|
int oldlen, len;
|
|
va_list arglist;
|
|
|
|
va_start(arglist, fmt);
|
|
len = vsnprintf(NULL, 0, fmt, arglist);
|
|
va_end(arglist);
|
|
|
|
oldlen = (s ? strlen(s) : 0);
|
|
s = realloc(s, oldlen + len + 1);
|
|
|
|
va_start(arglist, fmt);
|
|
vsnprintf(s + oldlen, len + 1, fmt, arglist);
|
|
va_end(arglist);
|
|
return s;
|
|
}
|
|
|
|
void logpv(const char *fmt, va_list arglist)
|
|
{
|
|
char *str;
|
|
vasprintf(&str, fmt, arglist);
|
|
write(log_fd, str, strlen(str));
|
|
}
|
|
|
|
void logp(const char *fmt, ...)
|
|
{
|
|
va_list arglist;
|
|
|
|
va_start(arglist, fmt);
|
|
logpv(fmt, arglist);
|
|
write(log_fd, "\n", 1);
|
|
va_end(arglist);
|
|
}
|
|
|
|
void barf(const char *fmt, ...)
|
|
{
|
|
va_list arglist;
|
|
|
|
write(log_fd, "FATAL: ", strlen("FATAL: "));
|
|
va_start(arglist, fmt);
|
|
logpv(fmt, arglist);
|
|
write(log_fd, "\n", 1);
|
|
va_end(arglist);
|
|
exit(1);
|
|
}
|
|
|
|
void barf_perror(const char *fmt, ...)
|
|
{
|
|
int err = errno;
|
|
va_list arglist;
|
|
|
|
fprintf(stderr, "FATAL: ");
|
|
va_start(arglist, fmt);
|
|
logpv(fmt, arglist);
|
|
va_end(arglist);
|
|
logp(": %s", strerror(err));
|
|
exit(1);
|
|
}
|
|
|
|
/* This is all squid allows in a POST by default anyway. */
|
|
#define MAXIMUM_LEN 40000
|
|
|
|
/* This version adds one byte (for nul term) */
|
|
void *grab_input(int fd, unsigned long *size)
|
|
{
|
|
int ret;
|
|
void *buffer;
|
|
|
|
buffer = malloc(MAXIMUM_LEN+1);
|
|
*size = 0;
|
|
while ((ret = read(fd, buffer + *size, MAXIMUM_LEN - *size)) > 0) {
|
|
*size += ret;
|
|
if (*size == MAXIMUM_LEN)
|
|
barf("too much input");
|
|
}
|
|
if (ret < 0)
|
|
barf_perror("read failure");
|
|
|
|
((char *)buffer)[*size] = '\0';
|
|
return buffer;
|
|
}
|
|
|
|
void _delete_arr(void *p, unsigned len, unsigned off, unsigned num, size_t s)
|
|
{
|
|
assert(off + num <= len);
|
|
memmove(p + off*s, p + (off+num)*s, (len - (off+num))*s);
|
|
}
|