utils.h File Reference

#include <stdio.h>
#include <stdlib.h>
#include <dirent.h>
#include <string.h>
#include <unistd.h>
#include <limits.h>
#include <errno.h>
#include "core.h"
#include "log.h"
#include "copy.h"
#include "system.h"
#include "strlist.h"
#include "utils.h"
#include "ini.h"

Include dependency graph for utils.h:

Go to the source code of this file.

Macros
#define	PATH_ENV_VAR   "PATH"
#define	DIR_SEP   "/"
#define	PATH_SEP   ":"
#define	LINE_SEP   "\n"
#define	STASIS_RANDOM_GENERATOR_FILE   NULL
#define	NEED_SRAND   1
#define	STASIS_XML_PRETTY_PRINT_PROG   "xmllint"
#define	STASIS_XML_PRETTY_PRINT_ARGS   "--format"
#define	STASIS_COLOR_RED   "\x1b[1;91m"
	Set output color to red.
#define	STASIS_COLOR_GREEN   "\x1b[1;92m"
	Set output color to green.
#define	STASIS_COLOR_YELLOW   "\x1b[1;93m"
	Set output color to yellow.
#define	STASIS_COLOR_BLUE   "\x1b[1;94m"
	Set output color to blue.
#define	STASIS_COLOR_WHITE   "\x1b[1;97m"
	Set output color to white.
#define	STASIS_COLOR_RESET   "\x1b[0;37m\x1b[0m"
	Reset output color to terminal default.
#define	STASIS_MSG_SUCCESS   0
#define	STASIS_MSG_NOP   1 << 0
	Suppress printing of the message text.
#define	STASIS_MSG_ERROR   1 << 1
	The message is an error.
#define	STASIS_MSG_WARN   1 << 2
	The message is a warning.
#define	STASIS_MSG_L1   1 << 3
	The message will be indented once.
#define	STASIS_MSG_L2   1 << 4
	The message will be indented twice.
#define	STASIS_MSG_L3   1 << 5
	The message will be indented thrice.
#define	STASIS_MSG_RESTRICT   1 << 6
	The message will only be printed in verbose mode.

Functions
int	pushd (const char *path)
int	popd (void)
char *	expandpath (const char *_path)
int	rmtree (char *_path)
char **	file_readlines (const char *filename, size_t start, size_t limit, ReaderFn *readerFn)
char *	path_basename (char *path)
char *	path_dirname (char *path)
char *	find_program (const char *name)
int	touch (const char *filename)
int	git_clone (struct Process *proc, char *url, char *destdir, char *gitref)
char *	git_describe (const char *path)
char *	git_rev_parse (const char *path, char *args)
int	path_store (char **destptr, size_t maxlen, const char *base, const char *path)
void	msg (unsigned type, char *fmt,...)
void	debug_shell ()
char *	xmkstemp (FILE **fp, const char *mode)
int	isempty_dir (const char *path)
int	xml_pretty_print_in_place (const char *filename, const char *pretty_print_prog, const char *pretty_print_args)
int	fix_tox_conf (const char *filename, char **result, size_t maxlen)
char *	collapse_whitespace (char **s)
int	redact_sensitive (const char **to_redact, size_t to_redact_size, char *src, char *dest, size_t maxlen)
struct StrList *	listdir (const char *path)
long	get_cpu_count ()
int	mkdirs (const char *_path, mode_t mode)

Function Documentation

collapse_whitespace()

char * collapse_whitespace

(

char **

s

)

Collapse all whitespace in a string (to single spaces)

Parameters

s	address of string to modify

Returns

expandpath()

char * expandpath

(

const char *

_path

)

Expand "~" to the user's home directory

char *home = expandpath("~");             // == /home/username
char *config = expandpath("~/.config");   // == /home/username/.config
char *nope = expandpath("/tmp/test");     // == /tmp/test
char *nada = expandpath("/~/broken");     // == /~/broken
 
free(home);
free(config);
free(nope);
free(nada);

Parameters

_path

(Must start with a ~)

Returns

success=expanded path or original path, failure=NULL

Here is the call graph for this function:

find_program()

char * find_program

(

const char *

name

)

Scan PATH directories for a named program

Parameters

name	program name

Returns

path to program, or NULL on error

fix_tox_conf()

int fix_tox_conf	(	const char *	filename,
		char **	result,
		size_t	maxlen )

Applies STASIS fixups to a tox ini config

Parameters

filename	path to tox.ini
result	path to processed configuration
maxlen

Returns

0 on success, -1 on error

Parameters

filename	/path/to/tox.ini
result	path of replacement tox.ini configuration
maxlen

Returns

0 on success, -1 on error

Here is the call graph for this function:

get_cpu_count()

long get_cpu_count

(

)

Get CPU count

Returns

CPU count on success, zero on error

git_clone()

int git_clone	(	struct Process *	proc,
		char *	url,
		char *	destdir,
		char *	gitref )

Clone a git repository

struct Process proc;
memset(proc, 0, sizeof(proc));
 
if (git_clone(&proc, "https://github.com/myuser/myrepo", "./repos", "unstable_branch")) {
    fprintf(stderr, "Failed to clone repository\n");
    exit(1);
}
 
if (pushd("./repos/myrepo")) {
    fprintf(stderr, "Unable to enter repository directory\n");
} else {
    // do something with repository
    popd();
}

See also

pushd

Parameters

proc	Process struct
url	URL (or file system path) of repoistory to clone
destdir	destination directory
gitref	commit/branch/tag of checkout (NULL will use HEAD of default branch for repo)

Returns

exit code from "git"

Here is the call graph for this function:

git_describe()

char * git_describe

(

const char *

path

)

Git describe wrapper

Parameters

path	to repository

Returns

output from "git describe", or NULL on error

Here is the call graph for this function:

git_rev_parse()

char * git_rev_parse	(	const char *	path,
		char *	args )

Git rev-parse wrapper

Parameters

path	to repository
args	to pass to git rev-parse

Returns

output from "git rev-parse", or NULL on error

Here is the call graph for this function:

isempty_dir()

int isempty_dir

(

const char *

path

)

Is the path an empty directory structure?

if (isempty_dir("/some/path")) {
    fprintf(stderr, "The directory is is empty!\n");
} else {
    printf("The directory contains dirs/files\n");
}

Parameters

path

directory

Returns

0 = no, 1 = yes

listdir()

struct StrList * listdir

(

const char *

path

)

Given a directory path, return a list of files

struct StrList *files;
 
basepath = ".";
files = listdir(basepath);
for (size_t i = 0; i < strlist_count(files); i++) {
    char *filename = strlist_item(files, i);
    printf("%s/%s\n", basepath, filename);
}
guard_strlist_free(&files);

Parameters

path	of a directory

Returns

a StrList containing file names

Retrieve file names in a directory (no metadata, non-recursive)

Parameters

path	directory path

Returns

StrList structure

Here is the call graph for this function:

mkdirs()

int mkdirs	(	const char *	_path,
		mode_t	mode )

Create all leafs in directory path

Parameters

_path	directory path to create
mode	mode_t permissions

Returns

path_basename()

char * path_basename

(

char *

path

)

Strip directory from file name Note: Caller is responsible for freeing memory

Parameters

_path

Returns

success=file name, failure=NULL

path_dirname()

char * path_dirname

(

char *

path

)

Return parent directory of file, or the parent of a directory

Parameters

path

Returns

success=directory, failure=empty string

path_store()

int path_store	(	char **	destptr,
		size_t	maxlen,
		const char *	base,
		const char *	path )

Helper function to initialize simple STASIS internal path strings

char *mypath = NULL;
 
if (path_store(&mypath, PATH_MAX, "/some", "path")) {
    fprintf(stderr, "Unable to allocate memory for path elements\n");
    exit(1);
}
// mypath is allocated to size PATH_MAX and contains the string: /some/path
// base+path will truncate at maxlen - 1

Parameters

destptr	address of destination string pointer
maxlen	maximum length of the path
base	path
path	to append to base

Returns

0 on success, -1 on error

Here is the call graph for this function:

popd()

int popd

(

void

)

Return from directory. Pop last path from directory stack.

See also

pushd

Returns

0 on success, -1 if stack is empty

Here is the call graph for this function:

pushd()

int pushd

(

const char *

path

)

Change directory. Push path on directory stack.

pushd("/somepath");
 
FILE fp = fopen("somefile", "w"); // i.e. /somepath/somefile
fprintf(fp, "Hello world.\n");
fclose(fp);
 
popd();

Parameters

path	of directory

Returns

0 on success, -1 on error

Here is the call graph for this function:

redact_sensitive()

int redact_sensitive	(	const char **	to_redact,
		size_t	to_redact_size,
		char *	src,
		char *	dest,
		size_t	maxlen )

Write REDACTED in dest for each occurrence of to_redacted token present in src

char command[PATH_MAX] = {0};
char command_redacted[PATH_MAX] = {0};
const char *password = "abc123";
const char *host = "myhostname";
const char *to_redact_case1[] = {password, host, NULL};
const char *to_redact_case2[] = {password, "--host", NULL};
const char *to_redact_case3[] = {password, "--host", host, NULL};
 
sprintf(command, "echo %s | program --host=%s -", password, host);
 
// CASE 1
redact_sensitive(to_redact_case1, command, command_redacted, sizeof(command_redacted) - 1);
printf("executing: %s\n", command_redacted);
// User sees:
// executing: echo ***REDACTED*** | program --host=***REDACTED*** -
system(command);
 
// CASE 2 remove an entire argument
redact_sensitive(to_redact_case2, command, command_redacted, sizeof(command_redacted) - 1);
printf("executing: %s\n", command_redacted);
// User sees:
// executing: echo ***REDACTED*** | program ***REDACTED*** -
system(command);
 
// CASE 3 remove it all (noisy)
redact_sensitive(to_redact_case3, command, command_redacted, sizeof(command_redacted) - 1);
printf("executing: %s\n", command_redacted);
// User sees:
// executing: echo ***REDACTED*** | program ***REDACTED***=***REDACTED*** -
system(command);

Parameters

to_redact	array of tokens to redact
src	input string
dest	output string
maxlen	maximum length of dest byte array

Returns

0 on success, -1 on error

Replace sensitive text in strings with REDACTED

Parameters

to_redact	a list of tokens to redact
to_redact_size	limit to n tokens in list
src	to read
dest	to write modified string
maxlen	maximum length of dest string

Returns

0 on success, -1 on error

Here is the call graph for this function:

rmtree()

int rmtree

(

char *

_path

)

Remove a directory tree recursively

mkdirs("a/b/c");
rmtree("a");
// a/b/c is removed

Parameters

_path

Returns

0 on success, -1 on error

Here is the call graph for this function:

touch()

int touch

(

const char *

filename

)

Create an empty file, or update modified timestamp on an existing file

Parameters

filename

file to touch

Returns

0 on success, 1 on error

xmkstemp()

char * xmkstemp	(	FILE **	fp,
		const char *	mode )

Creates a temporary file returning an open file pointer via fp, and the path to the file. The caller is responsible for closing fp and free()ing the returned file path.

FILE *fp = NULL;
char *tempfile = xmkstemp(&fp, "r+");
if (!fp || !tempfile) {
    fprintf(stderr, "Failed to generate temporary file for read/write\n");
    exit(1);
}

Parameters

fp	pointer to FILE (to be initialized)
mode	fopen() style file mode string

Returns

system path to the temporary file

NULL on failure

Here is the call graph for this function:

xml_pretty_print_in_place()

int xml_pretty_print_in_place	(	const char *	filename,
		const char *	pretty_print_prog,
		const char *	pretty_print_args )

Rewrite an XML file with a pretty printer command

Parameters

filename	path to modify
pretty_print_prog	program to call
pretty_print_args	arguments to pass to program

Returns

0 on success, -1 on error

Here is the call graph for this function: