140 lines
3.0 KiB
C
140 lines
3.0 KiB
C
/*
|
|
* Copyright (c) 2011, 2012, 2013, 2014, 2015, 2022 Jonas 'Sortie' Termansen.
|
|
*
|
|
* Permission to use, copy, modify, and 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.
|
|
*
|
|
* util.c
|
|
* Utility functions.
|
|
*/
|
|
|
|
#include <stdbool.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "util.h"
|
|
|
|
char* strdup_safe(const char* string)
|
|
{
|
|
return string ? strdup(string) : NULL;
|
|
}
|
|
|
|
const char* getenv_safe_def(const char* name, const char* def)
|
|
{
|
|
const char* ret = getenv(name);
|
|
return ret ? ret : def;
|
|
}
|
|
|
|
const char* getenv_safe(const char* name)
|
|
{
|
|
return getenv_safe_def(name, "");
|
|
}
|
|
|
|
bool array_add(void*** array_ptr,
|
|
size_t* used_ptr,
|
|
size_t* length_ptr,
|
|
void* value)
|
|
{
|
|
void** array;
|
|
memcpy(&array, array_ptr, sizeof(array)); // Strict aliasing.
|
|
|
|
if ( *used_ptr == *length_ptr )
|
|
{
|
|
size_t length = *length_ptr;
|
|
if ( !length )
|
|
length = 4;
|
|
void** new_array = reallocarray(array, length, 2 * sizeof(void*));
|
|
if ( !new_array )
|
|
return false;
|
|
array = new_array;
|
|
memcpy(array_ptr, &array, sizeof(array)); // Strict aliasing.
|
|
*length_ptr = length * 2;
|
|
}
|
|
|
|
memcpy(array + (*used_ptr)++, &value, sizeof(value)); // Strict aliasing.
|
|
|
|
return true;
|
|
}
|
|
|
|
void stringbuf_begin(struct stringbuf* buf)
|
|
{
|
|
buf->length = 0;
|
|
buf->allocated_size = 32;
|
|
if ( !(buf->string = (char*) malloc(buf->allocated_size)) )
|
|
{
|
|
buf->allocated_size = 0;
|
|
return;
|
|
}
|
|
buf->string[0] = '\0';
|
|
}
|
|
|
|
bool might_need_shell_quote(char c)
|
|
{
|
|
switch ( c )
|
|
{
|
|
case '|':
|
|
case '&':
|
|
case ';':
|
|
case '<':
|
|
case '>':
|
|
case '(':
|
|
case ')':
|
|
case '$':
|
|
case '`':
|
|
case '\\':
|
|
case '"':
|
|
case '\'':
|
|
case ' ':
|
|
case '\t':
|
|
case '*':
|
|
case '?':
|
|
case '[':
|
|
case '#':
|
|
case '~':
|
|
case '=':
|
|
case '%':
|
|
return true;
|
|
default:
|
|
return false;
|
|
}
|
|
}
|
|
|
|
void stringbuf_append_c(struct stringbuf* buf, char c)
|
|
{
|
|
if ( !buf->string )
|
|
return;
|
|
if ( buf->length + 1 == buf->allocated_size )
|
|
{
|
|
// TODO: Avoid overflow.
|
|
size_t new_allocated_size = 2 * buf->allocated_size;
|
|
char* new_string = (char*) realloc(buf->string, new_allocated_size);
|
|
if ( !new_string )
|
|
{
|
|
buf->string = NULL;
|
|
buf->length = 0;
|
|
buf->allocated_size = 0;
|
|
return;
|
|
}
|
|
buf->string = new_string;
|
|
buf->allocated_size = new_allocated_size;
|
|
}
|
|
buf->string[buf->length++] = c;
|
|
buf->string[buf->length] = '\0';
|
|
}
|
|
|
|
char* stringbuf_finish(struct stringbuf* buf)
|
|
{
|
|
if ( !buf->string )
|
|
return NULL;
|
|
return buf->string;
|
|
}
|