/*
* Error handling for libcdecl.
- * Copyright © 2011-2012, 2021 Nick Bowler
+ * Copyright © 2011-2012, 2021, 2023 Nick Bowler
*
* 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
*/
#include <config.h>
+#include <stdio.h>
#include <stdlib.h>
-#include <string.h>
#include <assert.h>
-#include <glthread/lock.h>
-#include <glthread/tls.h>
-
#include "cdecl.h"
#include "cdecl-internal.h"
-gl_once_define(static, tls_initialized);
-static gl_tls_key_t tls_key;
+#include <glthread/lock.h>
+#include <glthread/tls.h>
+
+#include "errmsg.h"
struct err_state {
struct cdecl_error err;
char str[FLEXIBLE_ARRAY_MEMBER];
};
-/* This error is reserved for extremely dire out-of-memory conditions. */
-static struct err_state err_no_mem = {
- .err = {
- .code = CDECL_ENOMEM,
- .str = NULL,
- },
-};
+/* This pre-initialized error is reserved for dire out-of-memory conditions. */
+static struct cdecl_error err_no_mem;
+static gl_tls_key_t tls_key;
+
+static void free_err(void *err)
+{
+ if (err == &err_no_mem)
+ return;
+
+ free(err);
+}
-const char *cdecl__strerror(unsigned code)
+static void set_err(unsigned code, struct cdecl_error *err)
{
-#include "errtab.h"
+ static const char errmsgs[] = STRTAB_INITIALIZER;
switch (code) {
- case CDECL_ENOMEM: return gettext(strtab+err_enomem);
- case CDECL_ENOPARSE: return gettext(strtab+err_enoparse);
+ case CDECL__ENOMEM:
+ err->code = CDECL_ENOMEM;
+ break;
+ default:
+ err->code = CDECL_ENOPARSE;
+ break;
}
- assert(0);
+ err->str = _(&errmsgs[code]);
}
-static void free_err(void *err)
+static void initialize_cb(void)
{
- if (err == &err_no_mem)
- return;
-
- free(err);
+#if ENABLE_NLS
+ bindtextdomain(PACKAGE, LOCALEDIR);
+ bindtextdomain("bison-runtime", BISON_LOCALEDIR);
+#endif
+ set_err(CDECL__ENOMEM, &err_no_mem);
+ gl_tls_key_init(tls_key, free_err);
}
-static void set_error(const struct cdecl_error *err)
+static void *alloc_err_state(void *old, size_t buf_size)
{
struct err_state *state;
- size_t need_len = 0;
-
- if (err->str) {
- need_len = strlen(err->str) + 1;
+ void *p;
+
+ state = p = realloc(old, offsetof(struct err_state, str) + buf_size);
+ if (state) {
+ state->nstr = buf_size;
+ } else if (old) {
+ /* Failed allocation, but existing state is still good */
+ p = old;
+ } else {
+ /* Failed allocation, no existing state */
+ p = &err_no_mem;
}
+ gl_tls_set(tls_key, p);
+ return state;
+}
+
+static struct err_state *get_err_state(void)
+{
+ void *state;
+
+ gl_once_define(static, tls_initialized)
+ gl_once(tls_initialized, initialize_cb);
+
state = gl_tls_get(tls_key);
if (state == &err_no_mem)
state = NULL;
- if (!state || state->nstr < need_len) {
- struct err_state *tmp;
-
- tmp = realloc(state, sizeof *state + need_len);
- if (!tmp) {
- /* Re-use the existing state buffer, if any. */
- if (state)
- state->err = err_no_mem.err;
- else
- state = &err_no_mem;
-
- gl_tls_set(tls_key, state);
- return;
- }
+ if (!state)
+ return alloc_err_state(state, 100);
+ return state;
+}
- state = tmp;
- state->nstr = need_len;
- gl_tls_set(tls_key, state);
- }
- state->err = *err;
- if (err->str) {
- memcpy(state->str, err->str, need_len);
- state->err.str = state->str;
- } else {
- state->err.str = cdecl__strerror(state->err.code);
- }
+#if ENABLE_NLS
+/*
+ * Initialize gettext indirectly via get_err_state.
+ */
+void cdecl__init_i18n(void)
+{
+ get_err_state();
}
+#endif
-static void initialize(void)
+/*
+ * Set the library error to one of the preset messages defined in errmsg.h
+ * (CDECL__Exxx).
+ */
+void cdecl__errmsg(unsigned msg)
{
- cdecl__init_i18n();
- err_no_mem.err.str = cdecl__strerror(CDECL_ENOMEM);
+ struct err_state *state;
- gl_tls_key_init(tls_key, free_err);
+ state = get_err_state();
+ if (!state)
+ return;
- /*
- * This default error message is a stop-gap measure until all library
- * error conditions use the new interface.
- */
- set_error(&(const struct cdecl_error){ .code = CDECL_ENOPARSE });
+ set_err(msg, &state->err);
}
-const struct cdecl_error *cdecl_get_error(void)
+/*
+ * Sets the library error to code; fmt is a printf-style string that may use
+ * up to one %s directive, to refer to arg.
+ */
+void cdecl__err(unsigned code, const char *fmt, const char *arg)
{
struct err_state *state;
+ int rc, try = 0;
- gl_once(tls_initialized, initialize);
+ state = get_err_state();
+ if (!state)
+ return;
- state = gl_tls_get(tls_key);
- assert(state);
+retry:
+ rc = snprintf(state->str, state->nstr, fmt, arg);
+ if (rc > 0 && rc >= state->nstr) {
+ assert(try++ == 0 && rc < SIZE_MAX / 4);
+ state = alloc_err_state(state, (size_t)(rc+1u) * 3 / 2);
+ if (!state)
+ return;
- return &state->err;
+ goto retry;
+ }
+
+ state->err.str = state->str;
+ state->err.code = code;
}
-void cdecl__set_error(const struct cdecl_error *err)
+const struct cdecl_error *cdecl_get_error(void)
{
- gl_once(tls_initialized, initialize);
+ struct err_state *state = get_err_state();
- set_error(err);
+ return state ? &state->err : NULL;
}