Annotation of embedaddon/libpdel/tmpl/test/main.c, revision 1.1
1.1 ! misho 1:
! 2: /*
! 3: * Copyright (c) 2001-2002 Packet Design, LLC.
! 4: * All rights reserved.
! 5: *
! 6: * Subject to the following obligations and disclaimer of warranty,
! 7: * use and redistribution of this software, in source or object code
! 8: * forms, with or without modifications are expressly permitted by
! 9: * Packet Design; provided, however, that:
! 10: *
! 11: * (i) Any and all reproductions of the source or object code
! 12: * must include the copyright notice above and the following
! 13: * disclaimer of warranties; and
! 14: * (ii) No rights are granted, in any manner or form, to use
! 15: * Packet Design trademarks, including the mark "PACKET DESIGN"
! 16: * on advertising, endorsements, or otherwise except as such
! 17: * appears in the above copyright notice or in the software.
! 18: *
! 19: * THIS SOFTWARE IS BEING PROVIDED BY PACKET DESIGN "AS IS", AND
! 20: * TO THE MAXIMUM EXTENT PERMITTED BY LAW, PACKET DESIGN MAKES NO
! 21: * REPRESENTATIONS OR WARRANTIES, EXPRESS OR IMPLIED, REGARDING
! 22: * THIS SOFTWARE, INCLUDING WITHOUT LIMITATION, ANY AND ALL IMPLIED
! 23: * WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE,
! 24: * OR NON-INFRINGEMENT. PACKET DESIGN DOES NOT WARRANT, GUARANTEE,
! 25: * OR MAKE ANY REPRESENTATIONS REGARDING THE USE OF, OR THE RESULTS
! 26: * OF THE USE OF THIS SOFTWARE IN TERMS OF ITS CORRECTNESS, ACCURACY,
! 27: * RELIABILITY OR OTHERWISE. IN NO EVENT SHALL PACKET DESIGN BE
! 28: * LIABLE FOR ANY DAMAGES RESULTING FROM OR ARISING OUT OF ANY USE
! 29: * OF THIS SOFTWARE, INCLUDING WITHOUT LIMITATION, ANY DIRECT,
! 30: * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, PUNITIVE, OR CONSEQUENTIAL
! 31: * DAMAGES, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES, LOSS OF
! 32: * USE, DATA OR PROFITS, HOWEVER CAUSED AND UNDER ANY THEORY OF
! 33: * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
! 34: * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
! 35: * THE USE OF THIS SOFTWARE, EVEN IF PACKET DESIGN IS ADVISED OF
! 36: * THE POSSIBILITY OF SUCH DAMAGE.
! 37: *
! 38: * Author: Archie Cobbs <archie@freebsd.org>
! 39: */
! 40:
! 41: #include <stdio.h>
! 42: #include <stdlib.h>
! 43: #include <unistd.h>
! 44: #include <string.h>
! 45: #include <signal.h>
! 46: #include <stdarg.h>
! 47: #include <termios.h>
! 48: #include <limits.h>
! 49: #include <errno.h>
! 50: #include <err.h>
! 51:
! 52: #include <pdel/structs/structs.h>
! 53: #include <pdel/structs/type/array.h>
! 54: #include <pdel/tmpl/tmpl.h>
! 55: #include <pdel/util/typed_mem.h>
! 56:
! 57: #define TMPL_MEM_TYPE "tmpl_test.tmpl"
! 58: #define CTX_MEM_TYPE "tmpl_test.ctx"
! 59:
! 60: #define EOF_STRING "<<EOF>>"
! 61:
! 62: #ifndef TCSASOFT
! 63: #define TCSASOFT 0
! 64: #endif
! 65:
! 66: struct private_info {
! 67: FILE *input;
! 68: };
! 69:
! 70: /*
! 71: * Internal functions
! 72: */
! 73: static tmpl_handler_t readfile_func;
! 74: static tmpl_handler_t display_func;
! 75: static tmpl_handler_t error_func;
! 76: static tmpl_handler_t input_func;
! 77:
! 78: static tmpl_handler_t handler;
! 79: static tmpl_errfmtr_t errfmtr;
! 80:
! 81: /*
! 82: * Internal variables
! 83: */
! 84: static struct tmpl_func function_list[] = {
! 85: { "display", 0, INT_MAX, display_func },
! 86: { "error_test", 0, INT_MAX, error_func },
! 87: { "func1", 0, INT_MAX, display_func },
! 88: { "func2", 0, INT_MAX, display_func },
! 89: { "func3", 0, INT_MAX, display_func },
! 90: { "input", 0, 1, input_func },
! 91: { "readfile", 1, 1, readfile_func },
! 92: };
! 93:
! 94: int
! 95: main(int ac, char **av)
! 96: {
! 97: int flags = TMPL_SKIP_NL_WHITE;
! 98: struct private_info priv;
! 99: struct tmpl_ctx *ctx;
! 100: struct tmpl *tmpl;
! 101: int show_stats = 0;
! 102: int num_errors;
! 103: char *errmsg;
! 104: FILE *fp;
! 105: int ch;
! 106:
! 107: /* Parse command line arguments */
! 108: while ((ch = getopt(ac, av, "ns")) != -1) {
! 109: switch (ch) {
! 110: case 'n':
! 111: flags &= ~TMPL_SKIP_NL_WHITE;
! 112: break;
! 113: case 's':
! 114: show_stats = 1;
! 115: break;
! 116: default:
! 117: goto usage;
! 118: }
! 119: }
! 120: ac -= optind;
! 121: av += optind;
! 122:
! 123: if (ac < 1)
! 124: usage: errx(1, "usage: tmpl [-n] filename [ func arg1 arg2 ...]");
! 125:
! 126: /* Enable typed memory */
! 127: if (typed_mem_enable() == -1)
! 128: err(1, "typed_mem_enable");
! 129:
! 130: /* Read input file to create template */
! 131: if ((fp = fopen(av[0], "r")) == NULL)
! 132: err(1, "%s", av[0]);
! 133: if ((tmpl = tmpl_create(fp, &num_errors, TMPL_MEM_TYPE)) == NULL)
! 134: err(1, "tmpl_create: %s", av[0]);
! 135: fclose(fp);
! 136: if (num_errors != 0)
! 137: fprintf(stderr, "There were %d parse errors.\n", num_errors);
! 138:
! 139: /* Create template context */
! 140: memset(&priv, 0, sizeof(priv));
! 141: priv.input = stdin;
! 142: if ((ctx = tmpl_ctx_create(&priv,
! 143: CTX_MEM_TYPE, handler, errfmtr)) == NULL)
! 144: err(1, "tmpl_ctx_create");
! 145:
! 146: /* Execute template */
! 147: if (tmpl_execute(tmpl, ctx, stdout, flags) == -1)
! 148: err(1, "tmpl_execute");
! 149:
! 150: /* Execute user defined function */
! 151: ac--;
! 152: av++;
! 153: if (ac > 0 && tmpl_execute_func(ctx,
! 154: stdout, &errmsg, ac, av, 0) == -1) {
! 155: errx(1, "tmpl_execute_func: %s",
! 156: errmsg ? errmsg : strerror(errno));
! 157: }
! 158:
! 159: /* Free template and context */
! 160: tmpl_ctx_destroy(&ctx);
! 161: tmpl_destroy(&tmpl);
! 162:
! 163: /* Done */
! 164: if (show_stats)
! 165: typed_mem_dump(stderr);
! 166: return (0);
! 167: }
! 168:
! 169: static char *
! 170: handler(struct tmpl_ctx *ctx, char **errmsgp, int ac, char **av)
! 171: {
! 172: return (tmpl_list_handler(ctx, function_list,
! 173: sizeof(function_list) / sizeof(*function_list), errmsgp, ac, av));
! 174: }
! 175:
! 176: static char *
! 177: errfmtr(struct tmpl_ctx *ctx, const char *errmsg)
! 178: {
! 179: const char *const mtype = tmpl_ctx_get_mtype(ctx);
! 180: char *string;
! 181:
! 182: ASPRINTF(mtype, &string, "[ ERROR: %s ]", errmsg);
! 183: return (string);
! 184: }
! 185:
! 186: static char *
! 187: error_func(struct tmpl_ctx *ctx, char **errmsgp, int ac, char **av)
! 188: {
! 189: const char *const mtype = tmpl_ctx_get_mtype(ctx);
! 190:
! 191: *errmsgp = STRDUP(mtype, "Sample error message here");
! 192: return (NULL);
! 193: }
! 194:
! 195: static char *
! 196: display_func(struct tmpl_ctx *ctx, char **errmsgp, int ac, char **av)
! 197: {
! 198: const char *const mtype = tmpl_ctx_get_mtype(ctx);
! 199: char buf[1024];
! 200: int i;
! 201:
! 202: /* Normal handler output */
! 203: snprintf(buf, sizeof(buf), "[ %s(", av[0]);
! 204: for (i = 1; i < ac; i++) {
! 205: snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf),
! 206: "%s%s", av[i], (i == ac - 1) ? "" : ", ");
! 207: }
! 208: snprintf(buf + strlen(buf), sizeof(buf) - strlen(buf), ") ]");
! 209: return (STRDUP(mtype, buf));
! 210: }
! 211:
! 212: static char *
! 213: readfile_func(struct tmpl_ctx *ctx, char **errmsgp, int ac, char **av)
! 214: {
! 215: const char *const mtype = tmpl_ctx_get_mtype(ctx);
! 216: int errno_save;
! 217: char *s, *t;
! 218: FILE *fp;
! 219: int slen;
! 220: int nr;
! 221:
! 222: /* Read file into memory */
! 223: if ((fp = fopen(av[1], "r")) == NULL)
! 224: return (NULL);
! 225: for (s = NULL, slen = 0; 1; slen += nr) {
! 226: if ((t = REALLOC(mtype, s, slen + 1024)) == NULL) {
! 227: errno_save = errno;
! 228: FREE(mtype, s);
! 229: s = NULL;
! 230: errno = errno_save;
! 231: break;
! 232: }
! 233: s = t;
! 234: if ((nr = fread(s + slen, 1, 1024, fp)) == 0) {
! 235: s[slen] = '\0';
! 236: break;
! 237: }
! 238: }
! 239:
! 240: /* Close file and return string */
! 241: errno_save = errno;
! 242: fclose(fp);
! 243: errno = errno_save;
! 244: return (s);
! 245: }
! 246:
! 247: /*
! 248: * @input()
! 249: *
! 250: * Reads one line of input from the input stream and returns it.
! 251: * If an optional first argument is supplied and its numeric value
! 252: * is non-zero, then the input is assumed to be a terminal device
! 253: * and echo will be disabled (useful for inputting passwords).
! 254: *
! 255: * If no input stream was provided, or EOF is detected, an empty
! 256: * string is returned and "input_eof" is set to "1" (otherwise it
! 257: * will be set to "0").
! 258: */
! 259: static char *
! 260: input_func(struct tmpl_ctx *ctx, char **errmsgp, int ac, char **av)
! 261: {
! 262: const struct private_info *const priv = tmpl_ctx_get_arg(ctx);
! 263: const char *const mtype = tmpl_ctx_get_mtype(ctx);
! 264: int quiet = (ac > 1 && strtol(av[1], NULL, 0) != 0);
! 265: struct termios term, oterm;
! 266: char *rtn = NULL;
! 267: char buf[1024];
! 268:
! 269: /* If no input supplied, act as if EOF */
! 270: if (priv->input == NULL) {
! 271: quiet = 0;
! 272: if (tmpl_ctx_set_var(ctx, "input_eof", "1"))
! 273: goto fail;
! 274: rtn = STRDUP(mtype, "");
! 275: goto fail;
! 276: }
! 277:
! 278: /* Turn echo off if requested */
! 279: if (quiet) {
! 280: if (tcgetattr(fileno(priv->input), &oterm) == -1)
! 281: quiet = 0;
! 282: else {
! 283: term = oterm;
! 284: term.c_lflag &= ~(ECHO|ECHOE|ECHOK|ECHOKE);
! 285: if (tcsetattr(fileno(priv->input),
! 286: TCSADRAIN|TCSASOFT, &term) == -1)
! 287: quiet = 0;
! 288: }
! 289: }
! 290:
! 291: /* Read one line of input */
! 292: if (tmpl_ctx_set_var(ctx, "input_eof", "0"))
! 293: goto fail;
! 294: if (fgets(buf, sizeof(buf), priv->input) == NULL) {
! 295: if (ferror(priv->input))
! 296: goto fail;
! 297: if (tmpl_ctx_set_var(ctx, "input_eof", "1"))
! 298: goto fail;
! 299: rtn = STRDUP(mtype, "");
! 300: goto fail;
! 301: }
! 302:
! 303: /* Trim trailing newline */
! 304: if (buf[strlen(buf) - 1] == '\n')
! 305: buf[strlen(buf) - 1] = '\0';
! 306:
! 307: /* Copy it to use the right memory type */
! 308: if ((rtn = STRDUP(mtype, buf)) == NULL)
! 309: goto fail;
! 310:
! 311: fail:
! 312: /* Clean up and return */
! 313: if (quiet)
! 314: tcsetattr(fileno(priv->input), TCSADRAIN|TCSASOFT, &oterm);
! 315: return (rtn);
! 316: }
! 317:
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>
![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to main.c CVS log ![[TXT]](/icons/cvsweb/text.gif){kind=icon}
![[DIR]](/icons/cvsweb/dir.gif){kind=icon}
Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / libpdel / tmpl / test