/* WARNING: This file was generated by dkct. Changes you make here will be lost if dkct is run again! You should modify the original source and run dkct on it. Original source: dkt-rnd.ctr */ /* Copyright (C) 2011-2013, Dirk Krause Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met: * Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer. * Redistributions in binary form must reproduce the above opyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution. * Neither the name of the author nor the names of contributors may be used to endorse or promote products derived from this software without specific prior written permission. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ /** @file dkt-rnd.c The dkt-rnd module. */ #line 10 "dkt-rnd.ctr" #include "dk3all.h" #include "dkt.h" #line 17 "dkt-rnd.ctr" /** Flag: Can continue. This is reset by the signal handlers. */ static #if DK3_HAVE_VOLATILE volatile #endif int dkt_rnd_can_continue = 1; /** Job structure for dkt erase-disk. */ typedef struct { dk3_app_t *app; /**< Application. */ dk3_option_set_t *opt; /**< Option set. */ dkChar const * const *msg; /**< Localized messages. */ dkChar const * const *kwnl; /**< Keywords, not localized. */ char *b; /**< Buffer. */ dk3_um_t nb; /**< Number of blocks. */ size_t bs; /**< Block size. */ int exval; /**< Exit status code. */ } DKT_RND_J; /** Data for the option set. */ static dk3_option_t const dkt_rnd_options[] = { { dkT('b'), dkT("block-size"), 1 }, { dkT('n'), dkT("number"), 1 } }; /** Number of options in the dkt_rnd_options array. */ static size_t const dkt_rnd_szoptions = sizeof(dkt_rnd_options)/sizeof(dk3_option_t); /** Signal handler. @param signo Signal number. */ static dk3_signal_ret_t dkt_rnd_sig_handler(int signo) { dk3signal_refresh(signo,dkt_rnd_int_handler); dkt_rnd_can_continue = 0; } /** Initialize job structure. @param j Job structure. */ static void dkt_rnd_job_init(DKT_RND_J *j) { dk3_um_t um = DK3_MAX_UINT_0; j->app = NULL; j->opt = NULL; j->msg = NULL; j->kwnl = NULL; j->exval = DKT_RESULT_ERR_UNSPECIFIC; um++; j->nb = um; j->bs = 512; } /** Clean up job structure. @param j Job structure. */ static void dkt_rnd_job_cleanup(DKT_RND_J *j) { if(j->opt) { dk3opt_close(j->opt); } j->opt = NULL; } /** Process command line arguments. @param j Job structure. @return 1 on success (can continue), 0 on error (abort). */ static int dkt_rnd_process_arguments(DKT_RND_J *j) { dk3_um_t um; int back = 0; int xargc = 0; /* Number of cmd line args. */ unsigned bs = 512; /* Block size. */ dkChar const *arg = NULL; /* Current argument. */ dkChar const * const *xargv = NULL; /* Cmd line args array. */ #line 117 "dkt-rnd.ctr" xargc = dk3app_get_argc(j->app); xargv = dk3app_get_argv(j->app); xargc--; xargc--; xargv++; xargv++; j->opt = dk3opt_open_app( dkt_rnd_options, dkt_rnd_szoptions, dkT('\0'), NULL, xargc, xargv, j->app ); if(j->opt) { if(0 == dk3opt_get_error_code(j->opt)) { back = 1; if(dk3opt_is_set(j->opt, dkT('b'))) { arg = dk3opt_get_short_arg(j->opt, dkT('b')); if(arg) { if(dk3sf_sscanf3(arg, dkT("%u"), &bs) == 1) { j->bs = (size_t)bs; } else { /* ERROR: Not numeric! */ dk3app_log_i3(j->app, DK3_LL_ERROR, 141, 142, arg); back = 0; j->exval = DKT_RESULT_ERR_OPTION; } } else { back = 0; j->exval = DKT_RESULT_ERR_OPTION; } } if(dk3opt_is_set(j->opt, dkT('n'))) { arg = dk3opt_get_short_arg(j->opt, dkT('n')); if(arg) { if(dk3ma_uintmax_from_string(&um, arg)) { j->nb = um; } else { /* ERROR: Not numeric! */ dk3app_log_i3(j->app, DK3_LL_ERROR, 141, 142, arg); back = 0; j->exval = DKT_RESULT_ERR_OPTION; } } else { back = 0; j->exval = DKT_RESULT_ERR_OPTION; } } } else { j->exval = DKT_RESULT_ERR_OPTION; } } else { j->exval = DKT_RESULT_ERR_OPTION; } #line 162 "dkt-rnd.ctr" return back; } /** Run with buffer set up. @param j Job structure. */ static void dkt_rnd_run_with_buffer(DKT_RND_J *j) { dk3_um_t blocks = DK3_MAX_UINT_0; #ifdef SIGINT dk3_signal_disp_t disp_int; #endif #ifdef SIGQUIT dk3_signal_disp_t disp_quit; #endif #ifdef SIGTERM dk3_signal_disp_t disp_term; #endif #ifdef SIGHUP dk3_signal_disp_t disp_hup; #endif #ifdef SIGPIPE dk3_signal_disp_t disp_pipe; #endif #if DK3_ON_WINDOWS int oldmode; #endif #line 194 "dkt-rnd.ctr" if(dk3app_rand_init(j->app, NULL)) { /* Set new signal dispositions, save old dispositions. */ #ifdef SIGINT disp_int = dk3signal_set(SIGINT, dkt_rnd_sig_handler); #endif #ifdef SIGQUIT disp_quit = dk3signal_set(SIGQUIT, dkt_rnd_sig_handler); #endif #ifdef SIGTERM disp_term = dk3signal_set(SIGTERM, dkt_rnd_sig_handler); #endif #ifdef SIGHUP disp_hup = dk3signal_set(SIGHUP, dkt_rnd_sig_handler); #endif #ifdef SIGPIPE disp_pipe = dk3signal_set(SIGPIPE, dkt_rnd_sig_handler); #endif /* Write buffer to standard output. */ #if DK3_ON_WINDOWS oldmode = _setmode(1, _O_BINARY); #endif do { #line 220 "dkt-rnd.ctr" if(dk3app_rand_bytes(j->app, (void *)(j->b), j->bs)) { (void)dk3sf_write_app(1, j->b, j->bs, j->app); blocks++; if(j->nb) { if(blocks >= j->nb) { dkt_rnd_can_continue = 0; #line 226 "dkt-rnd.ctr" } } } else { /* ERROR: Failed to obtain random bytes! */ dk3app_log_i1(j->app, DK3_LL_ERROR, 177); } } while(dkt_rnd_can_continue); #if DK3_ON_WINDOWS _setmode(1, oldmode); #endif /* Restore old signal dispositions. */ #ifdef SIGINT dk3signal_set(SIGINT, disp_int); #endif #ifdef SIGQUIT dk3signal_set(SIGQUIT, disp_quit); #endif #ifdef SIGTERM dk3signal_set(SIGTERM, disp_term); #endif #ifdef SIGHUP dk3signal_set(SIGHUP, disp_hup); #endif #ifdef SIGPIPE dk3signal_set(SIGPIPE, disp_pipe); #endif dk3app_rand_end(j->app); } #line 257 "dkt-rnd.ctr" } /** Run with built-in buffer. @param j Job structure. */ static void dkt_rnd_run_with_builtin_buffer(DKT_RND_J *j) { char bu[DKT_RND_BUFFER_SIZE]; #line 270 "dkt-rnd.ctr" j->b = bu; dkt_rnd_run_with_buffer(j); j->b = NULL; #line 273 "dkt-rnd.ctr" } /** Allocate buffer and run. @param j Job structure. */ static void dkt_rnd_run_with_allocated_buffer (DKT_RND_J *j) { char *bu = NULL; #line 286 "dkt-rnd.ctr" bu = dk3_new_app(char,j->bs,j->app); if(bu) { #line 288 "dkt-rnd.ctr" j->b = bu; dkt_rnd_run_with_buffer(j); j->b = NULL; dk3_delete(bu); } else { #line 293 "dkt-rnd.ctr" j->exval = DKT_RESULT_ERR_MEMORY; } #line 295 "dkt-rnd.ctr" } /** Produce the random output. @param j Job structure. */ static void dkt_rnd_run(DKT_RND_J *j) { #line 307 "dkt-rnd.ctr" if(j->bs > 0) { if(j->bs > DKT_RND_BUFFER_SIZE) { dkt_rnd_run_with_allocated_buffer(j); } else { dkt_rnd_run_with_builtin_buffer(j); } } else { /* ERROR: Invalid buffer size! */ dk3app_log_1(j->app, DK3_LL_ERROR, j->msg, 61); j->exval = DKT_RESULT_ERR_OPTION; } #line 318 "dkt-rnd.ctr" } int dkt_rnd( dk3_app_t *app, dkChar const *sn, dkChar const * const *msg, dkChar const * const *kwnl ) { int back = DKT_RESULT_ERR_UNSPECIFIC; DKT_RND_J j; #line 333 "dkt-rnd.ctr" dkt_rnd_job_init(&j); j.app = app; j.msg = msg; j.kwnl = kwnl; if(dkt_rnd_process_arguments(&j)) { #line 336 "dkt-rnd.ctr" j.exval = DKT_RESULT_OK; dkt_rnd_run(&j); } back = j.exval; dkt_rnd_job_cleanup(&j); #line 342 "dkt-rnd.ctr" return back; } /* vim: set ai sw=2 : */