diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/main.c | 91 | ||||
| -rw-r--r-- | src/print.c | 94 |
2 files changed, 185 insertions, 0 deletions
diff --git a/src/main.c b/src/main.c new file mode 100644 index 0000000..631b3c2 --- /dev/null +++ b/src/main.c @@ -0,0 +1,91 @@ +// SPDX-License-Identifier: GPL-3.0-or-later +/* + * Copyright (C) 2024-Ω Miquel Sabaté Solà <mssola@mssola.com> + * + * 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 + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see <https://www.gnu.org/licenses/>. + */ + +#define _GNU_SOURCE + +#include <errno.h> +#include <sys/types.h> +#include <sys/stat.h> +#include <unistd.h> + +#include <pudc/print.h> +#include <pudc/pudc.h> +#include <pudc/compiler.h> + +/** + * Ensure that our current PID and TID are all 1, otherwise halt execution. + */ +void ensure_pid_tid_one(void) +{ + pid_t pid, tid; + + pid = getpid(); + if (pid != 1) { + pr_error(true, "'pudc' is supposed to be PID 1, but it is PID %d", pid); + } + + tid = gettid(); + if (tid != 1) { + pr_error(true, "'pudc' is supposed to be TID 1, but it is TID %d", tid); + } +} + +/** + * Ensure that the given @path is an existing directory. + * + * NOTE: this function will halt execution in case of error. + */ +void ensure_dir(const char *const path) +{ + struct stat st; + int status = stat(path, &st); + + if (!status) { + // Nothing to be done! + if (S_ISDIR(st.st_mode)) { + pr_debug("directory '%s' found", path); + return; + } + + // It exists but it's not a directory, bail out. + pr_error(true, "'%s' exists and it's not a directory", path); + } + + pr_error(true, "failed to stat '%s'", path); +} + +__noreturn int main(void) +{ + pr_info("starting pudc v%s", PUDC_VERSION); + + /* + * Early checks. + */ + ensure_pid_tid_one(); + ensure_dir("/run/pudc"); + + // TODO + halt(); + + /* + * We should never reach this point, and if we do just halt. The second + * 'halt' is just to make the compiler happy. + */ + pr_error(true, "pudc is no longer accepting requests. Halting execution..", "asd"); + halt(); +} diff --git a/src/print.c b/src/print.c new file mode 100644 index 0000000..364a20b --- /dev/null +++ b/src/print.c @@ -0,0 +1,94 @@ +// SPDX-License-Identifier: GPL-3.0-or-later +/* + * Copyright (C) 2024-Ω Miquel Sabaté Solà <mssola@mssola.com> + * + * 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 + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see <https://www.gnu.org/licenses/>. + */ + +#include <stdarg.h> +#include <stdbool.h> +#include <time.h> +#include <errno.h> +#include <string.h> + +#include <pudc/print.h> +#include <pudc/pudc.h> + +/** + * Prints the current local time into the given @stream. If fetching the current + * local time fails, then it does nothing. + */ +void pr_time(FILE *stream) +{ + time_t current_time = time(NULL); + struct tm *tm_info = NULL; + char buffer[32]; + + if (current_time != (time_t)-1) { + tm_info = localtime(¤t_time); + strftime(buffer, sizeof(buffer), "%Y-%m-%d %H:%M:%S", tm_info); + fprintf(stream, "[%s] ", buffer); + fflush(stream); + } +} + +void pr_out(bool should_halt, FILE *stream, enum print_type type, const char *const message, ...) +{ + va_list args; + + pr_time(stream); + + fprintf(stream, "pudc ("); + fflush(stream); + + switch (type) { + case debug: + fprintf(stream, "debug"); + break; + case error: + fprintf(stream, "error"); + break; + case info: + default: + fprintf(stream, "info"); + break; + }; + fflush(stream); + + fprintf(stream, "): "); + fflush(stream); + + va_start(args, message); + vfprintf(stream, message, args); + fflush(stream); + va_end(args); + + /* + * 'errno' is quite unreliable, so only deal with it if this was called via + * 'pr_error'. + */ + if (type == error) { + fprintf(stream, ": "); + fflush(stream); + perror(NULL); + fflush(stream); + } else { + fprintf(stream, ".\n"); + fflush(stream); + } + + if (should_halt) { + halt(); + } +} |