src/lib/cmdline.c

   1 /*
   2  * Copyright (c) 2009, Olivier MATZ <zer0@droids-corp.org>
   3  * All rights reserved.
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions are met:
   6  *
   7  *     * Redistributions of source code must retain the above copyright
   8  *       notice, this list of conditions and the following disclaimer.
   9  *     * Redistributions in binary form must reproduce the above copyright
  10  *       notice, this list of conditions and the following disclaimer in the
  11  *       documentation and/or other materials provided with the distribution.
  12  *     * Neither the name of the University of California, Berkeley nor the
  13  *       names of its contributors may be used to endorse or promote products
  14  *       derived from this software without specific prior written permission.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND ANY
  17  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  18  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  19  * DISCLAIMED. IN NO EVENT SHALL THE REGENTS AND CONTRIBUTORS BE LIABLE FOR ANY
  20  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  21  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  22  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  23  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  25  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26  */
  27
  28 #include <stdio.h>
  29 #include <string.h>
  30 #include <unistd.h>
  31 #include <stdlib.h>
  32 #include <stdarg.h>
  33 #include <inttypes.h>
  34 #include <fcntl.h>
  35
  36 #include <sys/socket.h>
  37 #include <netinet/in.h>
  38 #include <sys/socket.h>
  39 #include <sys/un.h>
  40
  41 #include "cmdline_parse.h"
  42 #include "cmdline_rdline.h"
  43 #include "cmdline.h"
  44
  45
  46 #define PROMPT_SUFFIX "> "
  47
  48
  49 /**********************/
  50
  51 void
  52 cmdline_valid_buffer(struct rdline *rdl, const char *buf,
  53                      __attribute__((unused)) unsigned int size)
  54 {
  55         struct cmdline *cl = rdl->opaque;
  56         int ret;
  57         ret = cmdline_parse(cl, buf);
  58         if (ret == CMDLINE_PARSE_AMBIGUOUS)
  59                 cmdline_printf(cl, "Ambiguous command\n");
  60         else if (ret == CMDLINE_PARSE_NOMATCH)
  61                 cmdline_printf(cl, "Command not found\n");
  62         else if (ret == CMDLINE_PARSE_BAD_ARGS)
  63                 cmdline_printf(cl, "Bad arguments\n");
  64 }
  65
  66 int
  67 cmdline_complete_buffer(struct rdline *rdl, const char *buf,
  68                         char *dstbuf, unsigned int dstsize,
  69                         int *state)
  70 {
  71         struct cmdline *cl = rdl->opaque;
  72         return cmdline_complete(cl, buf, state, dstbuf, dstsize);
  73 }
  74
  75 void
  76 cmdline_write_char(struct rdline *rdl, char c)
  77 {
  78         struct cmdline *cl = rdl->opaque;
  79         if (cl->s_out >= 0) {
  80                 write(cl->s_out, &c, 1);
  81         }
  82 }
  83
  84
  85 void
  86 cmdline_set_prompt(struct cmdline *cl, const char *prompt)
  87 {
  88         snprintf(cl->prompt, sizeof(cl->prompt), prompt);
  89 }
  90
  91 struct cmdline *
  92 cmdline_new(cmdline_parse_ctx_t *ctx, const char *prompt, int s_in, int s_out)
  93 {
  94         struct cmdline *cl;
  95         cl = malloc(sizeof(struct cmdline));
  96         if (cl == NULL)
  97                 return NULL;
  98         memset(cl, 0, sizeof(struct cmdline));
  99         cl->s_in = s_in;
 100         cl->s_out = s_out;
 101         cl->ctx = ctx;
 102
 103         rdline_init(&cl->rdl, cmdline_write_char,
 104                     cmdline_valid_buffer, cmdline_complete_buffer);
 105         cl->rdl.opaque = cl;
 106         cmdline_set_prompt(cl, prompt);
 107         rdline_newline(&cl->rdl, cl->prompt);
 108
 109         return cl;
 110 }
 111
 112 void
 113 cmdline_free(struct cmdline *cl)
 114 {
 115         dprintf("called\n");
 116         if (cl->s_in > 2)
 117                 close(cl->s_in);
 118         if (cl->s_out != cl->s_in && cl->s_out > 2)
 119                 close(cl->s_out);
 120         free(cl);
 121 }
 122
 123 void
 124 cmdline_printf(const struct cmdline *cl, const char *fmt, ...)
 125 {
 126         va_list ap;
 127         char buffer[BUFSIZ];
 128
 129         va_start(ap, fmt);
 130         vsnprintf(buffer, 512, fmt, ap);
 131         va_end(ap);
 132         if (cl->s_out >= 0) {
 133                 write(cl->s_out, buffer, strlen(buffer));
 134         }
 135 }
 136
 137 int
 138 cmdline_in(struct cmdline *cl, const char *buf, int size)
 139 {
 140         const char *history, *buffer;
 141         int ret = 0;
 142         int i, same;
 143
 144         /* XXX use defines instead of hardcoded values */
 145         for (i=0; i<size; i++) {
 146                 ret = rdline_char_in(&cl->rdl, buf[i]);
 147
 148                 if (ret == 1) {
 149                         buffer = rdline_get_buffer(&cl->rdl);
 150                         history = rdline_get_history_item(&cl->rdl, 0);
 151                         if (history) {
 152                                 same = !memcmp(buffer, history, strlen(history)) &&
 153                                         buffer[strlen(history)] == '\n';
 154                         }
 155                         else
 156                                 same = 0;
 157                         if (strlen(buffer) > 1 && !same)
 158                                 rdline_add_history(&cl->rdl, buffer);
 159                         rdline_newline(&cl->rdl, cl->prompt);
 160                 }
 161                 else if (ret == -2)
 162                         return -1;
 163         }
 164         return i;
 165 }