From f4603994a39431c8cee036216a6d15ab7281b0c6 Mon Sep 17 00:00:00 2001 From: midipix Date: Mon, 14 Dec 2015 01:58:21 -0500 Subject: initial commit. --- src/driver/sfrt_driver_ctx.c | 169 +++++++++++++++++++++++++++++++++++++++++++ src/driver/sfrt_unit_ctx.c | 56 ++++++++++++++ 2 files changed, 225 insertions(+) create mode 100644 src/driver/sfrt_driver_ctx.c create mode 100644 src/driver/sfrt_unit_ctx.c (limited to 'src/driver') diff --git a/src/driver/sfrt_driver_ctx.c b/src/driver/sfrt_driver_ctx.c new file mode 100644 index 0000000..75afd0c --- /dev/null +++ b/src/driver/sfrt_driver_ctx.c @@ -0,0 +1,169 @@ +#include +#include +#include + +#define ARGV_DRIVER + +#include +#include "sofort_driver_impl.h" +#include "argv/argv.h" + +extern const struct argv_option sfrt_default_options[]; + +struct sfrt_driver_ctx_alloc { + struct argv_meta * meta; + struct sfrt_driver_ctx_impl ctx; + uint64_t guard; + const char * units[]; +}; + +static uint32_t sfrt_argv_flags(uint32_t flags) +{ + uint32_t ret = 0; + + if (flags & SFRT_DRIVER_VERBOSITY_NONE) + ret |= ARGV_VERBOSITY_NONE; + + if (flags & SFRT_DRIVER_VERBOSITY_ERRORS) + ret |= ARGV_VERBOSITY_ERRORS; + + if (flags & SFRT_DRIVER_VERBOSITY_STATUS) + ret |= ARGV_VERBOSITY_STATUS; + + return ret; +} + +static int sfrt_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] ...\n" "Options:\n", + program); + + argv_usage(stdout,header,options,arg); + argv_free(meta); + + return SFRT_USAGE; +} + +static struct sfrt_driver_ctx_impl * sfrt_driver_ctx_alloc(struct argv_meta * meta, size_t nunits) +{ + struct sfrt_driver_ctx_alloc * ictx; + size_t size; + struct argv_entry * entry; + const char ** units; + + size = sizeof(struct sfrt_driver_ctx_alloc); + size += (nunits+1)*sizeof(const char *); + + if (!(ictx = calloc(size,1))) + return 0; + + for (entry=meta->entries,units=ictx->units; entry->fopt || entry->arg; entry++) + if (!entry->fopt) + *units++ = entry->arg; + + ictx->ctx.ctx.units = ictx->units; + return &ictx->ctx; +} + +int sfrt_get_driver_ctx_fail(struct argv_meta * meta) +{ + argv_free(meta); + return -1; +} + +int sfrt_get_driver_ctx( + const char ** argv, + const char ** envp, + uint32_t flags, + struct sfrt_driver_ctx ** pctx) +{ + struct sfrt_driver_ctx_impl * ctx; + const struct argv_option * options; + struct argv_meta * meta; + struct argv_entry * entry; + size_t nunits; + uint64_t dflags; + uint64_t aflags; + const char * program; + const char * astring; + + options = sfrt_default_options; + + if (!(meta = argv_get(argv,options,sfrt_argv_flags(flags)))) + return -1; + + dflags = 0; + aflags = 0; + nunits = 0; + astring = 0; + program = argv_program_name(argv[0]); + + if (!argv[1] && (flags & SFRT_DRIVER_VERBOSITY_USAGE)) + return sfrt_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 & SFRT_DRIVER_VERBOSITY_USAGE) + return sfrt_driver_usage(program,entry->arg,options,meta); + + case TAG_VERSION: + dflags |= SFRT_DRIVER_VERSION; + break; + + case TAG_OUTPUT_DUMMY: + astring = entry->arg; + break; + + case TAG_OUTPUT_PROPERTY: + if (!(strcmp(entry->arg,"name"))) + aflags |= SFRT_OUTPUT_NAME; + else if (!(strcmp(entry->arg,"address"))) + aflags |= SFRT_OUTPUT_ADDRESS; + break; + } + } else + nunits++; + } + + if (!(ctx = sfrt_driver_ctx_alloc(meta,nunits))) + return sfrt_get_driver_ctx_fail(meta); + + ctx->ctx.program = program; + ctx->cctx.drvflags = dflags; + ctx->cctx.actflags = aflags; + ctx->cctx.anystring = astring; + + ctx->ctx.cctx = &ctx->cctx; + + *pctx = &ctx->ctx; + return SFRT_OK; +} + +static void sfrt_free_driver_ctx_impl(struct sfrt_driver_ctx_alloc * ictx) +{ + argv_free(ictx->meta); + free(ictx); +} + +void sfrt_free_driver_ctx(struct sfrt_driver_ctx * ctx) +{ + struct sfrt_driver_ctx_alloc * ictx; + uintptr_t addr; + + if (ctx) { + addr = (uintptr_t)ctx - offsetof(struct sfrt_driver_ctx_alloc,ctx); + addr = addr - offsetof(struct sfrt_driver_ctx_impl,ctx); + ictx = (struct sfrt_driver_ctx_alloc *)addr; + sfrt_free_driver_ctx_impl(ictx); + } +} diff --git a/src/driver/sfrt_unit_ctx.c b/src/driver/sfrt_unit_ctx.c new file mode 100644 index 0000000..cef9913 --- /dev/null +++ b/src/driver/sfrt_unit_ctx.c @@ -0,0 +1,56 @@ +#include +#include +#include +#include +#include + +#include +#include "sofort_driver_impl.h" + +static int sfrt_free_unit_ctx_impl(struct sfrt_unit_ctx_impl * ctx, int status) +{ + if (ctx) { + sfrt_unmap_input(&ctx->map); + free(ctx); + } + + return status; +} + +int sfrt_get_unit_ctx( + const struct sfrt_driver_ctx * dctx, + const char * path, + struct sfrt_unit_ctx ** pctx) +{ + struct sfrt_unit_ctx_impl * ctx; + + if (!dctx || !(ctx = calloc(sizeof(*ctx),1))) + return -1; + + if (sfrt_map_input(-1,path,PROT_READ,&ctx->map)) + return sfrt_free_unit_ctx_impl(ctx,-1); + + memcpy(&ctx->cctx,dctx->cctx, + sizeof(ctx->cctx)); + + ctx->path = path; + + ctx->uctx.path = &ctx->path; + ctx->uctx.map = &ctx->map; + ctx->uctx.cctx = &ctx->cctx; + + *pctx = &ctx->uctx; + return 0; +} + +void sfrt_free_unit_ctx(struct sfrt_unit_ctx * ctx) +{ + struct sfrt_unit_ctx_impl * ictx; + uintptr_t addr; + + if (ctx) { + addr = (uintptr_t)ctx - offsetof(struct sfrt_unit_ctx_impl,uctx); + ictx = (struct sfrt_unit_ctx_impl *)addr; + sfrt_free_unit_ctx_impl(ictx,0); + } +} -- cgit v1.2.3