lib/clippy.c

   1 /*
   2  * clippy (CLI preparator in python) main executable
   3  * Copyright (C) 2016-2017  David Lamparter for NetDEF, Inc.
   4  *
   5  * This program is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License as published by the Free
   7  * Software Foundation; either version 2 of the License, or (at your option)
   8  * any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful, but WITHOUT
  11  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  12  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  13  * more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along
  16  * with this program; see the file COPYING; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
  18  */
  19
  20 #include "config.h"
  21 #include <Python.h>
  22 #include <string.h>
  23 #include <stdlib.h>
  24 #include <wchar.h>
  25 #include "getopt.h"
  26
  27 #include "command_graph.h"
  28 #include "clippy.h"
  29
  30 #if PY_MAJOR_VERSION >= 3
  31 #define pychar wchar_t
  32 static wchar_t *wconv(const char *s)
  33 {
  34         size_t outlen = s ? mbstowcs(NULL, s, 0) : 0;
  35         wchar_t *out = malloc((outlen + 1) * sizeof(wchar_t));
  36
  37         if (outlen > 0)
  38                 mbstowcs(out, s, outlen);
  39         out[outlen] = 0;
  40         return out;
  41 }
  42 #else
  43 #define pychar char
  44 #define wconv(x) x
  45 #endif
  46
  47 int main(int argc, char **argv)
  48 {
  49         pychar **wargv;
  50
  51 #if PY_VERSION_HEX >= 0x03040000 /* 3.4 */
  52         Py_SetStandardStreamEncoding("UTF-8", NULL);
  53 #endif
  54         wchar_t *name = wconv(argv[0]);
  55         Py_SetProgramName(name);
  56         PyImport_AppendInittab("_clippy", command_py_init);
  57
  58         Py_Initialize();
  59
  60         wargv = malloc(argc * sizeof(pychar *));
  61         for (int i = 1; i < argc; i++)
  62                 wargv[i - 1] = wconv(argv[i]);
  63         PySys_SetArgv(argc - 1, wargv);
  64
  65         const char *pyfile = argc > 1 ? argv[1] : NULL;
  66         FILE *fp;
  67         if (pyfile) {
  68                 fp = fopen(pyfile, "r");
  69                 if (!fp) {
  70                         fprintf(stderr, "%s: %s\n", pyfile, strerror(errno));
  71                         return 1;
  72                 }
  73         } else {
  74                 fp = stdin;
  75                 char *ver = strdup(Py_GetVersion());
  76                 char *cr = strchr(ver, '\n');
  77                 if (cr)
  78                         *cr = ' ';
  79                 fprintf(stderr, "clippy interactive shell\n(Python %s)\n", ver);
  80                 free(ver);
  81                 PyRun_SimpleString(
  82                         "import rlcompleter, readline\n"
  83                         "readline.parse_and_bind('tab: complete')");
  84         }
  85
  86         if (PyRun_AnyFile(fp, pyfile)) {
  87                 if (PyErr_Occurred())
  88                         PyErr_Print();
  89                 return 1;
  90         }
  91         Py_Finalize();
  92
  93 #if PY_MAJOR_VERSION >= 3
  94         for (int i = 1; i < argc; i++)
  95                 free(wargv[i - 1]);
  96 #endif
  97         free(name);
  98         free(wargv);
  99         return 0;
 100 }
 101
 102 /* and now for the ugly part... provide simplified logging functions so we
 103  * don't need to link libzebra (which would be a circular build dep) */
 104
 105 #ifdef __ASSERT_FUNCTION
 106 #undef __ASSERT_FUNCTION
 107 #endif
 108
 109 #include "log.h"
 110 #include "zassert.h"
 111
 112 void vzlogx(const struct xref_logmsg *xref, int prio,
 113             const char *format, va_list args)
 114 {
 115         vfprintf(stderr, format, args);
 116         fputs("\n", stderr);
 117 }
 118
 119 void _zlog_assert_failed(const char *assertion, const char *file,
 120                          unsigned int line, const char *function)
 121 {
 122         fprintf(stderr,
 123                 "Assertion `%s' failed in file %s, line %u, function %s",
 124                 assertion, file, line, (function ? function : "?"));
 125         abort();
 126 }
 127
 128 void memory_oom(size_t size, const char *name)
 129 {
 130         abort();
 131 }