source: EcnlProtoTool/trunk/prototool/src/mruby.c@ 279

Last change on this file since 279 was 279, checked in by coas-nagasima, 7 years ago

ファイルを追加、更新。
  • Property svn:eol-style set to native
  • Property svn:keywords set to Id
  • Property svn:mime-type set to text/x-csrc
File size: 5.2 KB
Line 
1#include <stdlib.h>
2#include <string.h>
3
4#include "mruby.h"
5#include "mruby/array.h"
6#include "mruby/compile.h"
7#include "mruby/dump.h"
8#include "mruby/variable.h"
9
10#ifdef MRB_DISABLE_STDIO
11static void
12p(mrb_state *mrb, mrb_value obj)
13{
14 mrb_value val = mrb_inspect(mrb, obj);
15
16 fwrite(RSTRING_PTR(val), RSTRING_LEN(val), 1, stdout);
17 putc('\n', stdout);
18}
19#else
20#define p(mrb,obj) mrb_p(mrb,obj)
21#endif
22
23struct _args
24{
25 FILE *rfp;
26 char* cmdline;
27 mrb_bool fname : 1;
28 mrb_bool mrbfile : 1;
29 mrb_bool check_syntax : 1;
30 mrb_bool verbose : 1;
31 int argc;
32 char** argv;
33};
34
35static void
36usage(const char *name)
37{
38 static const char *const usage_msg[] = {
39 "switches:",
40 "-b load and execute RiteBinary (mrb) file",
41 "-c check syntax only",
42 "-e 'command' one line of script",
43 "-v print version number, then run in verbose mode",
44 "--verbose run in verbose mode",
45 "--version print the version",
46 "--copyright print the copyright",
47 NULL
48 };
49 const char *const *p = usage_msg;
50
51 printf("Usage: %s [switches] programfile\n", name);
52 while (*p)
53 printf(" %s\n", *p++);
54}
55
56static int
57parse_args(mrb_state *mrb, int argc, char **argv, struct _args *args)
58{
59 char **origargv = argv;
60 static const struct _args args_zero = { 0 };
61
62 *args = args_zero;
63
64 for (argc--, argv++; argc > 0; argc--, argv++) {
65 char *item;
66 if (argv[0][0] != '-') break;
67
68 if (strlen(*argv) <= 1) {
69 argc--; argv++;
70 args->rfp = stdin;
71 break;
72 }
73
74 item = argv[0] + 1;
75 switch (*item++) {
76 case 'b':
77 args->mrbfile = TRUE;
78 break;
79 case 'c':
80 args->check_syntax = TRUE;
81 break;
82 case 'e':
83 if (item[0]) {
84 goto append_cmdline;
85 }
86 else if (argc > 1) {
87 argc--; argv++;
88 item = argv[0];
89 append_cmdline:
90 if (!args->cmdline) {
91 size_t buflen;
92 char *buf;
93
94 buflen = strlen(item) + 1;
95 buf = (char *)mrb_malloc(mrb, buflen);
96 memcpy(buf, item, buflen);
97 args->cmdline = buf;
98 }
99 else {
100 size_t cmdlinelen;
101 size_t itemlen;
102
103 cmdlinelen = strlen(args->cmdline);
104 itemlen = strlen(item);
105 args->cmdline =
106 (char *)mrb_realloc(mrb, args->cmdline, cmdlinelen + itemlen + 2);
107 args->cmdline[cmdlinelen] = '\n';
108 memcpy(args->cmdline + cmdlinelen + 1, item, itemlen + 1);
109 }
110 }
111 else {
112 printf("%s: No code specified for -e\n", *origargv);
113 return EXIT_SUCCESS;
114 }
115 break;
116 case 'v':
117 if (!args->verbose) mrb_show_version(mrb);
118 args->verbose = TRUE;
119 break;
120 case '-':
121 if (strcmp((*argv) + 2, "version") == 0) {
122 mrb_show_version(mrb);
123 exit(EXIT_SUCCESS);
124 }
125 else if (strcmp((*argv) + 2, "verbose") == 0) {
126 args->verbose = TRUE;
127 break;
128 }
129 else if (strcmp((*argv) + 2, "copyright") == 0) {
130 mrb_show_copyright(mrb);
131 exit(EXIT_SUCCESS);
132 }
133 default:
134 return EXIT_FAILURE;
135 }
136 }
137
138 if (args->rfp == NULL && args->cmdline == NULL) {
139 if (*argv == NULL) args->rfp = stdin;
140 else {
141 args->rfp = fopen(argv[0], args->mrbfile ? "rb" : "r");
142 if (args->rfp == NULL) {
143 printf("%s: Cannot open program file. (%s)\n", *origargv, *argv);
144 return EXIT_FAILURE;
145 }
146 args->fname = TRUE;
147 args->cmdline = argv[0];
148 argc--; argv++;
149 }
150 }
151 args->argv = (char **)mrb_realloc(mrb, args->argv, sizeof(char*) * (argc + 1));
152 memcpy(args->argv, argv, (argc + 1) * sizeof(char*));
153 args->argc = argc;
154
155 return EXIT_SUCCESS;
156}
157
158static void
159cleanup(mrb_state *mrb, struct _args *args)
160{
161 if (args->rfp && args->rfp != stdin)
162 fclose(args->rfp);
163 if (!args->fname)
164 mrb_free(mrb, args->cmdline);
165 mrb_free(mrb, args->argv);
166 mrb_close(mrb);
167}
168
169int
170mruby_main(int argc, char **argv)
171{
172 mrb_state *mrb = mrb_open();
173 int n = -1;
174 int i;
175 struct _args args;
176 mrb_value ARGV;
177 mrbc_context *c;
178 mrb_value v;
179 mrb_sym zero_sym;
180
181 if (mrb == NULL) {
182 fputs("Invalid mrb_state, exiting mruby\n", stderr);
183 return EXIT_FAILURE;
184 }
185
186 n = parse_args(mrb, argc, argv, &args);
187 if (n == EXIT_FAILURE || (args.cmdline == NULL && args.rfp == NULL)) {
188 cleanup(mrb, &args);
189 usage(argv[0]);
190 return n;
191 }
192 else {
193 int ai = mrb_gc_arena_save(mrb);
194 ARGV = mrb_ary_new_capa(mrb, args.argc);
195 for (i = 0; i < args.argc; i++) {
196 mrb_ary_push(mrb, ARGV, mrb_str_new_cstr(mrb, args.argv[i]));
197 }
198 mrb_define_global_const(mrb, "ARGV", ARGV);
199
200 c = mrbc_context_new(mrb);
201 if (args.verbose)
202 c->dump_result = TRUE;
203 if (args.check_syntax)
204 c->no_exec = TRUE;
205
206 /* Set $0 */
207 zero_sym = mrb_intern_lit(mrb, "$0");
208 if (args.rfp) {
209 const char *cmdline;
210 cmdline = args.cmdline ? args.cmdline : "-";
211 mrbc_filename(mrb, c, cmdline);
212 mrb_gv_set(mrb, zero_sym, mrb_str_new_cstr(mrb, cmdline));
213 }
214 else {
215 mrbc_filename(mrb, c, "-e");
216 mrb_gv_set(mrb, zero_sym, mrb_str_new_lit(mrb, "-e"));
217 }
218
219 /* Load program */
220 if (args.mrbfile) {
221 v = mrb_load_irep_file_cxt(mrb, args.rfp, c);
222 }
223 else if (args.rfp) {
224 v = mrb_load_file_cxt(mrb, args.rfp, c);
225 }
226 else {
227 v = mrb_load_string_cxt(mrb, args.cmdline, c);
228 }
229
230 mrb_gc_arena_restore(mrb, ai);
231 mrbc_context_free(mrb, c);
232 if (mrb->exc) {
233 if (!mrb_undef_p(v)) {
234 mrb_print_error(mrb);
235 }
236 n = -1;
237 }
238 else if (args.check_syntax) {
239 printf("Syntax OK\n");
240 }
241 }
242 cleanup(mrb, &args);
243
244 return n == 0 ? EXIT_SUCCESS : EXIT_FAILURE;
245}
Note: See TracBrowser for help on using the repository browser.