diff options
Diffstat (limited to 'src/driver')
| -rw-r--r-- | src/driver/ptyc_amain.c | 74 | ||||
| -rw-r--r-- | src/driver/ptyc_driver_ctx.c | 209 |
2 files changed, 283 insertions, 0 deletions
diff --git a/src/driver/ptyc_amain.c b/src/driver/ptyc_amain.c new file mode 100644 index 0000000..4762d3a --- /dev/null +++ b/src/driver/ptyc_amain.c @@ -0,0 +1,74 @@ +/*********************************************************/ +/* ptycon: a pty-console bridge */ +/* Copyright (C) 2016 Z. Gilboa */ +/* Released under GPLv2 and GPLv3; see COPYING.PTYCON. */ +/*********************************************************/ + +#include <ntapi/ntapi.h> +#include <stdio.h> +#include <unistd.h> + +#include <ptycon/ptycon.h> +#include "ptycon_init_impl.h" +#include "ptycon_driver_impl.h" +#include "ptycon_nolibc_impl.h" + +#ifndef PTYC_DRIVER_FLAGS +#define PTYC_DRIVER_FLAGS PTYC_DRIVER_VERBOSITY_ERRORS \ + | PTYC_DRIVER_VERBOSITY_USAGE +#endif + +static const char vermsg[] = "%s%s%s (git://midipix.org/ptycon): " + "version %s%d.%d.%d%s.\n" + "[commit reference: %s%s%s]\n"; + +static const char * const ptyc_ver_color[6] = { + "\x1b[1m\x1b[35m","\x1b[0m", + "\x1b[1m\x1b[32m","\x1b[0m", + "\x1b[1m\x1b[34m","\x1b[0m" +}; + +static const char * const ptyc_ver_plain[6] = { + "","", + "","", + "","" +}; + +static ssize_t ptyc_version(struct ptyc_driver_ctx * dctx) +{ + const struct ptyc_source_version * verinfo; + const char * const * verclr; + + verinfo = ptyc_source_version(); + verclr = isatty(STDOUT_FILENO) ? ptyc_ver_color : ptyc_ver_plain; + + return fprintf(stdout,vermsg, + verclr[0],dctx->program,verclr[1], + verclr[2],verinfo->major,verinfo->minor, + verinfo->revision,verclr[3], + verclr[4],verinfo->commit,verclr[5]); +} + +static int ptyc_exit(struct ptyc_driver_ctx * dctx, int nerrors) +{ + ptyc_free_driver_ctx(dctx); + return nerrors ? 2 : 0; +} + +int ptyc_main(int argc, char ** argv, char ** envp) +{ + int ret; + struct ptyc_driver_ctx * dctx; + + if ((ret = ptyc_init())) + return ret; + + if ((ret = ptyc_get_driver_ctx(argv,envp,PTYC_DRIVER_FLAGS,&dctx))) + return (ret == PTYC_USAGE) ? !--argc : 2; + + if (dctx->cctx->drvflags & PTYC_DRIVER_VERSION) + if ((ptyc_version(dctx)) < 0) + return ptyc_exit(dctx,2); + + return ptyc_exit(dctx,ret); +} diff --git a/src/driver/ptyc_driver_ctx.c b/src/driver/ptyc_driver_ctx.c new file mode 100644 index 0000000..cc50e36 --- /dev/null +++ b/src/driver/ptyc_driver_ctx.c @@ -0,0 +1,209 @@ +/*********************************************************/ +/* ptycon: a pty-console bridge */ +/* Copyright (C) 2016 Z. Gilboa */ +/* Released under GPLv2 and GPLv3; see COPYING.PTYCON. */ +/*********************************************************/ + +#include <ntapi/ntapi.h> +#include <stdint.h> + +#include <ptycon/ptycon.h> +#include "ptycon_init_impl.h" +#include "ptycon_nolibc_impl.h" + +#define ARGV_DRIVER + +#include "ptycon_version.h" +#include "ptycon_driver_impl.h" +#include "argv/argv.h" + +/* ntapi accessor table */ +const ntapi_vtbl * ptyc_ntapi; + +/* package info */ +static const struct ptyc_source_version ptyc_src_version = { + PTYC_TAG_VER_MAJOR, + PTYC_TAG_VER_MINOR, + PTYC_TAG_VER_PATCH, + PTYCON_GIT_VERSION +}; + +struct ptyc_driver_ctx_alloc { + struct argv_meta * meta; + struct ptyc_driver_ctx_impl ctx; + uint64_t guard; + const char * units[]; +}; + +static uint32_t ptyc_argv_flags(uint32_t flags) +{ + uint32_t ret = 0; + + if (flags & PTYC_DRIVER_VERBOSITY_NONE) + ret |= ARGV_VERBOSITY_NONE; + + if (flags & PTYC_DRIVER_VERBOSITY_ERRORS) + ret |= ARGV_VERBOSITY_ERRORS; + + if (flags & PTYC_DRIVER_VERBOSITY_STATUS) + ret |= ARGV_VERBOSITY_STATUS; + + return ret; +} + +static int ptyc_driver_usage( + const char * program, + const char * arg, + const struct argv_option * options, + struct argv_meta * meta) +{ + char header[512]; + + snprintf(header,sizeof(header), + "Usage: %s [options] <file>...\n" "Options:\n", + program); + + argv_usage(stdout,header,options,arg); + argv_free(meta); + + return PTYC_USAGE; +} + +static struct ptyc_driver_ctx_impl * ptyc_driver_ctx_alloc( + struct argv_meta * meta, + const struct ptyc_common_ctx * cctx, + size_t nunits) +{ + struct ptyc_driver_ctx_alloc * ictx; + size_t size; + struct argv_entry * entry; + const char ** units; + + size = sizeof(struct ptyc_driver_ctx_alloc); + size += (nunits+1)*sizeof(const char *); + + if (!(ictx = calloc(1,size))) + return 0; + + if (cctx) + memcpy(&ictx->ctx.cctx,cctx,sizeof(*cctx)); + + for (entry=meta->entries,units=ictx->units; entry->fopt || entry->arg; entry++) + if (!entry->fopt) + *units++ = entry->arg; + + ictx->meta = meta; + ictx->ctx.ctx.units = ictx->units; + return &ictx->ctx; +} + +static int ptyc_get_driver_ctx_fail(struct argv_meta * meta) +{ + argv_free(meta); + return -1; +} + +int ptyc_get_driver_ctx( + char ** argv, + char ** envp, + uint32_t flags, + struct ptyc_driver_ctx ** pctx) +{ + struct ptyc_driver_ctx_impl * ctx; + struct ptyc_common_ctx cctx; + const struct argv_option * options; + struct argv_meta * meta; + struct argv_entry * entry; + size_t nunits; + const char * program; + + (void)envp; + + if (ptyc_init()) + return -1; + + options = ptyc_default_options; + + if (!(meta = argv_get(argv,options,ptyc_argv_flags(flags)))) + return -1; + + nunits = 0; + program = argv_program_name(argv[0]); + memset(&cctx,0,sizeof(cctx)); + cctx.drvflags = flags; + + if (!argv[1] && (flags & PTYC_DRIVER_VERBOSITY_USAGE)) + return ptyc_driver_usage(program,0,options,meta); + + /* get options, count units */ + for (entry=meta->entries; entry->fopt || entry->arg; entry++) { + if (entry->fopt) { + switch (entry->tag) { + case TAG_HELP: + if (flags & PTYC_DRIVER_VERBOSITY_USAGE) + return ptyc_driver_usage(program,entry->arg,options,meta); + + case TAG_VERSION: + cctx.drvflags |= PTYC_DRIVER_VERSION; + break; + } + } else + nunits++; + } + + if (!(ctx = ptyc_driver_ctx_alloc(meta,&cctx,nunits))) + return ptyc_get_driver_ctx_fail(meta); + + ctx->ctx.program = program; + ctx->ctx.cctx = &ctx->cctx; + + *pctx = &ctx->ctx; + return PTYC_OK; +} + +int ptyc_create_driver_ctx( + const struct ptyc_common_ctx * cctx, + struct ptyc_driver_ctx ** pctx) +{ + struct argv_meta * meta; + struct ptyc_driver_ctx_impl * ctx; + char * argv[] = {"ptycon_driver",0}; + + if (ptyc_init()) + return -1; + + if (!(meta = argv_get(argv,ptyc_default_options,0))) + return -1; + + if (!(ctx = ptyc_driver_ctx_alloc(meta,cctx,0))) + return ptyc_get_driver_ctx_fail(0); + + ctx->ctx.cctx = &ctx->cctx; + memcpy(&ctx->cctx,cctx,sizeof(*cctx)); + *pctx = &ctx->ctx; + return PTYC_OK; +} + +static void ptyc_free_driver_ctx_impl(struct ptyc_driver_ctx_alloc * ictx) +{ + argv_free(ictx->meta); + free(ictx); +} + +void ptyc_free_driver_ctx(struct ptyc_driver_ctx * ctx) +{ + struct ptyc_driver_ctx_alloc * ictx; + uintptr_t addr; + + if (ctx) { + addr = (uintptr_t)ctx - offsetof(struct ptyc_driver_ctx_alloc,ctx); + addr = addr - offsetof(struct ptyc_driver_ctx_impl,ctx); + ictx = (struct ptyc_driver_ctx_alloc *)addr; + ptyc_free_driver_ctx_impl(ictx); + } +} + +const struct ptyc_source_version * ptyc_source_version(void) +{ + return &ptyc_src_version; +} |