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, unsigned int size)
  53 {
  54         struct cmdline *cl = rdl->opaque;
  55         int ret;
  56         ret = cmdline_parse(cl, buf);
  57         if (ret == CMDLINE_PARSE_AMBIGUOUS)
  58                 cmdline_printf(cl, "Ambiguous command\n");
  59         else if (ret == CMDLINE_PARSE_NOMATCH)
  60                 cmdline_printf(cl, "Command not found\n");
  61         else if (ret == CMDLINE_PARSE_BAD_ARGS)
  62                 cmdline_printf(cl, "Bad arguments\n");
  63 }
  64
  65 int
  66 cmdline_complete_buffer(struct rdline *rdl, const char *buf,
  67                         char *dstbuf, unsigned int dstsize,
  68                         int *state)
  69 {
  70         struct cmdline *cl = rdl->opaque;
  71         return cmdline_complete(cl, buf, state, dstbuf, dstsize);
  72 }
  73
  74 void
  75 cmdline_write_char(struct rdline *rdl, char c)
  76 {
  77         struct cmdline *cl = rdl->opaque;
  78         if (cl->s_out >= 0) {
  79                 write(cl->s_out, &c, 1);
  80         }
  81 }
  82
  83
  84 void
  85 cmdline_set_prompt(struct cmdline *cl, const char *prompt)
  86 {
  87         snprintf(cl->prompt, sizeof(cl->prompt), prompt);
  88 }
  89
  90 struct cmdline *
  91 cmdline_new(cmdline_parse_ctx_t *ctx, const char *prompt, int s_in, int s_out)
  92 {
  93         struct cmdline *cl;
  94         cl = malloc(sizeof(struct cmdline));
  95         if (cl == NULL)
  96                 return NULL;
  97         memset(cl, 0, sizeof(struct cmdline));
  98         cl->s_in = s_in;
  99         cl->s_out = s_out;
 100         cl->ctx = ctx;
 101
 102         rdline_init(&cl->rdl, cmdline_write_char,
 103                     cmdline_valid_buffer, cmdline_complete_buffer);
 104         cl->rdl.opaque = cl;
 105         cmdline_set_prompt(cl, prompt);
 106         rdline_newline(&cl->rdl, cl->prompt);
 107
 108         return cl;
 109 }
 110
 111 void
 112 cmdline_free(struct cmdline *cl)
 113 {
 114         dprintf("called\n");
 115         if (cl->s_in > 2)
 116                 close(cl->s_in);
 117         if (cl->s_out != cl->s_in && cl->s_out > 2)
 118                 close(cl->s_out);
 119         free(cl);
 120 }
 121
 122 void
 123 cmdline_printf(const struct cmdline *cl, const char *fmt, ...)
 124 {
 125         va_list ap;
 126         char buffer[BUFSIZ];
 127
 128         va_start(ap, fmt);
 129         vsnprintf(buffer, 512, fmt, ap);
 130         va_end(ap);
 131         if (cl->s_out >= 0) {
 132                 write(cl->s_out, buffer, strlen(buffer));
 133         }
 134 }
 135
 136 int
 137 cmdline_in(struct cmdline *cl, const char *buf, int size)
 138 {
 139         const char *history, *buffer;
 140         int ret = 0;
 141         int i, same;
 142
 143         /* XXX use defines instead of hardcoded values */
 144         for (i=0; i<size; i++) {
 145                 ret = rdline_char_in(&cl->rdl, buf[i]);
 146
 147                 if (ret == 1) {
 148                         buffer = rdline_get_buffer(&cl->rdl);
 149                         history = rdline_get_history_item(&cl->rdl, 0);
 150                         if (history) {
 151                                 same = !memcmp(buffer, history, strlen(history)) &&
 152                                         buffer[strlen(history)] == '\n';
 153                         }
 154                         else
 155                                 same = 0;
 156                         if (strlen(buffer) > 1 && !same)
 157                                 rdline_add_history(&cl->rdl, buffer);
 158                         rdline_newline(&cl->rdl, cl->prompt);
 159                 }
 160                 else if (ret == -2)
 161                         return -1;
 162         }
 163         return i;
 164 }