diff options
| -rw-r--r-- | include/proto_stdio.h | 30 | ||||
| -rw-r--r-- | src/daemon/meson.build | 2 | ||||
| -rw-r--r-- | src/daemon/proto_stdio.c | 151 | ||||
| -rw-r--r-- | src/daemon/proto_stdio_private.h | 47 |
4 files changed, 230 insertions, 0 deletions
diff --git a/include/proto_stdio.h b/include/proto_stdio.h new file mode 100644 index 0000000..c54ef9c --- /dev/null +++ b/include/proto_stdio.h @@ -0,0 +1,30 @@ +/* + * Usurpation – null (stdio) im proto + * + * Copyright (C) 2019 Gediminas Jakutis + * Copyright (C) 2019 Paulius Ratkevičius + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; version 2.1 + * of the License. + * + * 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 + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA + */ + +#ifndef USURPATION_PROTO_STDIO_H +#define USURPATION_PROTO_STDIO_H + +int proto_stdio_init(void); +void proto_stdio_close(void); +void message_receive(char *); +char *message_send(void); + +#endif /* USURPATION_PROTO_STDIO_H */ diff --git a/src/daemon/meson.build b/src/daemon/meson.build index aa0aa75..d900949 100644 --- a/src/daemon/meson.build +++ b/src/daemon/meson.build @@ -3,8 +3,10 @@ d_filenames = [ 'settings.c', 'net.c', 'purple.c', + 'proto_stdio.c', 'settings_private.h', 'purple_private.h', + 'proto_stdio_private.h', ] d_conf_filenames = [ diff --git a/src/daemon/proto_stdio.c b/src/daemon/proto_stdio.c new file mode 100644 index 0000000..f0a22a6 --- /dev/null +++ b/src/daemon/proto_stdio.c @@ -0,0 +1,151 @@ +/* + * Usurpation – null (stdio) im proto + * + * Copyright (C) 2019 Gediminas Jakutis + * Copyright (C) 2019 Paulius Ratkevičius + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; version 2.1 + * of the License. + * + * 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 + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA + */ + +#include <errno.h> +#include "proto_stdio.h" +#include "proto_stdio_private.h" + + +void message_receive(char *arg) +{ + int cancelstate; + int done = 0; + + while (!done) { + pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &cancelstate); + pthread_mutex_lock(&state.out_m); + if (!state.writebuf) { + state.writebuf = strdup(arg); + done = 1; + } + pthread_mutex_unlock(&state.out_m); + pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, &cancelstate); + pthread_testcancel(); + } +} + +char *message_send(void) +{ + char *ret; + int cancelstate; + + pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &cancelstate); + pthread_mutex_lock(&state.out_m); + ret = strdup(state.readbuf); + free(state.readbuf); + state.readbuf = NULL; + state.readbufsize = 0; + pthread_mutex_unlock(&state.out_m); + pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, &cancelstate); + pthread_testcancel(); + + return ret; +} + +static void *read_stdin(void *arg) +{ + int cancelstate; + + (void) arg; + + while(1) { + pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &cancelstate); + pthread_mutex_lock(&state.in_m); + + if (!state.readbuf) { + state.readbufsize = getline(&state.readbuf, NULL, stdin); + } + + pthread_mutex_unlock(&state.in_m); + pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, &cancelstate); + pthread_testcancel(); + nanosleep(&respite, NULL); + } + + return NULL; +} + + +static void *write_stdout(void *arg) +{ + int cancelstate; + + (void) arg; + + while(1) { + pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &cancelstate); + pthread_mutex_lock(&state.out_m); + + if (state.writebuf) { + if (state.writebuf[strlen(state.writebuf) - 1] == '\n') { + printf("%s", state.writebuf); + } else { + printf("%s\n", state.writebuf); + } + + free(state.writebuf); + state.writebuf = NULL; + state.writebufsize = 0; + } + + pthread_mutex_unlock(&state.out_m); + pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, &cancelstate); + pthread_testcancel(); + nanosleep(&respite, NULL); + } + + return NULL; +} + + +int proto_stdio_init(void) +{ + + int ret = 0; + + /* aquire the """singleton""" mutex */ + if (pthread_mutex_trylock(&state.mutex)) { + /* TODO: use proper error numbers */ + ret = 1; + } else { + if ((ret = pthread_create(&state.stdio_in, NULL, read_stdin, NULL))) { + proto_stdio_close(); + } + + if ((ret = pthread_create(&state.stdio_out, NULL, write_stdout, NULL))) { + proto_stdio_close(); + } + } + + return ret; +} + +void proto_stdio_close(void) +{ + if (pthread_mutex_trylock(&state.mutex) == EBUSY) { + pthread_cancel(state.stdio_in); + pthread_cancel(state.stdio_out); + pthread_join(state.stdio_in, NULL); + pthread_join(state.stdio_out, NULL); + } + + pthread_mutex_unlock(&state.mutex); +} diff --git a/src/daemon/proto_stdio_private.h b/src/daemon/proto_stdio_private.h new file mode 100644 index 0000000..209d191 --- /dev/null +++ b/src/daemon/proto_stdio_private.h @@ -0,0 +1,47 @@ +/* + * Usurpation – null (stdio) im proto + * + * Copyright (C) 2019 Gediminas Jakutis + * Copyright (C) 2019 Paulius Ratkevičius + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; version 2.1 + * of the License. + * + * 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 + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA + */ + +#ifndef USURPATION_PROTO_STDIO_PRIVATE_H +#define USURPATION_PROTO_STDIO_PRIVATE_H + +#include <string.h> +#include <stdlib.h> +#include <stdio.h> +#include <sys/types.h> +#include <pthread.h> + +static struct state { + pthread_mutex_t mutex; + pthread_t stdio_in; + pthread_t stdio_out; + pthread_mutex_t in_m; + pthread_mutex_t out_m; + char *readbuf; + char *writebuf; + ssize_t readbufsize; + ssize_t writebufsize; +} state = {PTHREAD_MUTEX_INITIALIZER, 0, 0, PTHREAD_MUTEX_INITIALIZER, PTHREAD_MUTEX_INITIALIZER, NULL, NULL, 0, 0}; + +static void *read_stdin(void *arg); +static void *write_stdout(void *arg); + +static const struct timespec respite = {0, 10 * 1000 * 1000}; /* 10ms */ + +#endif /* USURPATION_PROTO_STDIO_PRIVATE_H */ |