diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/blogc-git-receiver/pre-receive-parser.c | 77 | ||||
-rw-r--r-- | src/blogc-git-receiver/pre-receive-parser.h | 14 | ||||
-rw-r--r-- | src/blogc-git-receiver/pre-receive.c | 91 | ||||
-rw-r--r-- | src/common/stdin.c | 23 | ||||
-rw-r--r-- | src/common/stdin.h | 14 |
5 files changed, 138 insertions, 81 deletions
diff --git a/src/blogc-git-receiver/pre-receive-parser.c b/src/blogc-git-receiver/pre-receive-parser.c new file mode 100644 index 0000000..a842a9d --- /dev/null +++ b/src/blogc-git-receiver/pre-receive-parser.c @@ -0,0 +1,77 @@ +/* + * blogc: A blog compiler. + * Copyright (C) 2015-2016 Rafael G. Martins <rafael@rafaelmartins.eng.br> + * + * This program can be distributed under the terms of the BSD License. + * See the file LICENSE. + */ + +#include <stdlib.h> +#include <string.h> +#include "../common/utils.h" +#include "pre-receive-parser.h" + +typedef enum { + START_OLD = 1, + OLD, + START_NEW, + NEW, + START_REF, + REF +} input_state_t; + + +char* +bgr_pre_receive_parse(const char *input) +{ + input_state_t state = START_OLD; + size_t start = 0; + size_t start_new = 0; + size_t input_len = strlen(input); + + for (size_t current = 0; current < input_len; current++) { + + char c = input[current]; + + switch (state) { + case START_OLD: + start = current; + state = OLD; + break; + case OLD: + if (c != ' ') + break; + // no need to store old + state = START_NEW; + break; + case START_NEW: + start = current; + state = NEW; + break; + case NEW: + if (c != ' ') + break; + state = START_REF; + start_new = start; + break; + case START_REF: + start = current; + state = REF; + break; + case REF: + if (c != '\n') + break; + state = START_OLD; + // we just care about a ref (refs/heads/master), everything + // else is disposable :) + if ((current - start == 17) && + (0 == strncmp("refs/heads/master", input + start, 17))) + { + return bc_strndup(input + start_new, start - 1 - start_new); + } + break; + } + } + + return NULL; +} diff --git a/src/blogc-git-receiver/pre-receive-parser.h b/src/blogc-git-receiver/pre-receive-parser.h new file mode 100644 index 0000000..3a605da --- /dev/null +++ b/src/blogc-git-receiver/pre-receive-parser.h @@ -0,0 +1,14 @@ +/* + * blogc: A blog compiler. + * Copyright (C) 2015-2016 Rafael G. Martins <rafael@rafaelmartins.eng.br> + * + * This program can be distributed under the terms of the BSD License. + * See the file LICENSE. + */ + +#ifndef _PRE_RECEIVE_PARSER_H +#define _PRE_RECEIVE_PARSER_H + +char* bgr_pre_receive_parse(const char *input); + +#endif /* _PRE_RECEIVE_PARSER_H */ diff --git a/src/blogc-git-receiver/pre-receive.c b/src/blogc-git-receiver/pre-receive.c index 8c35537..ca239da 100644 --- a/src/blogc-git-receiver/pre-receive.c +++ b/src/blogc-git-receiver/pre-receive.c @@ -17,12 +17,10 @@ #include <sys/stat.h> #include <dirent.h> #include <time.h> - #include "../common/utils.h" - -#ifndef BUFFER_SIZE -#define BUFFER_SIZE 4096 -#endif +#include "../common/stdin.h" +#include "pre-receive-parser.h" +#include "pre-receive.h" static unsigned int @@ -91,93 +89,26 @@ rmdir_recursive(const char *dir) } -typedef enum { - START_OLD = 1, - OLD, - START_NEW, - NEW, - START_REF, - REF -} input_state_t; - - int bgr_pre_receive_hook(int argc, char *argv[]) { - int c; - char buffer[BUFFER_SIZE]; - - input_state_t state = START_OLD; - size_t i = 0; - size_t start = 0; - int rv = 0; - char *new = NULL; - char *master = NULL; + char buffer[4096]; - while (EOF != (c = getc(stdin))) { - - buffer[i] = (char) c; - - switch (state) { - case START_OLD: - start = i; - state = OLD; - break; - case OLD: - if (c != ' ') - break; - // no need to store old - state = START_NEW; - break; - case START_NEW: - start = i; - state = NEW; - break; - case NEW: - if (c != ' ') - break; - state = START_REF; - new = strndup(buffer + start, i - start); - break; - case START_REF: - start = i; - state = REF; - break; - case REF: - if (c != '\n') - break; - state = START_OLD; - // we just care about a ref (refs/heads/master), everything - // else is disposable :) - if (!((i - start == 17) && - (0 == strncmp("refs/heads/master", buffer + start, 17)))) - { - free(new); - new = NULL; - break; - } - master = new; - break; - } - - if (++i >= BUFFER_SIZE) { - fprintf(stderr, "error: pre-receive hook payload is too big.\n"); - rv = 1; - goto cleanup2; - } - } + char *input = bc_stdin_read(); + char *master = bgr_pre_receive_parse(input); + free(input); if (master == NULL) { fprintf(stderr, "warning: no reference to master branch found. " "nothing to deploy.\n"); - goto cleanup2; + return rv; } char *repo_dir = NULL; char *output_dir = NULL; - if (NULL == getcwd(buffer, BUFFER_SIZE)) { + if (NULL == getcwd(buffer, sizeof(buffer))) { fprintf(stderr, "error: failed to get repository remote path: %s\n", strerror(errno)); rv = 1; @@ -247,7 +178,7 @@ bgr_pre_receive_hook(int argc, char *argv[]) } char *htdocs_sym = NULL; - ssize_t htdocs_sym_len = readlink("htdocs", buffer, BUFFER_SIZE); + ssize_t htdocs_sym_len = readlink("htdocs", buffer, sizeof(buffer)); if (0 < htdocs_sym_len) { if (0 != unlink("htdocs")) { fprintf(stderr, "error: failed to remove symlink (%s/htdocs): %s\n", @@ -275,7 +206,5 @@ cleanup: free(output_dir); rmdir_recursive(dir); free(repo_dir); -cleanup2: - free(new); return rv; } diff --git a/src/common/stdin.c b/src/common/stdin.c new file mode 100644 index 0000000..2664c3e --- /dev/null +++ b/src/common/stdin.c @@ -0,0 +1,23 @@ +/* + * blogc: A blog compiler. + * Copyright (C) 2015-2016 Rafael G. Martins <rafael@rafaelmartins.eng.br> + * + * This program can be distributed under the terms of the BSD License. + * See the file LICENSE. + */ + +#include <stdio.h> +#include "utils.h" +#include "stdin.h" + + +// splitted in single file to make it easier to test +char* +bc_stdin_read(void) +{ + int c; + bc_string_t *rv = bc_string_new(); + while (EOF != (c = fgetc(stdin))) + bc_string_append_c(rv, c); + return bc_string_free(rv, false); +} diff --git a/src/common/stdin.h b/src/common/stdin.h new file mode 100644 index 0000000..ebf042d --- /dev/null +++ b/src/common/stdin.h @@ -0,0 +1,14 @@ +/* + * blogc: A blog compiler. + * Copyright (C) 2015-2016 Rafael G. Martins <rafael@rafaelmartins.eng.br> + * + * This program can be distributed under the terms of the BSD License. + * See the file LICENSE. + */ + +#ifndef _STDIN_H +#define _STDIN_H + +char* bc_stdin_read(void); + +#endif /* _STDIN_H */ |