1 /* vi:set ts=8 sts=4 sw=4 noet:
2 *
3 * MzScheme interface by Sergey Khorev <sergey.khorev@gmail.com>
4 * Based on work by Brent Fulgham <bfulgham@debian.org>
5 * (Based on lots of help from Matthew Flatt)
6 *
7 * This consists of six parts:
8 * 1. MzScheme interpreter main program
9 * 2. Routines that handle the external interface between MzScheme and
10 * Vim.
11 * 3. MzScheme input/output handlers: writes output via [e]msg().
12 * 4. Implementation of the Vim Features for MzScheme
13 * 5. Vim Window-related Manipulation Functions.
14 * 6. Vim Buffer-related Manipulation Functions
15 *
16 * NOTES
17 * 1. Memory, allocated with scheme_malloc*, need not to be freed explicitly,
18 * garbage collector will do it self
19 * 2. Requires at least NORMAL features. I can't imagine why one may want
20 * to build with SMALL or TINY features but with MzScheme interface.
21 * 3. I don't use K&R-style functions. Anyways, MzScheme headers are ANSI.
22 */
23
24 #include "vim.h"
25
26 #include "if_mzsch.h"
27
28 // Only do the following when the feature is enabled. Needed for "make
29 // depend".
30 #if defined(FEAT_MZSCHEME) || defined(PROTO)
31
32 #ifdef PROTO
33 typedef int Scheme_Object;
34 typedef int Scheme_Closed_Prim;
35 typedef int Scheme_Env;
36 typedef int Scheme_Hash_Table;
37 typedef int Scheme_Type;
38 typedef int Scheme_Thread;
39 typedef int Scheme_Closed_Prim;
40 typedef int mzshort;
41 typedef int Scheme_Prim;
42 typedef int HINSTANCE;
43 #endif
44
45 /*
46 * scheme_register_tls_space is only available on 32-bit Windows until
47 * racket-6.3. See
48 * http://docs.racket-lang.org/inside/im_memoryalloc.html?q=scheme_register_tls_space
49 */
50 #if MZSCHEME_VERSION_MAJOR >= 500 && defined(MSWIN) \
51 && defined(USE_THREAD_LOCAL) \
52 && (!defined(_WIN64) || MZSCHEME_VERSION_MAJOR >= 603)
53 # define HAVE_TLS_SPACE 1
54 #endif
55
56 /*
57 * Since version 4.x precise GC requires trampolined startup.
58 * Futures and places in version 5.x need it too.
59 */
60 #if defined(MZ_PRECISE_GC) && MZSCHEME_VERSION_MAJOR >= 400 \
61 || MZSCHEME_VERSION_MAJOR >= 500 \
62 && (defined(MZ_USE_FUTURES) || defined(MZ_USE_PLACES))
63 # define TRAMPOLINED_MZVIM_STARTUP
64 #endif
65
66 // Base data structures
67 #define SCHEME_VIMBUFFERP(obj) SAME_TYPE(SCHEME_TYPE(obj), mz_buffer_type)
68 #define SCHEME_VIMWINDOWP(obj) SAME_TYPE(SCHEME_TYPE(obj), mz_window_type)
69
70 typedef struct
71 {
72 Scheme_Object so;
73 buf_T *buf;
74 } vim_mz_buffer;
75
76 #define INVALID_BUFFER_VALUE ((buf_T *)(-1))
77
78 typedef struct
79 {
80 Scheme_Object so;
81 win_T *win;
82 } vim_mz_window;
83
84 #define INVALID_WINDOW_VALUE ((win_T *)(-1))
85
86 /*
87 * Prims that form MzScheme Vim interface
88 */
89 typedef struct
90 {
91 Scheme_Closed_Prim *prim;
92 char *name;
93 int mina; // arity information
94 int maxa;
95 } Vim_Prim;
96
97 typedef struct
98 {
99 char *name;
100 Scheme_Object *port;
101 } Port_Info;
102
103 /*
104 *========================================================================
105 * Vim-Control Commands
106 *========================================================================
107 */
108 /*
109 *========================================================================
110 * Utility functions for the vim/mzscheme interface
111 *========================================================================
112 */
113 #ifdef HAVE_SANDBOX
114 static Scheme_Object *sandbox_file_guard(int, Scheme_Object **);
115 static Scheme_Object *sandbox_network_guard(int, Scheme_Object **);
116 static void sandbox_check(void);
117 #endif
118 // Buffer-related commands
119 static Scheme_Object *buffer_new(buf_T *buf);
120 static Scheme_Object *get_buffer_by_num(void *, int, Scheme_Object **);
121 static vim_mz_buffer *get_vim_curr_buffer(void);
122
123 // Window-related commands
124 static Scheme_Object *window_new(win_T *win);
125 static vim_mz_window *get_vim_curr_window(void);
126
127 /*
128 *========================================================================
129 * Internal Function Prototypes
130 *========================================================================
131 */
132 static int vim_error_check(void);
133 static int do_mzscheme_command(exarg_T *, void *, Scheme_Closed_Prim *what);
134 static int startup_mzscheme(void);
135 static char *string_to_line(Scheme_Object *obj);
136 #if MZSCHEME_VERSION_MAJOR >= 501
137 # define OUTPUT_LEN_TYPE intptr_t
138 #else
139 # define OUTPUT_LEN_TYPE long
140 #endif
141 static void do_output(char *mesg, OUTPUT_LEN_TYPE len);
142 static void do_printf(char *format, ...) ATTRIBUTE_FORMAT_PRINTF(1, 2);
143 static void do_flush(void);
144 static Scheme_Object *_apply_thunk_catch_exceptions(
145 Scheme_Object *, Scheme_Object **);
146 static Scheme_Object *extract_exn_message(Scheme_Object *v);
147 static Scheme_Object *do_eval(void *, int noargc, Scheme_Object **noargv);
148 static Scheme_Object *do_load(void *, int noargc, Scheme_Object **noargv);
149 static void register_vim_exn(void);
150 static vim_mz_buffer *get_buffer_arg(const char *fname, int argnum,
151 int argc, Scheme_Object **argv);
152 static vim_mz_window *get_window_arg(const char *fname, int argnum,
153 int argc, Scheme_Object **argv);
154 static int line_in_range(linenr_T, buf_T *);
155 static void check_line_range(linenr_T, buf_T *);
156 static void mz_fix_cursor(int lo, int hi, int extra);
157
158 static int eval_with_exn_handling(void *, Scheme_Closed_Prim *,
159 Scheme_Object **ret);
160 static void make_modules(void);
161 static void init_exn_catching_apply(void);
162 static int mzscheme_env_main(Scheme_Env *env, int argc, char **argv);
163 static int mzscheme_init(void);
164 #ifdef FEAT_EVAL
165 static Scheme_Object *vim_to_mzscheme(typval_T *vim_value);
166 static Scheme_Object *vim_to_mzscheme_impl(typval_T *vim_value, int depth,
167 Scheme_Hash_Table *visited);
168 static int mzscheme_to_vim(Scheme_Object *obj, typval_T *tv);
169 static int mzscheme_to_vim_impl(Scheme_Object *obj, typval_T *tv, int depth,
170 Scheme_Hash_Table *visited);
171 static Scheme_Object *vim_funcref(void *data, int argc, Scheme_Object **argv);
172 #endif
173
174 #ifdef MZ_PRECISE_GC
buffer_size_proc(void * obj UNUSED)175 static int buffer_size_proc(void *obj UNUSED)
176 {
177 return gcBYTES_TO_WORDS(sizeof(vim_mz_buffer));
178 }
buffer_mark_proc(void * obj)179 static int buffer_mark_proc(void *obj)
180 {
181 return buffer_size_proc(obj);
182 }
buffer_fixup_proc(void * obj)183 static int buffer_fixup_proc(void *obj)
184 {
185 // apparently not needed as the object will be uncollectable while
186 // the buffer is alive
187 // vim_mz_buffer* buf = (vim_mz_buffer*) obj;
188 // buf->buf->b_mzscheme_ref = GC_fixup_self(obj);
189
190 return buffer_size_proc(obj);
191 }
window_size_proc(void * obj UNUSED)192 static int window_size_proc(void *obj UNUSED)
193 {
194 return gcBYTES_TO_WORDS(sizeof(vim_mz_window));
195 }
window_mark_proc(void * obj)196 static int window_mark_proc(void *obj)
197 {
198 return window_size_proc(obj);
199 }
window_fixup_proc(void * obj)200 static int window_fixup_proc(void *obj)
201 {
202 // apparently not needed as the object will be uncollectable while
203 // the window is alive
204 // vim_mz_window* win = (vim_mz_window*) obj;
205 // win->win->w_mzscheme_ref = GC_fixup_self(obj);
206 //
207 return window_size_proc(obj);
208 }
209 // with precise GC, w_mzscheme_ref and b_mzscheme_ref are immobile boxes
210 // containing pointers to a window/buffer
211 // with conservative GC these are simply pointers
212 # define WINDOW_REF(win) *(vim_mz_window **)((win)->w_mzscheme_ref)
213 # define BUFFER_REF(buf) *(vim_mz_buffer **)((buf)->b_mzscheme_ref)
214 #else
215 # define WINDOW_REF(win) (vim_mz_window *)((win)->w_mzscheme_ref)
216 # define BUFFER_REF(buf) (vim_mz_buffer *)((buf)->b_mzscheme_ref)
217 #endif
218
219 #if defined(DYNAMIC_MZSCHEME) || defined(PROTO)
220 static Scheme_Object *dll_scheme_eof;
221 static Scheme_Object *dll_scheme_false;
222 static Scheme_Object *dll_scheme_void;
223 static Scheme_Object *dll_scheme_null;
224 static Scheme_Object *dll_scheme_true;
225
226 static Scheme_Thread **dll_scheme_current_thread_ptr;
227
228 static void (**dll_scheme_console_printf_ptr)(char *str, ...);
229 static void (**dll_scheme_console_output_ptr)(char *str, OUTPUT_LEN_TYPE len);
230 static void (**dll_scheme_notify_multithread_ptr)(int on);
231
232 static void *(*dll_GC_malloc)(size_t size_in_bytes);
233 static void *(*dll_GC_malloc_atomic)(size_t size_in_bytes);
234 static Scheme_Env *(*dll_scheme_basic_env)(void);
235 static void (*dll_scheme_check_threads)(void);
236 static void (*dll_scheme_register_static)(void *ptr, long size);
237 static void (*dll_scheme_set_stack_base)(void *base, int no_auto_statics);
238 static void (*dll_scheme_add_global)(const char *name, Scheme_Object *val,
239 Scheme_Env *env);
240 static void (*dll_scheme_add_global_symbol)(Scheme_Object *name,
241 Scheme_Object *val, Scheme_Env *env);
242 static Scheme_Object *(*dll_scheme_apply)(Scheme_Object *rator, int num_rands,
243 Scheme_Object **rands);
244 static Scheme_Object *(*dll_scheme_builtin_value)(const char *name);
245 # if MZSCHEME_VERSION_MAJOR >= 299
246 static Scheme_Object *(*dll_scheme_byte_string_to_char_string)(Scheme_Object *s);
247 static Scheme_Object *(*dll_scheme_make_path)(const char *chars);
248 # endif
249 static void (*dll_scheme_close_input_port)(Scheme_Object *port);
250 static void (*dll_scheme_count_lines)(Scheme_Object *port);
251 #if MZSCHEME_VERSION_MAJOR < 360
252 static Scheme_Object *(*dll_scheme_current_continuation_marks)(void);
253 #else
254 static Scheme_Object *(*dll_scheme_current_continuation_marks)(Scheme_Object *prompt_tag);
255 #endif
256 static void (*dll_scheme_display)(Scheme_Object *obj, Scheme_Object *port);
257 static char *(*dll_scheme_display_to_string)(Scheme_Object *obj, OUTPUT_LEN_TYPE *len);
258 static int (*dll_scheme_eq)(Scheme_Object *obj1, Scheme_Object *obj2);
259 static Scheme_Object *(*dll_scheme_do_eval)(Scheme_Object *obj,
260 int _num_rands, Scheme_Object **rands, int val);
261 static void (*dll_scheme_dont_gc_ptr)(void *p);
262 static Scheme_Object *(*dll_scheme_eval)(Scheme_Object *obj, Scheme_Env *env);
263 static Scheme_Object *(*dll_scheme_eval_string)(const char *str,
264 Scheme_Env *env);
265 static Scheme_Object *(*dll_scheme_eval_string_all)(const char *str,
266 Scheme_Env *env, int all);
267 static void (*dll_scheme_finish_primitive_module)(Scheme_Env *env);
268 # if MZSCHEME_VERSION_MAJOR < 299
269 static char *(*dll_scheme_format)(char *format, int flen, int argc,
270 Scheme_Object **argv, long *rlen);
271 # else
272 static char *(*dll_scheme_format_utf8)(char *format, int flen, int argc,
273 Scheme_Object **argv, OUTPUT_LEN_TYPE *rlen);
274 static Scheme_Object *(*dll_scheme_get_param)(Scheme_Config *c, int pos);
275 # endif
276 static void (*dll_scheme_gc_ptr_ok)(void *p);
277 # if MZSCHEME_VERSION_MAJOR < 299
278 static char *(*dll_scheme_get_sized_string_output)(Scheme_Object *,
279 long *len);
280 # else
281 static char *(*dll_scheme_get_sized_byte_string_output)(Scheme_Object *,
282 OUTPUT_LEN_TYPE *len);
283 # endif
284 static Scheme_Object *(*dll_scheme_intern_symbol)(const char *name);
285 static Scheme_Object *(*dll_scheme_lookup_global)(Scheme_Object *symbol,
286 Scheme_Env *env);
287 static Scheme_Object *(*dll_scheme_make_closed_prim_w_arity)
288 (Scheme_Closed_Prim *prim, void *data, const char *name, mzshort mina,
289 mzshort maxa);
290 static Scheme_Object *(*dll_scheme_make_integer_value)(long i);
291 static Scheme_Object *(*dll_scheme_make_pair)(Scheme_Object *car,
292 Scheme_Object *cdr);
293 static Scheme_Object *(*dll_scheme_make_prim_w_arity)(Scheme_Prim *prim,
294 const char *name, mzshort mina, mzshort maxa);
295 # if MZSCHEME_VERSION_MAJOR < 299
296 static Scheme_Object *(*dll_scheme_make_string)(const char *chars);
297 static Scheme_Object *(*dll_scheme_make_string_output_port)();
298 # else
299 static Scheme_Object *(*dll_scheme_make_byte_string)(const char *chars);
300 static Scheme_Object *(*dll_scheme_make_byte_string_output_port)();
301 # endif
302 static Scheme_Object *(*dll_scheme_make_struct_instance)(Scheme_Object *stype,
303 int argc, Scheme_Object **argv);
304 static Scheme_Object **(*dll_scheme_make_struct_names)(Scheme_Object *base,
305 Scheme_Object *field_names, int flags, int *count_out);
306 static Scheme_Object *(*dll_scheme_make_struct_type)(Scheme_Object *base,
307 Scheme_Object *parent, Scheme_Object *inspector, int num_fields,
308 int num_uninit_fields, Scheme_Object *uninit_val,
309 Scheme_Object *properties
310 # if MZSCHEME_VERSION_MAJOR >= 299
311 , Scheme_Object *guard
312 # endif
313 );
314 static Scheme_Object **(*dll_scheme_make_struct_values)(
315 Scheme_Object *struct_type, Scheme_Object **names, int count,
316 int flags);
317 static Scheme_Type (*dll_scheme_make_type)(const char *name);
318 static Scheme_Object *(*dll_scheme_make_vector)(int size,
319 Scheme_Object *fill);
320 static void *(*dll_scheme_malloc_fail_ok)(void *(*f)(size_t), size_t);
321 static Scheme_Object *(*dll_scheme_open_input_file)(const char *name,
322 const char *who);
323 static Scheme_Env *(*dll_scheme_primitive_module)(Scheme_Object *name,
324 Scheme_Env *for_env);
325 static int (*dll_scheme_proper_list_length)(Scheme_Object *list);
326 static void (*dll_scheme_raise)(Scheme_Object *exn);
327 static Scheme_Object *(*dll_scheme_read)(Scheme_Object *port);
328 static void (*dll_scheme_signal_error)(const char *msg, ...);
329 static void (*dll_scheme_wrong_type)(const char *name, const char *expected,
330 int which, int argc, Scheme_Object **argv);
331 # if MZSCHEME_VERSION_MAJOR >= 299
332 static void (*dll_scheme_set_param)(Scheme_Config *c, int pos,
333 Scheme_Object *o);
334 static Scheme_Config *(*dll_scheme_current_config)(void);
335 static Scheme_Object *(*dll_scheme_char_string_to_byte_string)
336 (Scheme_Object *s);
337 static Scheme_Object *(*dll_scheme_char_string_to_path)
338 (Scheme_Object *s);
339 static void *(*dll_scheme_set_collects_path)(Scheme_Object *p);
340 # endif
341 static Scheme_Hash_Table *(*dll_scheme_make_hash_table)(int type);
342 static void (*dll_scheme_hash_set)(Scheme_Hash_Table *table,
343 Scheme_Object *key, Scheme_Object *value);
344 static Scheme_Object *(*dll_scheme_hash_get)(Scheme_Hash_Table *table,
345 Scheme_Object *key);
346 static Scheme_Object *(*dll_scheme_make_double)(double d);
347 static Scheme_Object *(*dll_scheme_make_sized_byte_string)(char *chars,
348 long len, int copy);
349 static Scheme_Object *(*dll_scheme_namespace_require)(Scheme_Object *req);
350 static Scheme_Object *(*dll_scheme_dynamic_wind)(void (*pre)(void *), Scheme_Object *(* volatile act)(void *), void (* volatile post)(void *), Scheme_Object *(*jmp_handler)(void *), void * volatile data);
351 # ifdef MZ_PRECISE_GC
352 static void *(*dll_GC_malloc_one_tagged)(size_t size_in_bytes);
353 static void (*dll_GC_register_traversers)(short tag, Size_Proc size, Mark_Proc mark, Fixup_Proc fixup, int is_constant_size, int is_atomic);
354 # endif
355 # if MZSCHEME_VERSION_MAJOR >= 400
356 static void (*dll_scheme_init_collection_paths)(Scheme_Env *global_env, Scheme_Object *extra_dirs);
357 static void **(*dll_scheme_malloc_immobile_box)(void *p);
358 static void (*dll_scheme_free_immobile_box)(void **b);
359 # endif
360 # if MZSCHEME_VERSION_MAJOR >= 500
361 # ifdef TRAMPOLINED_MZVIM_STARTUP
362 static int (*dll_scheme_main_setup)(int no_auto_statics, Scheme_Env_Main _main, int argc, char **argv);
363 # if defined(IMPLEMENT_THREAD_LOCAL_VIA_WIN_TLS) || MZSCHEME_VERSION_MAJOR >= 603
364 static void (*dll_scheme_register_tls_space)(void *tls_space, int _tls_index);
365 # endif
366 # endif
367 # if defined(IMPLEMENT_THREAD_LOCAL_VIA_WIN_TLS) || defined(IMPLEMENT_THREAD_LOCAL_EXTERNALLY_VIA_PROC)
368 static Thread_Local_Variables *(*dll_scheme_external_get_thread_local_variables)(void);
369 # endif
370 # endif
371 # if MZSCHEME_VERSION_MAJOR >= 600
372 static void (*dll_scheme_embedded_load)(intptr_t len, const char *s, int predefined);
373 static void (*dll_scheme_register_embedded_load)(intptr_t len, const char *s);
374 static void (*dll_scheme_set_config_path)(Scheme_Object *p);
375 # endif
376
377 #if defined(DYNAMIC_MZSCHEME) // not when defined(PROTO)
378
379 // arrays are imported directly
380 # define scheme_eof dll_scheme_eof
381 # define scheme_false dll_scheme_false
382 # define scheme_void dll_scheme_void
383 # define scheme_null dll_scheme_null
384 # define scheme_true dll_scheme_true
385
386 // pointers are GetProceAddress'ed as pointers to pointer
387 #if !defined(USE_THREAD_LOCAL) && !defined(LINK_EXTENSIONS_BY_TABLE)
388 # define scheme_current_thread (*dll_scheme_current_thread_ptr)
389 # endif
390 # define scheme_console_printf (*dll_scheme_console_printf_ptr)
391 # define scheme_console_output (*dll_scheme_console_output_ptr)
392 # define scheme_notify_multithread (*dll_scheme_notify_multithread_ptr)
393
394 // and functions in a usual way
395 # define GC_malloc dll_GC_malloc
396 # define GC_malloc_atomic dll_GC_malloc_atomic
397
398 # define scheme_add_global dll_scheme_add_global
399 # define scheme_add_global_symbol dll_scheme_add_global_symbol
400 # define scheme_apply dll_scheme_apply
401 # define scheme_basic_env dll_scheme_basic_env
402 # define scheme_builtin_value dll_scheme_builtin_value
403 # if MZSCHEME_VERSION_MAJOR >= 299
404 # define scheme_byte_string_to_char_string dll_scheme_byte_string_to_char_string
405 # define scheme_make_path dll_scheme_make_path
406 # endif
407 # define scheme_check_threads dll_scheme_check_threads
408 # define scheme_close_input_port dll_scheme_close_input_port
409 # define scheme_count_lines dll_scheme_count_lines
410 # define scheme_current_continuation_marks \
411 dll_scheme_current_continuation_marks
412 # define scheme_display dll_scheme_display
413 # define scheme_display_to_string dll_scheme_display_to_string
414 # define scheme_do_eval dll_scheme_do_eval
415 # define scheme_dont_gc_ptr dll_scheme_dont_gc_ptr
416 # define scheme_eq dll_scheme_eq
417 # define scheme_eval dll_scheme_eval
418 # define scheme_eval_string dll_scheme_eval_string
419 # define scheme_eval_string_all dll_scheme_eval_string_all
420 # define scheme_finish_primitive_module dll_scheme_finish_primitive_module
421 # if MZSCHEME_VERSION_MAJOR < 299
422 # define scheme_format dll_scheme_format
423 # else
424 # define scheme_format_utf8 dll_scheme_format_utf8
425 # endif
426 # define scheme_gc_ptr_ok dll_scheme_gc_ptr_ok
427 # if MZSCHEME_VERSION_MAJOR < 299
428 # define scheme_get_sized_byte_string_output dll_scheme_get_sized_string_output
429 # else
430 # define scheme_get_sized_byte_string_output \
431 dll_scheme_get_sized_byte_string_output
432 # define scheme_get_param dll_scheme_get_param
433 # endif
434 # define scheme_intern_symbol dll_scheme_intern_symbol
435 # define scheme_lookup_global dll_scheme_lookup_global
436 # define scheme_make_closed_prim_w_arity dll_scheme_make_closed_prim_w_arity
437 # define scheme_make_integer_value dll_scheme_make_integer_value
438 # define scheme_make_pair dll_scheme_make_pair
439 # define scheme_make_prim_w_arity dll_scheme_make_prim_w_arity
440 # if MZSCHEME_VERSION_MAJOR < 299
441 # define scheme_make_byte_string dll_scheme_make_string
442 # define scheme_make_byte_string_output_port dll_scheme_make_string_output_port
443 # else
444 # define scheme_make_byte_string dll_scheme_make_byte_string
445 # define scheme_make_byte_string_output_port \
446 dll_scheme_make_byte_string_output_port
447 # endif
448 # define scheme_make_struct_instance dll_scheme_make_struct_instance
449 # define scheme_make_struct_names dll_scheme_make_struct_names
450 # define scheme_make_struct_type dll_scheme_make_struct_type
451 # define scheme_make_struct_values dll_scheme_make_struct_values
452 # define scheme_make_type dll_scheme_make_type
453 # define scheme_make_vector dll_scheme_make_vector
454 # define scheme_malloc_fail_ok dll_scheme_malloc_fail_ok
455 # define scheme_open_input_file dll_scheme_open_input_file
456 # define scheme_primitive_module dll_scheme_primitive_module
457 # define scheme_proper_list_length dll_scheme_proper_list_length
458 # define scheme_raise dll_scheme_raise
459 # define scheme_read dll_scheme_read
460 # define scheme_register_static dll_scheme_register_static
461 # define scheme_set_stack_base dll_scheme_set_stack_base
462 # define scheme_signal_error dll_scheme_signal_error
463 # define scheme_wrong_type dll_scheme_wrong_type
464 # if MZSCHEME_VERSION_MAJOR >= 299
465 # define scheme_set_param dll_scheme_set_param
466 # define scheme_current_config dll_scheme_current_config
467 # define scheme_char_string_to_byte_string \
468 dll_scheme_char_string_to_byte_string
469 # define scheme_char_string_to_path \
470 dll_scheme_char_string_to_path
471 # define scheme_set_collects_path dll_scheme_set_collects_path
472 # endif
473 # define scheme_make_hash_table dll_scheme_make_hash_table
474 # define scheme_hash_set dll_scheme_hash_set
475 # define scheme_hash_get dll_scheme_hash_get
476 # define scheme_make_double dll_scheme_make_double
477 # define scheme_make_sized_byte_string dll_scheme_make_sized_byte_string
478 # define scheme_namespace_require dll_scheme_namespace_require
479 # define scheme_dynamic_wind dll_scheme_dynamic_wind
480 # ifdef MZ_PRECISE_GC
481 # define GC_malloc_one_tagged dll_GC_malloc_one_tagged
482 # define GC_register_traversers dll_GC_register_traversers
483 # endif
484 # if MZSCHEME_VERSION_MAJOR >= 400
485 # ifdef TRAMPOLINED_MZVIM_STARTUP
486 # define scheme_main_setup dll_scheme_main_setup
487 # if defined(IMPLEMENT_THREAD_LOCAL_VIA_WIN_TLS) || MZSCHEME_VERSION_MAJOR >= 603
488 # define scheme_register_tls_space dll_scheme_register_tls_space
489 # endif
490 # endif
491 # define scheme_init_collection_paths dll_scheme_init_collection_paths
492 # define scheme_malloc_immobile_box dll_scheme_malloc_immobile_box
493 # define scheme_free_immobile_box dll_scheme_free_immobile_box
494 # endif
495 # if MZSCHEME_VERSION_MAJOR >= 600
496 # define scheme_embedded_load dll_scheme_embedded_load
497 # define scheme_register_embedded_load dll_scheme_register_embedded_load
498 # define scheme_set_config_path dll_scheme_set_config_path
499 # endif
500
501 # if MZSCHEME_VERSION_MAJOR >= 500
502 # if defined(IMPLEMENT_THREAD_LOCAL_VIA_WIN_TLS) || defined(IMPLEMENT_THREAD_LOCAL_EXTERNALLY_VIA_PROC)
503 // define as function for macro in schthread.h
504 Thread_Local_Variables *
scheme_external_get_thread_local_variables(void)505 scheme_external_get_thread_local_variables(void)
506 {
507 return dll_scheme_external_get_thread_local_variables();
508 }
509 # endif
510 # endif
511
512 #endif
513
514 typedef struct
515 {
516 char *name;
517 void **ptr;
518 } Thunk_Info;
519
520 static Thunk_Info mzgc_imports[] = {
521 {"GC_malloc", (void **)&dll_GC_malloc},
522 {"GC_malloc_atomic", (void **)&dll_GC_malloc_atomic},
523 {NULL, NULL}};
524
525 static Thunk_Info mzsch_imports[] = {
526 {"scheme_eof", (void **)&dll_scheme_eof},
527 {"scheme_false", (void **)&dll_scheme_false},
528 {"scheme_void", (void **)&dll_scheme_void},
529 {"scheme_null", (void **)&dll_scheme_null},
530 {"scheme_true", (void **)&dll_scheme_true},
531 #if !defined(USE_THREAD_LOCAL) && !defined(LINK_EXTENSIONS_BY_TABLE)
532 {"scheme_current_thread", (void **)&dll_scheme_current_thread_ptr},
533 #endif
534 {"scheme_console_printf", (void **)&dll_scheme_console_printf_ptr},
535 {"scheme_console_output", (void **)&dll_scheme_console_output_ptr},
536 {"scheme_notify_multithread",
537 (void **)&dll_scheme_notify_multithread_ptr},
538 {"scheme_add_global", (void **)&dll_scheme_add_global},
539 {"scheme_add_global_symbol", (void **)&dll_scheme_add_global_symbol},
540 {"scheme_apply", (void **)&dll_scheme_apply},
541 {"scheme_basic_env", (void **)&dll_scheme_basic_env},
542 # if MZSCHEME_VERSION_MAJOR >= 299
543 {"scheme_byte_string_to_char_string", (void **)&dll_scheme_byte_string_to_char_string},
544 {"scheme_make_path", (void **)&dll_scheme_make_path},
545 # endif
546 {"scheme_builtin_value", (void **)&dll_scheme_builtin_value},
547 {"scheme_check_threads", (void **)&dll_scheme_check_threads},
548 {"scheme_close_input_port", (void **)&dll_scheme_close_input_port},
549 {"scheme_count_lines", (void **)&dll_scheme_count_lines},
550 {"scheme_current_continuation_marks",
551 (void **)&dll_scheme_current_continuation_marks},
552 {"scheme_display", (void **)&dll_scheme_display},
553 {"scheme_display_to_string", (void **)&dll_scheme_display_to_string},
554 {"scheme_do_eval", (void **)&dll_scheme_do_eval},
555 {"scheme_dont_gc_ptr", (void **)&dll_scheme_dont_gc_ptr},
556 {"scheme_eq", (void **)&dll_scheme_eq},
557 {"scheme_eval", (void **)&dll_scheme_eval},
558 {"scheme_eval_string", (void **)&dll_scheme_eval_string},
559 {"scheme_eval_string_all", (void **)&dll_scheme_eval_string_all},
560 {"scheme_finish_primitive_module",
561 (void **)&dll_scheme_finish_primitive_module},
562 # if MZSCHEME_VERSION_MAJOR < 299
563 {"scheme_format", (void **)&dll_scheme_format},
564 # else
565 {"scheme_format_utf8", (void **)&dll_scheme_format_utf8},
566 {"scheme_get_param", (void **)&dll_scheme_get_param},
567 #endif
568 {"scheme_gc_ptr_ok", (void **)&dll_scheme_gc_ptr_ok},
569 # if MZSCHEME_VERSION_MAJOR < 299
570 {"scheme_get_sized_string_output",
571 (void **)&dll_scheme_get_sized_string_output},
572 # else
573 {"scheme_get_sized_byte_string_output",
574 (void **)&dll_scheme_get_sized_byte_string_output},
575 #endif
576 {"scheme_intern_symbol", (void **)&dll_scheme_intern_symbol},
577 {"scheme_lookup_global", (void **)&dll_scheme_lookup_global},
578 {"scheme_make_closed_prim_w_arity",
579 (void **)&dll_scheme_make_closed_prim_w_arity},
580 {"scheme_make_integer_value", (void **)&dll_scheme_make_integer_value},
581 {"scheme_make_pair", (void **)&dll_scheme_make_pair},
582 {"scheme_make_prim_w_arity", (void **)&dll_scheme_make_prim_w_arity},
583 # if MZSCHEME_VERSION_MAJOR < 299
584 {"scheme_make_string", (void **)&dll_scheme_make_string},
585 {"scheme_make_string_output_port",
586 (void **)&dll_scheme_make_string_output_port},
587 # else
588 {"scheme_make_byte_string", (void **)&dll_scheme_make_byte_string},
589 {"scheme_make_byte_string_output_port",
590 (void **)&dll_scheme_make_byte_string_output_port},
591 # endif
592 {"scheme_make_struct_instance",
593 (void **)&dll_scheme_make_struct_instance},
594 {"scheme_make_struct_names", (void **)&dll_scheme_make_struct_names},
595 {"scheme_make_struct_type", (void **)&dll_scheme_make_struct_type},
596 {"scheme_make_struct_values", (void **)&dll_scheme_make_struct_values},
597 {"scheme_make_type", (void **)&dll_scheme_make_type},
598 {"scheme_make_vector", (void **)&dll_scheme_make_vector},
599 {"scheme_malloc_fail_ok", (void **)&dll_scheme_malloc_fail_ok},
600 {"scheme_open_input_file", (void **)&dll_scheme_open_input_file},
601 {"scheme_primitive_module", (void **)&dll_scheme_primitive_module},
602 {"scheme_proper_list_length", (void **)&dll_scheme_proper_list_length},
603 {"scheme_raise", (void **)&dll_scheme_raise},
604 {"scheme_read", (void **)&dll_scheme_read},
605 {"scheme_register_static", (void **)&dll_scheme_register_static},
606 {"scheme_set_stack_base", (void **)&dll_scheme_set_stack_base},
607 {"scheme_signal_error", (void **)&dll_scheme_signal_error},
608 {"scheme_wrong_type", (void **)&dll_scheme_wrong_type},
609 # if MZSCHEME_VERSION_MAJOR >= 299
610 {"scheme_set_param", (void **)&dll_scheme_set_param},
611 {"scheme_current_config", (void **)&dll_scheme_current_config},
612 {"scheme_char_string_to_byte_string",
613 (void **)&dll_scheme_char_string_to_byte_string},
614 {"scheme_char_string_to_path", (void **)&dll_scheme_char_string_to_path},
615 {"scheme_set_collects_path", (void **)&dll_scheme_set_collects_path},
616 # endif
617 {"scheme_make_hash_table", (void **)&dll_scheme_make_hash_table},
618 {"scheme_hash_set", (void **)&dll_scheme_hash_set},
619 {"scheme_hash_get", (void **)&dll_scheme_hash_get},
620 {"scheme_make_double", (void **)&dll_scheme_make_double},
621 {"scheme_make_sized_byte_string", (void **)&dll_scheme_make_sized_byte_string},
622 {"scheme_namespace_require", (void **)&dll_scheme_namespace_require},
623 {"scheme_dynamic_wind", (void **)&dll_scheme_dynamic_wind},
624 # ifdef MZ_PRECISE_GC
625 {"GC_malloc_one_tagged", (void **)&dll_GC_malloc_one_tagged},
626 {"GC_register_traversers", (void **)&dll_GC_register_traversers},
627 # endif
628 # if MZSCHEME_VERSION_MAJOR >= 400
629 # ifdef TRAMPOLINED_MZVIM_STARTUP
630 {"scheme_main_setup", (void **)&dll_scheme_main_setup},
631 # if defined(IMPLEMENT_THREAD_LOCAL_VIA_WIN_TLS) || MZSCHEME_VERSION_MAJOR >= 603
632 {"scheme_register_tls_space", (void **)&dll_scheme_register_tls_space},
633 # endif
634 # endif
635 {"scheme_init_collection_paths", (void **)&dll_scheme_init_collection_paths},
636 {"scheme_malloc_immobile_box", (void **)&dll_scheme_malloc_immobile_box},
637 {"scheme_free_immobile_box", (void **)&dll_scheme_free_immobile_box},
638 # endif
639 # if MZSCHEME_VERSION_MAJOR >= 500
640 # if defined(IMPLEMENT_THREAD_LOCAL_VIA_WIN_TLS) || defined(IMPLEMENT_THREAD_LOCAL_EXTERNALLY_VIA_PROC)
641 {"scheme_external_get_thread_local_variables", (void **)&dll_scheme_external_get_thread_local_variables},
642 # endif
643 # endif
644 # if MZSCHEME_VERSION_MAJOR >= 600
645 {"scheme_embedded_load", (void **)&dll_scheme_embedded_load},
646 {"scheme_register_embedded_load", (void **)&dll_scheme_register_embedded_load},
647 {"scheme_set_config_path", (void **)&dll_scheme_set_config_path},
648 # endif
649 {NULL, NULL}};
650
651 static HINSTANCE hMzGC = 0;
652 static HINSTANCE hMzSch = 0;
653
654 static void dynamic_mzscheme_end(void);
655 static int mzscheme_runtime_link_init(char *sch_dll, char *gc_dll,
656 int verbose);
657
658 static int
mzscheme_runtime_link_init(char * sch_dll,char * gc_dll,int verbose)659 mzscheme_runtime_link_init(char *sch_dll, char *gc_dll, int verbose)
660 {
661 Thunk_Info *thunk = NULL;
662
663 if (hMzGC && hMzSch)
664 return OK;
665 hMzSch = vimLoadLib(sch_dll);
666 hMzGC = vimLoadLib(gc_dll);
667
668 if (!hMzGC)
669 {
670 if (verbose)
671 semsg(_(e_loadlib), gc_dll, GetWin32Error());
672 return FAIL;
673 }
674
675 if (!hMzSch)
676 {
677 if (verbose)
678 semsg(_(e_loadlib), sch_dll, GetWin32Error());
679 return FAIL;
680 }
681
682 for (thunk = mzsch_imports; thunk->name; thunk++)
683 {
684 if ((*thunk->ptr =
685 (void *)GetProcAddress(hMzSch, thunk->name)) == NULL)
686 {
687 FreeLibrary(hMzSch);
688 hMzSch = 0;
689 FreeLibrary(hMzGC);
690 hMzGC = 0;
691 if (verbose)
692 semsg(_(e_loadfunc), thunk->name);
693 return FAIL;
694 }
695 }
696 for (thunk = mzgc_imports; thunk->name; thunk++)
697 {
698 if ((*thunk->ptr =
699 (void *)GetProcAddress(hMzGC, thunk->name)) == NULL)
700 {
701 FreeLibrary(hMzSch);
702 hMzSch = 0;
703 FreeLibrary(hMzGC);
704 hMzGC = 0;
705 if (verbose)
706 semsg(_(e_loadfunc), thunk->name);
707 return FAIL;
708 }
709 }
710 return OK;
711 }
712
713 int
mzscheme_enabled(int verbose)714 mzscheme_enabled(int verbose)
715 {
716 return mzscheme_runtime_link_init(
717 (char *)p_mzschemedll, (char *)p_mzschemegcdll, verbose) == OK;
718 }
719
720 static void
dynamic_mzscheme_end(void)721 dynamic_mzscheme_end(void)
722 {
723 if (hMzSch)
724 {
725 FreeLibrary(hMzSch);
726 hMzSch = 0;
727 }
728 if (hMzGC)
729 {
730 FreeLibrary(hMzGC);
731 hMzGC = 0;
732 }
733 }
734 #endif // DYNAMIC_MZSCHEME
735
736 #if MZSCHEME_VERSION_MAJOR < 299
737 # define GUARANTEED_STRING_ARG(proc, num) GUARANTEE_STRING(proc, num)
738 #else
739 static Scheme_Object *
guaranteed_byte_string_arg(char * proc,int num,int argc,Scheme_Object ** argv)740 guaranteed_byte_string_arg(char *proc, int num, int argc, Scheme_Object **argv)
741 {
742 if (SCHEME_BYTE_STRINGP(argv[num]))
743 {
744 return argv[num];
745 }
746 else if (SCHEME_CHAR_STRINGP(argv[num]))
747 {
748 Scheme_Object *tmp = NULL;
749 MZ_GC_DECL_REG(2);
750 MZ_GC_VAR_IN_REG(0, argv[num]);
751 MZ_GC_VAR_IN_REG(1, tmp);
752 MZ_GC_REG();
753 tmp = scheme_char_string_to_byte_string(argv[num]);
754 MZ_GC_UNREG();
755 return tmp;
756 }
757 else
758 scheme_wrong_type(proc, "string", num, argc, argv);
759 // unreachable
760 return scheme_void;
761 }
762 # define GUARANTEED_STRING_ARG(proc, num) guaranteed_byte_string_arg(proc, num, argc, argv)
763 #endif
764
765 // need to put it here for dynamic stuff to work
766 #if defined(INCLUDE_MZSCHEME_BASE)
767 # include "mzscheme_base.c"
768 #endif
769
770 /*
771 *========================================================================
772 * 1. MzScheme interpreter startup
773 *========================================================================
774 */
775
776 static Scheme_Type mz_buffer_type;
777 static Scheme_Type mz_window_type;
778
779 static int initialized = FALSE;
780 #ifdef DYNAMIC_MZSCHEME
781 static int disabled = FALSE;
782 #endif
783 static int load_base_module_failed = FALSE;
784
785 // global environment
786 static Scheme_Env *environment = NULL;
787 // output/error handlers
788 static Scheme_Object *curout = NULL;
789 static Scheme_Object *curerr = NULL;
790 // exn:vim exception
791 static Scheme_Object *exn_catching_apply = NULL;
792 static Scheme_Object *exn_p = NULL;
793 static Scheme_Object *exn_message = NULL;
794 static Scheme_Object *vim_exn = NULL; // Vim Error exception
795
796 #if !defined(MZ_PRECISE_GC) || MZSCHEME_VERSION_MAJOR < 400
797 static void *stack_base = NULL;
798 #endif
799
800 static long range_start;
801 static long range_end;
802
803 // MzScheme threads scheduling stuff
804 static int mz_threads_allow = 0;
805
806 #if defined(FEAT_GUI_MSWIN)
807 static void CALLBACK timer_proc(HWND, UINT, UINT_PTR, DWORD);
808 static UINT timer_id = 0;
809 #elif defined(FEAT_GUI_GTK)
810 static gboolean timer_proc(gpointer);
811 static guint timer_id = 0;
812 #elif defined(FEAT_GUI_MOTIF) || defined(FEAT_GUI_ATHENA)
813 static void timer_proc(XtPointer, XtIntervalId *);
814 static XtIntervalId timer_id = (XtIntervalId)0;
815 #endif
816
817 #if !defined(FEAT_GUI_MSWIN) || defined(VIMDLL) // Win32 console and Unix
818 void
mzvim_check_threads(void)819 mzvim_check_threads(void)
820 {
821 // Last time MzScheme threads were scheduled
822 static time_t mz_last_time = 0;
823
824 if (mz_threads_allow && p_mzq > 0)
825 {
826 time_t now = time(NULL);
827
828 if ((now - mz_last_time) * 1000 > p_mzq)
829 {
830 mz_last_time = now;
831 scheme_check_threads();
832 }
833 }
834 }
835 #endif
836
837 #if defined(MZSCHEME_GUI_THREADS) || defined(PROTO)
838 static void setup_timer(void);
839 static void remove_timer(void);
840
841 // timers are presented in GUI only
842 # if defined(FEAT_GUI_MSWIN)
843 static void CALLBACK
timer_proc(HWND hwnd UNUSED,UINT uMsg UNUSED,UINT_PTR idEvent UNUSED,DWORD dwTime UNUSED)844 timer_proc(HWND hwnd UNUSED, UINT uMsg UNUSED, UINT_PTR idEvent UNUSED, DWORD dwTime UNUSED)
845 # elif defined(FEAT_GUI_GTK)
846 static gboolean
847 timer_proc(gpointer data UNUSED)
848 # elif defined(FEAT_GUI_MOTIF) || defined(FEAT_GUI_ATHENA)
849 static void
850 timer_proc(XtPointer timed_out UNUSED, XtIntervalId *interval_id UNUSED)
851 # endif
852 {
853 scheme_check_threads();
854 # if defined(FEAT_GUI_GTK)
855 return TRUE; // continue receiving notifications
856 # elif defined(FEAT_GUI_MOTIF) || defined(FEAT_GUI_ATHENA)
857 // renew timeout
858 if (mz_threads_allow && p_mzq > 0)
859 timer_id = XtAppAddTimeOut(app_context, p_mzq,
860 timer_proc, NULL);
861 # endif
862 }
863
864 static void
setup_timer(void)865 setup_timer(void)
866 {
867 # if defined(FEAT_GUI_MSWIN)
868 timer_id = SetTimer(NULL, 0, p_mzq, timer_proc);
869 # elif defined(FEAT_GUI_GTK)
870 timer_id = g_timeout_add((guint)p_mzq, (GSourceFunc)timer_proc, NULL);
871 # elif defined(FEAT_GUI_MOTIF) || defined(FEAT_GUI_ATHENA)
872 timer_id = XtAppAddTimeOut(app_context, p_mzq, timer_proc, NULL);
873 # endif
874 }
875
876 static void
remove_timer(void)877 remove_timer(void)
878 {
879 # if defined(FEAT_GUI_MSWIN)
880 KillTimer(NULL, timer_id);
881 # elif defined(FEAT_GUI_GTK)
882 g_source_remove(timer_id);
883 # elif defined(FEAT_GUI_MOTIF) || defined(FEAT_GUI_ATHENA)
884 XtRemoveTimeOut(timer_id);
885 # endif
886 timer_id = 0;
887 }
888
889 void
mzvim_reset_timer(void)890 mzvim_reset_timer(void)
891 {
892 if (timer_id != 0)
893 remove_timer();
894 if (mz_threads_allow && p_mzq > 0 && gui.in_use)
895 setup_timer();
896 }
897
898 #endif // MZSCHEME_GUI_THREADS
899
900 static void
notify_multithread(int on)901 notify_multithread(int on)
902 {
903 mz_threads_allow = on;
904 #ifdef MZSCHEME_GUI_THREADS
905 if (on && timer_id == 0 && p_mzq > 0 && gui.in_use)
906 setup_timer();
907 if (!on && timer_id != 0)
908 remove_timer();
909 #endif
910 }
911
912 void
mzscheme_end(void)913 mzscheme_end(void)
914 {
915 // We can not unload the DLL. Racket's thread might be still alive.
916 #if 0
917 #ifdef DYNAMIC_MZSCHEME
918 dynamic_mzscheme_end();
919 #endif
920 #endif
921 }
922
923 #if HAVE_TLS_SPACE
924 # if defined(_MSC_VER)
925 static __declspec(thread) void *tls_space;
926 extern intptr_t _tls_index;
927 # elif defined(__MINGW32__)
928 static __thread void *tls_space;
929 extern intptr_t _tls_index;
930 # else
931 static THREAD_LOCAL void *tls_space;
932 static intptr_t _tls_index = 0;
933 # endif
934 #endif
935
936 /*
937 * mzscheme_main() is called early in main().
938 * We may call scheme_main_setup() which calls mzscheme_env_main() which then
939 * trampolines into vim_main2(), which never returns.
940 */
941 int
mzscheme_main(void)942 mzscheme_main(void)
943 {
944 int argc = 0;
945 char *argv = NULL;
946
947 #ifdef DYNAMIC_MZSCHEME
948 /*
949 * Racket requires trampolined startup. We can not load it later.
950 * If dynamic dll loading is failed, disable it.
951 */
952 if (!mzscheme_enabled(FALSE))
953 {
954 disabled = TRUE;
955 return vim_main2();
956 }
957 #endif
958 #ifdef HAVE_TLS_SPACE
959 scheme_register_tls_space(&tls_space, _tls_index);
960 #endif
961 #ifdef TRAMPOLINED_MZVIM_STARTUP
962 return scheme_main_setup(TRUE, mzscheme_env_main, argc, &argv);
963 #else
964 return mzscheme_env_main(NULL, argc, &argv);
965 #endif
966 }
967
968 static int
mzscheme_env_main(Scheme_Env * env,int argc UNUSED,char ** argv UNUSED)969 mzscheme_env_main(Scheme_Env *env, int argc UNUSED, char **argv UNUSED)
970 {
971 #ifdef TRAMPOLINED_MZVIM_STARTUP
972 // Scheme has created the environment for us
973 environment = env;
974 #else
975 # ifdef MZ_PRECISE_GC
976 Scheme_Object *dummy = NULL;
977 MZ_GC_DECL_REG(1);
978 MZ_GC_VAR_IN_REG(0, dummy);
979
980 stack_base = &__gc_var_stack__;
981 # else
982 int dummy = 0;
983 stack_base = (void *)&dummy;
984 # endif
985 #endif
986
987 vim_main2();
988 // not reached, vim_main2() will loop until exit()
989
990 return 0;
991 }
992
993 static Scheme_Object*
load_base_module(void * data)994 load_base_module(void *data)
995 {
996 scheme_namespace_require(scheme_intern_symbol((char *)data));
997 return scheme_null;
998 }
999
1000 static Scheme_Object *
load_base_module_on_error(void * data UNUSED)1001 load_base_module_on_error(void *data UNUSED)
1002 {
1003 load_base_module_failed = TRUE;
1004 return scheme_null;
1005 }
1006
1007 static int
startup_mzscheme(void)1008 startup_mzscheme(void)
1009 {
1010 #ifndef TRAMPOLINED_MZVIM_STARTUP
1011 scheme_set_stack_base(stack_base, 1);
1012 #endif
1013
1014 #ifndef TRAMPOLINED_MZVIM_STARTUP
1015 // in newer versions of precise GC the initial env has been created
1016 environment = scheme_basic_env();
1017 #endif
1018
1019 MZ_REGISTER_STATIC(environment);
1020 MZ_REGISTER_STATIC(curout);
1021 MZ_REGISTER_STATIC(curerr);
1022 MZ_REGISTER_STATIC(exn_catching_apply);
1023 MZ_REGISTER_STATIC(exn_p);
1024 MZ_REGISTER_STATIC(exn_message);
1025 MZ_REGISTER_STATIC(vim_exn);
1026
1027 MZ_GC_CHECK();
1028
1029 #ifdef INCLUDE_MZSCHEME_BASE
1030 // invoke function from generated and included mzscheme_base.c
1031 declare_modules(environment);
1032 #endif
1033
1034 // setup 'current-library-collection-paths' parameter
1035 # if MZSCHEME_VERSION_MAJOR >= 299
1036 {
1037 Scheme_Object *coll_path = NULL;
1038 int mustfree = FALSE;
1039 char_u *s;
1040
1041 MZ_GC_DECL_REG(1);
1042 MZ_GC_VAR_IN_REG(0, coll_path);
1043 MZ_GC_REG();
1044 // workaround for dynamic loading on windows
1045 s = vim_getenv((char_u *)"PLTCOLLECTS", &mustfree);
1046 if (s != NULL)
1047 {
1048 coll_path = scheme_make_path((char *)s);
1049 MZ_GC_CHECK();
1050 if (mustfree)
1051 vim_free(s);
1052 }
1053 # ifdef MZSCHEME_COLLECTS
1054 if (coll_path == NULL)
1055 {
1056 coll_path = scheme_make_path(MZSCHEME_COLLECTS);
1057 MZ_GC_CHECK();
1058 }
1059 # endif
1060 if (coll_path != NULL)
1061 {
1062 scheme_set_collects_path(coll_path);
1063 MZ_GC_CHECK();
1064 }
1065 MZ_GC_UNREG();
1066 }
1067 # else
1068 # ifdef MZSCHEME_COLLECTS
1069 {
1070 Scheme_Object *coll_string = NULL;
1071 Scheme_Object *coll_pair = NULL;
1072 Scheme_Config *config = NULL;
1073
1074 MZ_GC_DECL_REG(3);
1075 MZ_GC_VAR_IN_REG(0, coll_string);
1076 MZ_GC_VAR_IN_REG(1, coll_pair);
1077 MZ_GC_VAR_IN_REG(2, config);
1078 MZ_GC_REG();
1079 coll_string = scheme_make_byte_string(MZSCHEME_COLLECTS);
1080 MZ_GC_CHECK();
1081 coll_pair = scheme_make_pair(coll_string, scheme_null);
1082 MZ_GC_CHECK();
1083 config = scheme_current_config();
1084 MZ_GC_CHECK();
1085 scheme_set_param(config, MZCONFIG_COLLECTION_PATHS, coll_pair);
1086 MZ_GC_CHECK();
1087 MZ_GC_UNREG();
1088 }
1089 # endif
1090 #endif
1091
1092 # if MZSCHEME_VERSION_MAJOR >= 600
1093 {
1094 Scheme_Object *config_path = NULL;
1095 int mustfree = FALSE;
1096 char_u *s;
1097
1098 MZ_GC_DECL_REG(1);
1099 MZ_GC_VAR_IN_REG(0, config_path);
1100 MZ_GC_REG();
1101 // workaround for dynamic loading on windows
1102 s = vim_getenv((char_u *)"PLTCONFIGDIR", &mustfree);
1103 if (s != NULL)
1104 {
1105 config_path = scheme_make_path((char *)s);
1106 MZ_GC_CHECK();
1107 if (mustfree)
1108 vim_free(s);
1109 }
1110 #ifdef MZSCHEME_CONFIGDIR
1111 if (config_path == NULL)
1112 {
1113 config_path = scheme_make_path(MZSCHEME_CONFIGDIR);
1114 MZ_GC_CHECK();
1115 }
1116 #endif
1117 if (config_path != NULL)
1118 {
1119 scheme_set_config_path(config_path);
1120 MZ_GC_CHECK();
1121 }
1122 MZ_GC_UNREG();
1123 }
1124 # endif
1125
1126 #if MZSCHEME_VERSION_MAJOR >= 400
1127 scheme_init_collection_paths(environment, scheme_null);
1128 #endif
1129
1130 /*
1131 * versions 4.x do not provide Scheme bindings by default
1132 * we need to add them explicitly
1133 */
1134 {
1135 // use error handler to avoid abort
1136 scheme_dynamic_wind(NULL, load_base_module, NULL,
1137 load_base_module_on_error, "racket/base");
1138 if (load_base_module_failed)
1139 {
1140 load_base_module_failed = FALSE;
1141 scheme_dynamic_wind(NULL, load_base_module, NULL,
1142 load_base_module_on_error, "scheme/base");
1143 if (load_base_module_failed)
1144 return -1;
1145 }
1146 }
1147
1148 register_vim_exn();
1149 // use new environment to initialise exception handling
1150 init_exn_catching_apply();
1151
1152 // redirect output
1153 scheme_console_output = do_output;
1154 scheme_console_printf = do_printf;
1155
1156 #ifdef HAVE_SANDBOX
1157 {
1158 Scheme_Object *make_security_guard = NULL;
1159 MZ_GC_DECL_REG(1);
1160 MZ_GC_VAR_IN_REG(0, make_security_guard);
1161 MZ_GC_REG();
1162
1163 #if MZSCHEME_VERSION_MAJOR < 400
1164 {
1165 Scheme_Object *make_security_guard_symbol = NULL;
1166 MZ_GC_DECL_REG(1);
1167 MZ_GC_VAR_IN_REG(0, make_security_guard_symbol);
1168 MZ_GC_REG();
1169 make_security_guard_symbol = scheme_intern_symbol("make-security-guard");
1170 MZ_GC_CHECK();
1171 make_security_guard = scheme_lookup_global(
1172 make_security_guard_symbol, environment);
1173 MZ_GC_UNREG();
1174 }
1175 #else
1176 make_security_guard = scheme_builtin_value("make-security-guard");
1177 MZ_GC_CHECK();
1178 #endif
1179
1180 // setup sandbox guards
1181 if (make_security_guard != NULL)
1182 {
1183 Scheme_Object *args[3] = {NULL, NULL, NULL};
1184 Scheme_Object *guard = NULL;
1185 Scheme_Config *config = NULL;
1186 MZ_GC_DECL_REG(5);
1187 MZ_GC_ARRAY_VAR_IN_REG(0, args, 3);
1188 MZ_GC_VAR_IN_REG(3, guard);
1189 MZ_GC_VAR_IN_REG(4, config);
1190 MZ_GC_REG();
1191 config = scheme_current_config();
1192 MZ_GC_CHECK();
1193 args[0] = scheme_get_param(config, MZCONFIG_SECURITY_GUARD);
1194 MZ_GC_CHECK();
1195 args[1] = scheme_make_prim_w_arity(sandbox_file_guard,
1196 "sandbox-file-guard", 3, 3);
1197 args[2] = scheme_make_prim_w_arity(sandbox_network_guard,
1198 "sandbox-network-guard", 4, 4);
1199 guard = scheme_apply(make_security_guard, 3, args);
1200 MZ_GC_CHECK();
1201 scheme_set_param(config, MZCONFIG_SECURITY_GUARD, guard);
1202 MZ_GC_CHECK();
1203 MZ_GC_UNREG();
1204 }
1205 MZ_GC_UNREG();
1206 }
1207 #endif
1208 // Create buffer and window types for use in Scheme code
1209 mz_buffer_type = scheme_make_type("<vim-buffer>");
1210 MZ_GC_CHECK();
1211 mz_window_type = scheme_make_type("<vim-window>");
1212 MZ_GC_CHECK();
1213 #ifdef MZ_PRECISE_GC
1214 GC_register_traversers(mz_buffer_type,
1215 buffer_size_proc, buffer_mark_proc, buffer_fixup_proc,
1216 TRUE, TRUE);
1217 GC_register_traversers(mz_window_type,
1218 window_size_proc, window_mark_proc, window_fixup_proc,
1219 TRUE, TRUE);
1220 #endif
1221
1222 make_modules();
1223
1224 /*
1225 * setup callback to receive notifications
1226 * whether thread scheduling is (or not) required
1227 */
1228 scheme_notify_multithread = notify_multithread;
1229
1230 return 0;
1231 }
1232
1233 /*
1234 * This routine is called for each new invocation of MzScheme
1235 * to make sure things are properly initialized.
1236 */
1237 static int
mzscheme_init(void)1238 mzscheme_init(void)
1239 {
1240 if (!initialized)
1241 {
1242 #ifdef DYNAMIC_MZSCHEME
1243 if (disabled || !mzscheme_enabled(TRUE))
1244 {
1245 emsg(_("E815: Sorry, this command is disabled, the MzScheme libraries could not be loaded."));
1246 return -1;
1247 }
1248 #endif
1249 if (load_base_module_failed || startup_mzscheme())
1250 {
1251 emsg(_("E895: Sorry, this command is disabled, the MzScheme's racket/base module could not be loaded."));
1252 return -1;
1253 }
1254 initialized = TRUE;
1255 }
1256 {
1257 Scheme_Config *config = NULL;
1258 MZ_GC_DECL_REG(1);
1259 MZ_GC_VAR_IN_REG(0, config);
1260 MZ_GC_REG();
1261 config = scheme_current_config();
1262 MZ_GC_CHECK();
1263 // recreate ports each call effectively clearing these ones
1264 curout = scheme_make_byte_string_output_port();
1265 MZ_GC_CHECK();
1266 curerr = scheme_make_byte_string_output_port();
1267 MZ_GC_CHECK();
1268 scheme_set_param(config, MZCONFIG_OUTPUT_PORT, curout);
1269 MZ_GC_CHECK();
1270 scheme_set_param(config, MZCONFIG_ERROR_PORT, curerr);
1271 MZ_GC_CHECK();
1272 MZ_GC_UNREG();
1273 }
1274
1275 return 0;
1276 }
1277
1278 /*
1279 *========================================================================
1280 * 2. External Interface
1281 *========================================================================
1282 */
1283
1284 /*
1285 * Evaluate command with exception handling
1286 */
1287 static int
eval_with_exn_handling(void * data,Scheme_Closed_Prim * what,Scheme_Object ** ret)1288 eval_with_exn_handling(void *data, Scheme_Closed_Prim *what, Scheme_Object **ret)
1289 {
1290 Scheme_Object *value = NULL;
1291 Scheme_Object *exn = NULL;
1292 Scheme_Object *prim = NULL;
1293
1294 MZ_GC_DECL_REG(3);
1295 MZ_GC_VAR_IN_REG(0, value);
1296 MZ_GC_VAR_IN_REG(1, exn);
1297 MZ_GC_VAR_IN_REG(2, prim);
1298 MZ_GC_REG();
1299
1300 prim = scheme_make_closed_prim_w_arity(what, data, "mzvim", 0, 0);
1301 MZ_GC_CHECK();
1302 value = _apply_thunk_catch_exceptions(prim, &exn);
1303 MZ_GC_CHECK();
1304
1305 if (!value)
1306 {
1307 value = extract_exn_message(exn);
1308 // Got an exn?
1309 if (value)
1310 {
1311 scheme_display(value, curerr); // Send to stderr-vim
1312 MZ_GC_CHECK();
1313 do_flush();
1314 }
1315 MZ_GC_UNREG();
1316 // `raise' was called on some arbitrary value
1317 return FAIL;
1318 }
1319
1320 if (ret != NULL) // if pointer to retval supported give it up
1321 *ret = value;
1322 // Print any result, as long as it's not a void
1323 else if (!SCHEME_VOIDP(value))
1324 {
1325 scheme_display(value, curout); // Send to stdout-vim
1326 MZ_GC_CHECK();
1327 }
1328
1329 do_flush();
1330 MZ_GC_UNREG();
1331 return OK;
1332 }
1333
1334 /*
1335 * :mzscheme
1336 */
1337 static int
do_mzscheme_command(exarg_T * eap,void * data,Scheme_Closed_Prim * what)1338 do_mzscheme_command(exarg_T *eap, void *data, Scheme_Closed_Prim *what)
1339 {
1340 if (mzscheme_init())
1341 return FAIL;
1342
1343 range_start = eap->line1;
1344 range_end = eap->line2;
1345
1346 return eval_with_exn_handling(data, what, NULL);
1347 }
1348
1349 /*
1350 * Routine called by VIM when deleting a buffer
1351 */
1352 void
mzscheme_buffer_free(buf_T * buf)1353 mzscheme_buffer_free(buf_T *buf)
1354 {
1355 if (buf->b_mzscheme_ref)
1356 {
1357 vim_mz_buffer *bp = NULL;
1358 MZ_GC_DECL_REG(1);
1359 MZ_GC_VAR_IN_REG(0, bp);
1360 MZ_GC_REG();
1361
1362 bp = BUFFER_REF(buf);
1363 bp->buf = INVALID_BUFFER_VALUE;
1364 #ifndef MZ_PRECISE_GC
1365 scheme_gc_ptr_ok(bp);
1366 #else
1367 scheme_free_immobile_box(buf->b_mzscheme_ref);
1368 #endif
1369 buf->b_mzscheme_ref = NULL;
1370 MZ_GC_CHECK();
1371 MZ_GC_UNREG();
1372 }
1373 }
1374
1375 /*
1376 * Routine called by VIM when deleting a Window
1377 */
1378 void
mzscheme_window_free(win_T * win)1379 mzscheme_window_free(win_T *win)
1380 {
1381 if (win->w_mzscheme_ref)
1382 {
1383 vim_mz_window *wp = NULL;
1384 MZ_GC_DECL_REG(1);
1385 MZ_GC_VAR_IN_REG(0, wp);
1386 MZ_GC_REG();
1387 wp = WINDOW_REF(win);
1388 wp->win = INVALID_WINDOW_VALUE;
1389 #ifndef MZ_PRECISE_GC
1390 scheme_gc_ptr_ok(wp);
1391 #else
1392 scheme_free_immobile_box(win->w_mzscheme_ref);
1393 #endif
1394 win->w_mzscheme_ref = NULL;
1395 MZ_GC_CHECK();
1396 MZ_GC_UNREG();
1397 }
1398 }
1399
1400 /*
1401 * ":mzscheme" (or ":mz")
1402 */
1403 void
ex_mzscheme(exarg_T * eap)1404 ex_mzscheme(exarg_T *eap)
1405 {
1406 char_u *script;
1407
1408 script = script_get(eap, eap->arg);
1409 if (!eap->skip)
1410 {
1411 if (script == NULL)
1412 do_mzscheme_command(eap, eap->arg, do_eval);
1413 else
1414 {
1415 do_mzscheme_command(eap, script, do_eval);
1416 vim_free(script);
1417 }
1418 }
1419 }
1420
1421 static Scheme_Object *
do_load(void * data,int noargc UNUSED,Scheme_Object ** noargv UNUSED)1422 do_load(void *data, int noargc UNUSED, Scheme_Object **noargv UNUSED)
1423 {
1424 Scheme_Object *expr = NULL;
1425 Scheme_Object *result = NULL;
1426 char *file = NULL;
1427 Port_Info *pinfo = (Port_Info *)data;
1428
1429 MZ_GC_DECL_REG(3);
1430 MZ_GC_VAR_IN_REG(0, expr);
1431 MZ_GC_VAR_IN_REG(1, result);
1432 MZ_GC_VAR_IN_REG(2, file);
1433 MZ_GC_REG();
1434
1435 file = (char *)scheme_malloc_fail_ok(scheme_malloc_atomic, MAXPATHL + 1);
1436 MZ_GC_CHECK();
1437
1438 // make Vim expansion
1439 expand_env((char_u *)pinfo->name, (char_u *)file, MAXPATHL);
1440 pinfo->port = scheme_open_input_file(file, "mzfile");
1441 MZ_GC_CHECK();
1442 scheme_count_lines(pinfo->port); // to get accurate read error location
1443 MZ_GC_CHECK();
1444
1445 // Like REPL but print only last result
1446 while (!SCHEME_EOFP(expr = scheme_read(pinfo->port)))
1447 {
1448 result = scheme_eval(expr, environment);
1449 MZ_GC_CHECK();
1450 }
1451
1452 // errors will be caught in do_mzscheme_command and ex_mzfile
1453 scheme_close_input_port(pinfo->port);
1454 MZ_GC_CHECK();
1455 pinfo->port = NULL;
1456 MZ_GC_UNREG();
1457 return result;
1458 }
1459
1460 /*
1461 * :mzfile
1462 */
1463 void
ex_mzfile(exarg_T * eap)1464 ex_mzfile(exarg_T *eap)
1465 {
1466 Port_Info pinfo = {NULL, NULL};
1467
1468 MZ_GC_DECL_REG(1);
1469 MZ_GC_VAR_IN_REG(0, pinfo.port);
1470 MZ_GC_REG();
1471
1472 pinfo.name = (char *)eap->arg;
1473 if (do_mzscheme_command(eap, &pinfo, do_load) != OK
1474 && pinfo.port != NULL) // looks like port was not closed
1475 {
1476 scheme_close_input_port(pinfo.port);
1477 MZ_GC_CHECK();
1478 }
1479 MZ_GC_UNREG();
1480 }
1481
1482
1483 /*
1484 *========================================================================
1485 * Exception handling code -- cribbed form the MzScheme sources and
1486 * Matthew Flatt's "Inside PLT MzScheme" document.
1487 *========================================================================
1488 */
1489 static void
init_exn_catching_apply(void)1490 init_exn_catching_apply(void)
1491 {
1492 if (!exn_catching_apply)
1493 {
1494 char *e =
1495 "(lambda (thunk) "
1496 "(with-handlers ([void (lambda (exn) (cons #f exn))]) "
1497 "(cons #t (thunk))))";
1498
1499 exn_catching_apply = scheme_eval_string(e, environment);
1500 MZ_GC_CHECK();
1501 exn_p = scheme_builtin_value("exn?");
1502 MZ_GC_CHECK();
1503 exn_message = scheme_builtin_value("exn-message");
1504 MZ_GC_CHECK();
1505 }
1506 }
1507
1508 /*
1509 * This function applies a thunk, returning the Scheme value if there's
1510 * no exception, otherwise returning NULL and setting *exn to the raised
1511 * value (usually an exn structure).
1512 */
1513 static Scheme_Object *
_apply_thunk_catch_exceptions(Scheme_Object * f,Scheme_Object ** exn)1514 _apply_thunk_catch_exceptions(Scheme_Object *f, Scheme_Object **exn)
1515 {
1516 Scheme_Object *v;
1517
1518 v = _scheme_apply(exn_catching_apply, 1, &f);
1519 // v is a pair: (cons #t value) or (cons #f exn)
1520
1521 if (SCHEME_TRUEP(SCHEME_CAR(v)))
1522 return SCHEME_CDR(v);
1523 else
1524 {
1525 *exn = SCHEME_CDR(v);
1526 return NULL;
1527 }
1528 }
1529
1530 static Scheme_Object *
extract_exn_message(Scheme_Object * v)1531 extract_exn_message(Scheme_Object *v)
1532 {
1533 if (SCHEME_TRUEP(_scheme_apply(exn_p, 1, &v)))
1534 return _scheme_apply(exn_message, 1, &v);
1535 else
1536 return NULL; // Not an exn structure
1537 }
1538
1539 static Scheme_Object *
do_eval(void * s,int noargc UNUSED,Scheme_Object ** noargv UNUSED)1540 do_eval(void *s, int noargc UNUSED, Scheme_Object **noargv UNUSED)
1541 {
1542 return scheme_eval_string_all((char *)s, environment, TRUE);
1543 }
1544
1545 /*
1546 *========================================================================
1547 * 3. MzScheme I/O Handlers
1548 *========================================================================
1549 */
1550 static void
do_intrnl_output(char * mesg,int error)1551 do_intrnl_output(char *mesg, int error)
1552 {
1553 char *p, *prev;
1554
1555 prev = mesg;
1556 p = strchr(prev, '\n');
1557 while (p)
1558 {
1559 *p = '\0';
1560 if (error)
1561 emsg(prev);
1562 else
1563 msg(prev);
1564 prev = p + 1;
1565 p = strchr(prev, '\n');
1566 }
1567
1568 if (error)
1569 emsg(prev);
1570 else
1571 msg(prev);
1572 }
1573
1574 static void
do_output(char * mesg,OUTPUT_LEN_TYPE len UNUSED)1575 do_output(char *mesg, OUTPUT_LEN_TYPE len UNUSED)
1576 {
1577 // TODO: use len, the string may not be NUL terminated
1578 do_intrnl_output(mesg, 0);
1579 }
1580
1581 static void
do_err_output(char * mesg)1582 do_err_output(char *mesg)
1583 {
1584 do_intrnl_output(mesg, 1);
1585 }
1586
1587 static void
do_printf(char * format,...)1588 do_printf(char *format, ...)
1589 {
1590 do_intrnl_output(format, 1);
1591 }
1592
1593 static void
do_flush(void)1594 do_flush(void)
1595 {
1596 char *buff;
1597 OUTPUT_LEN_TYPE length;
1598
1599 buff = scheme_get_sized_byte_string_output(curerr, &length);
1600 MZ_GC_CHECK();
1601 if (length)
1602 {
1603 do_err_output(buff);
1604 return;
1605 }
1606
1607 buff = scheme_get_sized_byte_string_output(curout, &length);
1608 MZ_GC_CHECK();
1609 if (length)
1610 do_output(buff, length);
1611 }
1612
1613 /*
1614 *========================================================================
1615 * 4. Implementation of the Vim Features for MzScheme
1616 *========================================================================
1617 */
1618
1619 /*
1620 * (command {command-string})
1621 */
1622 static Scheme_Object *
vim_command(void * data,int argc,Scheme_Object ** argv)1623 vim_command(void *data, int argc, Scheme_Object **argv)
1624 {
1625 Vim_Prim *prim = (Vim_Prim *)data;
1626 Scheme_Object *cmd = NULL;
1627 MZ_GC_DECL_REG(1);
1628 MZ_GC_VAR_IN_REG(0, cmd);
1629 MZ_GC_REG();
1630 cmd = GUARANTEED_STRING_ARG(prim->name, 0);
1631
1632 // may be use do_cmdline_cmd?
1633 do_cmdline(BYTE_STRING_VALUE(cmd), NULL, NULL, DOCMD_NOWAIT|DOCMD_VERBOSE);
1634 update_screen(VALID);
1635
1636 MZ_GC_UNREG();
1637 raise_if_error();
1638 return scheme_void;
1639 }
1640
1641 /*
1642 * (eval {expr-string})
1643 */
1644 static Scheme_Object *
vim_eval(void * data UNUSED,int argc UNUSED,Scheme_Object ** argv UNUSED)1645 vim_eval(void *data UNUSED, int argc UNUSED, Scheme_Object **argv UNUSED)
1646 {
1647 #ifdef FEAT_EVAL
1648 Vim_Prim *prim = (Vim_Prim *)data;
1649 Scheme_Object *result = NULL;
1650 typval_T *vim_result;
1651 Scheme_Object *expr = NULL;
1652 MZ_GC_DECL_REG(2);
1653 MZ_GC_VAR_IN_REG(0, result);
1654 MZ_GC_VAR_IN_REG(1, expr);
1655 MZ_GC_REG();
1656 expr = GUARANTEED_STRING_ARG(prim->name, 0);
1657
1658 vim_result = eval_expr(BYTE_STRING_VALUE(expr), NULL);
1659
1660 if (vim_result == NULL)
1661 raise_vim_exn(_("invalid expression"));
1662
1663 result = vim_to_mzscheme(vim_result);
1664 MZ_GC_CHECK();
1665 free_tv(vim_result);
1666
1667 MZ_GC_UNREG();
1668 return result;
1669 #else
1670 raise_vim_exn(_("expressions disabled at compile time"));
1671 // unreachable
1672 return scheme_false;
1673 #endif
1674 }
1675
1676 /*
1677 * (range-start)
1678 */
1679 static Scheme_Object *
get_range_start(void * data UNUSED,int argc UNUSED,Scheme_Object ** argv UNUSED)1680 get_range_start(void *data UNUSED, int argc UNUSED, Scheme_Object **argv UNUSED)
1681 {
1682 return scheme_make_integer(range_start);
1683 }
1684
1685 /*
1686 * (range-end)
1687 */
1688 static Scheme_Object *
get_range_end(void * data UNUSED,int argc UNUSED,Scheme_Object ** argv UNUSED)1689 get_range_end(void *data UNUSED, int argc UNUSED, Scheme_Object **argv UNUSED)
1690 {
1691 return scheme_make_integer(range_end);
1692 }
1693
1694 /*
1695 * (beep)
1696 */
1697 static Scheme_Object *
mzscheme_beep(void * data UNUSED,int argc UNUSED,Scheme_Object ** argv UNUSED)1698 mzscheme_beep(void *data UNUSED, int argc UNUSED, Scheme_Object **argv UNUSED)
1699 {
1700 vim_beep(BO_LANG);
1701 return scheme_void;
1702 }
1703
1704 static Scheme_Object *M_global = NULL;
1705
1706 /*
1707 * (get-option {option-name}) [buffer/window]
1708 */
1709 static Scheme_Object *
get_option(void * data,int argc,Scheme_Object ** argv)1710 get_option(void *data, int argc, Scheme_Object **argv)
1711 {
1712 Vim_Prim *prim = (Vim_Prim *)data;
1713 long value;
1714 char *strval;
1715 getoption_T rc;
1716 Scheme_Object *rval = NULL;
1717 Scheme_Object *name = NULL;
1718 int opt_flags = 0;
1719 buf_T *save_curb = curbuf;
1720 win_T *save_curw = curwin;
1721
1722 MZ_GC_DECL_REG(2);
1723 MZ_GC_VAR_IN_REG(0, rval);
1724 MZ_GC_VAR_IN_REG(1, name);
1725 MZ_GC_REG();
1726
1727 name = GUARANTEED_STRING_ARG(prim->name, 0);
1728
1729 if (argc > 1)
1730 {
1731 if (M_global == NULL)
1732 {
1733 MZ_REGISTER_STATIC(M_global);
1734 M_global = scheme_intern_symbol("global");
1735 MZ_GC_CHECK();
1736 }
1737
1738 if (argv[1] == M_global)
1739 opt_flags = OPT_GLOBAL;
1740 else if (SCHEME_VIMBUFFERP(argv[1]))
1741 {
1742 curbuf = get_valid_buffer(argv[1]);
1743 opt_flags = OPT_LOCAL;
1744 }
1745 else if (SCHEME_VIMWINDOWP(argv[1]))
1746 {
1747 win_T *win = get_valid_window(argv[1]);
1748
1749 curwin = win;
1750 curbuf = win->w_buffer;
1751 opt_flags = OPT_LOCAL;
1752 }
1753 else
1754 scheme_wrong_type(prim->name, "vim-buffer/window", 1, argc, argv);
1755 }
1756
1757 rc = get_option_value(BYTE_STRING_VALUE(name), &value, (char_u **)&strval,
1758 opt_flags);
1759 curbuf = save_curb;
1760 curwin = save_curw;
1761
1762 switch (rc)
1763 {
1764 case gov_bool:
1765 case gov_number:
1766 MZ_GC_UNREG();
1767 return scheme_make_integer_value(value);
1768 case gov_string:
1769 rval = scheme_make_byte_string(strval);
1770 MZ_GC_CHECK();
1771 vim_free(strval);
1772 MZ_GC_UNREG();
1773 return rval;
1774 case gov_hidden_bool:
1775 case gov_hidden_number:
1776 case gov_hidden_string:
1777 MZ_GC_UNREG();
1778 raise_vim_exn(_("hidden option"));
1779 //NOTREACHED
1780 case gov_unknown:
1781 MZ_GC_UNREG();
1782 raise_vim_exn(_("unknown option"));
1783 //NOTREACHED
1784 }
1785 // unreachable
1786 return scheme_void;
1787 }
1788
1789 /*
1790 * (set-option {option-changing-string} [buffer/window])
1791 */
1792 static Scheme_Object *
set_option(void * data,int argc,Scheme_Object ** argv)1793 set_option(void *data, int argc, Scheme_Object **argv)
1794 {
1795 char_u *command = NULL;
1796 int opt_flags = 0;
1797 buf_T *save_curb = curbuf;
1798 win_T *save_curw = curwin;
1799 Vim_Prim *prim = (Vim_Prim *)data;
1800 Scheme_Object *cmd = NULL;
1801
1802 MZ_GC_DECL_REG(1);
1803 MZ_GC_VAR_IN_REG(0, cmd);
1804 MZ_GC_REG();
1805 cmd = GUARANTEED_STRING_ARG(prim->name, 0);
1806
1807 if (argc > 1)
1808 {
1809 if (M_global == NULL)
1810 {
1811 MZ_REGISTER_STATIC(M_global);
1812 M_global = scheme_intern_symbol("global");
1813 MZ_GC_CHECK();
1814 }
1815
1816 if (argv[1] == M_global)
1817 opt_flags = OPT_GLOBAL;
1818 else if (SCHEME_VIMBUFFERP(argv[1]))
1819 {
1820 curbuf = get_valid_buffer(argv[1]);
1821 opt_flags = OPT_LOCAL;
1822 }
1823 else if (SCHEME_VIMWINDOWP(argv[1]))
1824 {
1825 win_T *win = get_valid_window(argv[1]);
1826 curwin = win;
1827 curbuf = win->w_buffer;
1828 opt_flags = OPT_LOCAL;
1829 }
1830 else
1831 scheme_wrong_type(prim->name, "vim-buffer/window", 1, argc, argv);
1832 }
1833
1834 // do_set can modify cmd, make copy
1835 command = vim_strsave(BYTE_STRING_VALUE(cmd));
1836 MZ_GC_UNREG();
1837 do_set(command, opt_flags);
1838 vim_free(command);
1839 update_screen(NOT_VALID);
1840 curbuf = save_curb;
1841 curwin = save_curw;
1842 raise_if_error();
1843 return scheme_void;
1844 }
1845
1846 /*
1847 *===========================================================================
1848 * 5. Vim Window-related Manipulation Functions
1849 *===========================================================================
1850 */
1851
1852 /*
1853 * (curr-win)
1854 */
1855 static Scheme_Object *
get_curr_win(void * data UNUSED,int argc UNUSED,Scheme_Object ** argv UNUSED)1856 get_curr_win(void *data UNUSED, int argc UNUSED, Scheme_Object **argv UNUSED)
1857 {
1858 return (Scheme_Object *)get_vim_curr_window();
1859 }
1860
1861 /*
1862 * (win-count)
1863 */
1864 static Scheme_Object *
get_window_count(void * data UNUSED,int argc UNUSED,Scheme_Object ** argv UNUSED)1865 get_window_count(void *data UNUSED, int argc UNUSED, Scheme_Object **argv UNUSED)
1866 {
1867 int n = 0;
1868 win_T *w;
1869
1870 FOR_ALL_WINDOWS(w)
1871 ++n;
1872 return scheme_make_integer(n);
1873 }
1874
1875 /*
1876 * (get-win-list [buffer])
1877 */
1878 static Scheme_Object *
get_window_list(void * data,int argc,Scheme_Object ** argv)1879 get_window_list(void *data, int argc, Scheme_Object **argv)
1880 {
1881 Vim_Prim *prim = (Vim_Prim *)data;
1882 vim_mz_buffer *buf;
1883 Scheme_Object *list;
1884 win_T *w = firstwin;
1885
1886 buf = get_buffer_arg(prim->name, 0, argc, argv);
1887 list = scheme_null;
1888
1889 for ( ; w != NULL; w = w->w_next)
1890 if (w->w_buffer == buf->buf)
1891 {
1892 list = scheme_make_pair(window_new(w), list);
1893 MZ_GC_CHECK();
1894 }
1895
1896 return list;
1897 }
1898
1899 static Scheme_Object *
window_new(win_T * win)1900 window_new(win_T *win)
1901 {
1902 vim_mz_window *self = NULL;
1903
1904 MZ_GC_DECL_REG(1);
1905 MZ_GC_VAR_IN_REG(0, self);
1906
1907 // We need to handle deletion of windows underneath us.
1908 // If we add a "w_mzscheme_ref" field to the win_T structure,
1909 // then we can get at it in win_free() in vim.
1910 //
1911 // On a win_free() we set the Scheme object's win_T *field
1912 // to an invalid value. We trap all uses of a window
1913 // object, and reject them if the win_T *field is invalid.
1914 if (win->w_mzscheme_ref != NULL)
1915 return (Scheme_Object *)WINDOW_REF(win);
1916
1917 MZ_GC_REG();
1918 self = scheme_malloc_fail_ok(scheme_malloc_tagged, sizeof(vim_mz_window));
1919 CLEAR_POINTER(self);
1920 #ifndef MZ_PRECISE_GC
1921 scheme_dont_gc_ptr(self); // because win isn't visible to GC
1922 #else
1923 win->w_mzscheme_ref = scheme_malloc_immobile_box(NULL);
1924 #endif
1925 MZ_GC_CHECK();
1926 WINDOW_REF(win) = self;
1927 MZ_GC_CHECK();
1928 self->win = win;
1929 self->so.type = mz_window_type;
1930
1931 MZ_GC_UNREG();
1932 return (Scheme_Object *)self;
1933 }
1934
1935 /*
1936 * (get-win-num [window])
1937 */
1938 static Scheme_Object *
get_window_num(void * data UNUSED,int argc UNUSED,Scheme_Object ** argv UNUSED)1939 get_window_num(void *data UNUSED, int argc UNUSED, Scheme_Object **argv UNUSED)
1940 {
1941 int nr = 1;
1942 Vim_Prim *prim = (Vim_Prim *)data;
1943 win_T *win = get_window_arg(prim->name, 0, argc, argv)->win;
1944 win_T *wp;
1945
1946 for (wp = firstwin; wp != win; wp = wp->w_next)
1947 ++nr;
1948
1949 return scheme_make_integer(nr);
1950 }
1951
1952 /*
1953 * (get-win-by-num {windownum})
1954 */
1955 static Scheme_Object *
get_window_by_num(void * data,int argc,Scheme_Object ** argv)1956 get_window_by_num(void *data, int argc, Scheme_Object **argv)
1957 {
1958 Vim_Prim *prim = (Vim_Prim *)data;
1959 win_T *win = firstwin;
1960 int fnum;
1961
1962 fnum = SCHEME_INT_VAL(GUARANTEE_INTEGER(prim->name, 0));
1963 if (fnum < 1)
1964 scheme_signal_error(_("window index is out of range"));
1965
1966 for ( ; win != NULL; win = win->w_next, --fnum)
1967 if (fnum == 1) // to be 1-based
1968 return window_new(win);
1969
1970 return scheme_false;
1971 }
1972
1973 /*
1974 * (get-win-buffer [window])
1975 */
1976 static Scheme_Object *
get_window_buffer(void * data,int argc,Scheme_Object ** argv)1977 get_window_buffer(void *data, int argc, Scheme_Object **argv)
1978 {
1979 Vim_Prim *prim = (Vim_Prim *)data;
1980 vim_mz_window *win = get_window_arg(prim->name, 0, argc, argv);
1981
1982 return buffer_new(win->win->w_buffer);
1983 }
1984
1985 /*
1986 * (get-win-height [window])
1987 */
1988 static Scheme_Object *
get_window_height(void * data,int argc,Scheme_Object ** argv)1989 get_window_height(void *data, int argc, Scheme_Object **argv)
1990 {
1991 Vim_Prim *prim = (Vim_Prim *)data;
1992 vim_mz_window *win = get_window_arg(prim->name, 0, argc, argv);
1993
1994 return scheme_make_integer(win->win->w_height);
1995 }
1996
1997 /*
1998 * (set-win-height {height} [window])
1999 */
2000 static Scheme_Object *
set_window_height(void * data,int argc,Scheme_Object ** argv)2001 set_window_height(void *data, int argc, Scheme_Object **argv)
2002 {
2003 Vim_Prim *prim = (Vim_Prim *)data;
2004 vim_mz_window *win;
2005 win_T *savewin;
2006 int height;
2007
2008 win = get_window_arg(prim->name, 1, argc, argv);
2009 height = SCHEME_INT_VAL(GUARANTEE_INTEGER(prim->name, 0));
2010
2011 #ifdef FEAT_GUI
2012 need_mouse_correct = TRUE;
2013 #endif
2014
2015 savewin = curwin;
2016 curwin = win->win;
2017 win_setheight(height);
2018 curwin = savewin;
2019
2020 raise_if_error();
2021 return scheme_void;
2022 }
2023
2024 /*
2025 * (get-win-width [window])
2026 */
2027 static Scheme_Object *
get_window_width(void * data,int argc,Scheme_Object ** argv)2028 get_window_width(void *data, int argc, Scheme_Object **argv)
2029 {
2030 Vim_Prim *prim = (Vim_Prim *)data;
2031 vim_mz_window *win = get_window_arg(prim->name, 0, argc, argv);
2032
2033 return scheme_make_integer(win->win->w_width);
2034 }
2035
2036 /*
2037 * (set-win-width {width} [window])
2038 */
2039 static Scheme_Object *
set_window_width(void * data,int argc,Scheme_Object ** argv)2040 set_window_width(void *data, int argc, Scheme_Object **argv)
2041 {
2042 Vim_Prim *prim = (Vim_Prim *)data;
2043 vim_mz_window *win;
2044 win_T *savewin;
2045 int width = 0;
2046
2047 win = get_window_arg(prim->name, 1, argc, argv);
2048 width = SCHEME_INT_VAL(GUARANTEE_INTEGER(prim->name, 0));
2049
2050 # ifdef FEAT_GUI
2051 need_mouse_correct = TRUE;
2052 # endif
2053
2054 savewin = curwin;
2055 curwin = win->win;
2056 win_setwidth(width);
2057 curwin = savewin;
2058
2059 raise_if_error();
2060 return scheme_void;
2061 }
2062
2063 /*
2064 * (get-cursor [window]) -> (line . col)
2065 */
2066 static Scheme_Object *
get_cursor(void * data,int argc,Scheme_Object ** argv)2067 get_cursor(void *data, int argc, Scheme_Object **argv)
2068 {
2069 Vim_Prim *prim = (Vim_Prim *)data;
2070 vim_mz_window *win;
2071 pos_T pos;
2072
2073 win = get_window_arg(prim->name, 0, argc, argv);
2074 pos = win->win->w_cursor;
2075 return scheme_make_pair(scheme_make_integer_value((long)pos.lnum),
2076 scheme_make_integer_value((long)pos.col + 1));
2077 }
2078
2079 /*
2080 * (set-cursor (line . col) [window])
2081 */
2082 static Scheme_Object *
set_cursor(void * data,int argc,Scheme_Object ** argv)2083 set_cursor(void *data, int argc, Scheme_Object **argv)
2084 {
2085 Vim_Prim *prim = (Vim_Prim *)data;
2086 vim_mz_window *win;
2087 long lnum = 0;
2088 long col = 0;
2089
2090 #ifdef HAVE_SANDBOX
2091 sandbox_check();
2092 #endif
2093 win = get_window_arg(prim->name, 1, argc, argv);
2094 GUARANTEE_PAIR(prim->name, 0);
2095
2096 if (!SCHEME_INTP(SCHEME_CAR(argv[0]))
2097 || !SCHEME_INTP(SCHEME_CDR(argv[0])))
2098 scheme_wrong_type(prim->name, "integer pair", 0, argc, argv);
2099
2100 lnum = SCHEME_INT_VAL(SCHEME_CAR(argv[0]));
2101 col = SCHEME_INT_VAL(SCHEME_CDR(argv[0])) - 1;
2102
2103 check_line_range(lnum, win->win->w_buffer);
2104 // don't know how to catch invalid column value
2105
2106 win->win->w_cursor.lnum = lnum;
2107 win->win->w_cursor.col = col;
2108 win->win->w_set_curswant = TRUE;
2109 update_screen(VALID);
2110
2111 raise_if_error();
2112 return scheme_void;
2113 }
2114 /*
2115 *===========================================================================
2116 * 6. Vim Buffer-related Manipulation Functions
2117 *===========================================================================
2118 */
2119
2120 /*
2121 * (open-buff {filename})
2122 */
2123 static Scheme_Object *
mzscheme_open_buffer(void * data,int argc,Scheme_Object ** argv)2124 mzscheme_open_buffer(void *data, int argc, Scheme_Object **argv)
2125 {
2126 Vim_Prim *prim = (Vim_Prim *)data;
2127 int num = 0;
2128 Scheme_Object *onum = NULL;
2129 Scheme_Object *buf = NULL;
2130 Scheme_Object *fname;
2131
2132 MZ_GC_DECL_REG(3);
2133 MZ_GC_VAR_IN_REG(0, onum);
2134 MZ_GC_VAR_IN_REG(1, buf);
2135 MZ_GC_VAR_IN_REG(2, fname);
2136 MZ_GC_REG();
2137 fname = GUARANTEED_STRING_ARG(prim->name, 0);
2138
2139 #ifdef HAVE_SANDBOX
2140 sandbox_check();
2141 #endif
2142 // TODO make open existing file
2143 num = buflist_add(BYTE_STRING_VALUE(fname), BLN_LISTED | BLN_CURBUF);
2144
2145 if (num == 0)
2146 raise_vim_exn(_("couldn't open buffer"));
2147
2148 onum = scheme_make_integer(num);
2149 buf = get_buffer_by_num(data, 1, &onum);
2150 MZ_GC_UNREG();
2151 return buf;
2152 }
2153
2154 /*
2155 * (get-buff-by-num {buffernum})
2156 */
2157 static Scheme_Object *
get_buffer_by_num(void * data,int argc,Scheme_Object ** argv)2158 get_buffer_by_num(void *data, int argc, Scheme_Object **argv)
2159 {
2160 Vim_Prim *prim = (Vim_Prim *)data;
2161 buf_T *buf;
2162 int fnum;
2163
2164 fnum = SCHEME_INT_VAL(GUARANTEE_INTEGER(prim->name, 0));
2165
2166 FOR_ALL_BUFFERS(buf)
2167 if (buf->b_fnum == fnum)
2168 return buffer_new(buf);
2169
2170 return scheme_false;
2171 }
2172
2173 /*
2174 * (get-buff-by-name {buffername})
2175 */
2176 static Scheme_Object *
get_buffer_by_name(void * data,int argc,Scheme_Object ** argv)2177 get_buffer_by_name(void *data, int argc, Scheme_Object **argv)
2178 {
2179 Vim_Prim *prim = (Vim_Prim *)data;
2180 buf_T *buf;
2181 Scheme_Object *buffer = NULL;
2182 Scheme_Object *fname = NULL;
2183
2184 MZ_GC_DECL_REG(2);
2185 MZ_GC_VAR_IN_REG(0, buffer);
2186 MZ_GC_VAR_IN_REG(1, fname);
2187 MZ_GC_REG();
2188 fname = GUARANTEED_STRING_ARG(prim->name, 0);
2189 buffer = scheme_false;
2190
2191 FOR_ALL_BUFFERS(buf)
2192 {
2193 if (buf->b_ffname == NULL || buf->b_sfname == NULL)
2194 // empty string
2195 {
2196 if (BYTE_STRING_VALUE(fname)[0] == NUL)
2197 buffer = buffer_new(buf);
2198 }
2199 else if (!fnamecmp(buf->b_ffname, BYTE_STRING_VALUE(fname))
2200 || !fnamecmp(buf->b_sfname, BYTE_STRING_VALUE(fname)))
2201 {
2202 // either short or long filename matches
2203 buffer = buffer_new(buf);
2204 }
2205 }
2206
2207 MZ_GC_UNREG();
2208 return buffer;
2209 }
2210
2211 /*
2212 * (get-next-buff [buffer])
2213 */
2214 static Scheme_Object *
get_next_buffer(void * data,int argc,Scheme_Object ** argv)2215 get_next_buffer(void *data, int argc, Scheme_Object **argv)
2216 {
2217 Vim_Prim *prim = (Vim_Prim *)data;
2218 buf_T *buf = get_buffer_arg(prim->name, 0, argc, argv)->buf;
2219
2220 if (buf->b_next == NULL)
2221 return scheme_false;
2222 else
2223 return buffer_new(buf->b_next);
2224 }
2225
2226 /*
2227 * (get-prev-buff [buffer])
2228 */
2229 static Scheme_Object *
get_prev_buffer(void * data,int argc,Scheme_Object ** argv)2230 get_prev_buffer(void *data, int argc, Scheme_Object **argv)
2231 {
2232 Vim_Prim *prim = (Vim_Prim *)data;
2233 buf_T *buf = get_buffer_arg(prim->name, 0, argc, argv)->buf;
2234
2235 if (buf->b_prev == NULL)
2236 return scheme_false;
2237 else
2238 return buffer_new(buf->b_prev);
2239 }
2240
2241 /*
2242 * (get-buff-num [buffer])
2243 */
2244 static Scheme_Object *
get_buffer_num(void * data,int argc,Scheme_Object ** argv)2245 get_buffer_num(void *data, int argc, Scheme_Object **argv)
2246 {
2247 Vim_Prim *prim = (Vim_Prim *)data;
2248 vim_mz_buffer *buf = get_buffer_arg(prim->name, 0, argc, argv);
2249
2250 return scheme_make_integer(buf->buf->b_fnum);
2251 }
2252
2253 /*
2254 * (buff-count)
2255 */
2256 static Scheme_Object *
get_buffer_count(void * data UNUSED,int argc UNUSED,Scheme_Object ** argv UNUSED)2257 get_buffer_count(void *data UNUSED, int argc UNUSED, Scheme_Object **argv UNUSED)
2258 {
2259 buf_T *b;
2260 int n = 0;
2261
2262 FOR_ALL_BUFFERS(b) ++n;
2263 return scheme_make_integer(n);
2264 }
2265
2266 /*
2267 * (get-buff-name [buffer])
2268 */
2269 static Scheme_Object *
get_buffer_name(void * data,int argc,Scheme_Object ** argv)2270 get_buffer_name(void *data, int argc, Scheme_Object **argv)
2271 {
2272 Vim_Prim *prim = (Vim_Prim *)data;
2273 vim_mz_buffer *buf = get_buffer_arg(prim->name, 0, argc, argv);
2274
2275 return scheme_make_byte_string((char *)buf->buf->b_ffname);
2276 }
2277
2278 /*
2279 * (curr-buff)
2280 */
2281 static Scheme_Object *
get_curr_buffer(void * data UNUSED,int argc UNUSED,Scheme_Object ** argv UNUSED)2282 get_curr_buffer(void *data UNUSED, int argc UNUSED, Scheme_Object **argv UNUSED)
2283 {
2284 return (Scheme_Object *)get_vim_curr_buffer();
2285 }
2286
2287 static Scheme_Object *
buffer_new(buf_T * buf)2288 buffer_new(buf_T *buf)
2289 {
2290 vim_mz_buffer *self = NULL;
2291
2292 MZ_GC_DECL_REG(1);
2293 MZ_GC_VAR_IN_REG(0, self);
2294
2295 // We need to handle deletion of buffers underneath us.
2296 // If we add a "b_mzscheme_ref" field to the buf_T structure,
2297 // then we can get at it in buf_freeall() in vim.
2298 if (buf->b_mzscheme_ref)
2299 return (Scheme_Object *)BUFFER_REF(buf);
2300
2301 MZ_GC_REG();
2302 self = scheme_malloc_fail_ok(scheme_malloc_tagged, sizeof(vim_mz_buffer));
2303 CLEAR_POINTER(self);
2304 #ifndef MZ_PRECISE_GC
2305 scheme_dont_gc_ptr(self); // because buf isn't visible to GC
2306 #else
2307 buf->b_mzscheme_ref = scheme_malloc_immobile_box(NULL);
2308 #endif
2309 MZ_GC_CHECK();
2310 BUFFER_REF(buf) = self;
2311 MZ_GC_CHECK();
2312 self->buf = buf;
2313 self->so.type = mz_buffer_type;
2314
2315 MZ_GC_UNREG();
2316 return (Scheme_Object *)self;
2317 }
2318
2319 /*
2320 * (get-buff-size [buffer])
2321 *
2322 * Get the size (number of lines) in the current buffer.
2323 */
2324 static Scheme_Object *
get_buffer_size(void * data,int argc,Scheme_Object ** argv)2325 get_buffer_size(void *data, int argc, Scheme_Object **argv)
2326 {
2327 Vim_Prim *prim = (Vim_Prim *)data;
2328 vim_mz_buffer *buf = get_buffer_arg(prim->name, 0, argc, argv);
2329
2330 return scheme_make_integer(buf->buf->b_ml.ml_line_count);
2331 }
2332
2333 /*
2334 * (get-buff-line {linenr} [buffer])
2335 *
2336 * Get a line from the specified buffer. The line number is
2337 * in Vim format (1-based). The line is returned as a MzScheme
2338 * string object.
2339 */
2340 static Scheme_Object *
get_buffer_line(void * data,int argc,Scheme_Object ** argv)2341 get_buffer_line(void *data, int argc, Scheme_Object **argv)
2342 {
2343 Vim_Prim *prim = (Vim_Prim *)data;
2344 vim_mz_buffer *buf;
2345 int linenr;
2346 char_u *line;
2347
2348 buf = get_buffer_arg(prim->name, 1, argc, argv);
2349 linenr = SCHEME_INT_VAL(GUARANTEE_INTEGER(prim->name, 0));
2350 line = ml_get_buf(buf->buf, (linenr_T)linenr, FALSE);
2351
2352 raise_if_error();
2353 return scheme_make_byte_string((char *)line);
2354 }
2355
2356
2357 /*
2358 * (get-buff-line-list {start} {end} [buffer])
2359 *
2360 * Get a list of lines from the specified buffer. The line numbers
2361 * are in Vim format (1-based). The range is from lo up to, but not
2362 * including, hi. The list is returned as a list of string objects.
2363 */
2364 static Scheme_Object *
get_buffer_line_list(void * data,int argc,Scheme_Object ** argv)2365 get_buffer_line_list(void *data, int argc, Scheme_Object **argv)
2366 {
2367 Vim_Prim *prim = (Vim_Prim *)data;
2368 vim_mz_buffer *buf;
2369 int i, hi, lo, n;
2370 Scheme_Object *list = NULL;
2371
2372 MZ_GC_DECL_REG(1);
2373 MZ_GC_VAR_IN_REG(0, list);
2374 MZ_GC_REG();
2375
2376 buf = get_buffer_arg(prim->name, 2, argc, argv);
2377 list = scheme_null;
2378 hi = SCHEME_INT_VAL(GUARANTEE_INTEGER(prim->name, 1));
2379 lo = SCHEME_INT_VAL(GUARANTEE_INTEGER(prim->name, 0));
2380
2381 /*
2382 * Handle some error conditions
2383 */
2384 if (lo < 0)
2385 lo = 0;
2386
2387 if (hi < 0)
2388 hi = 0;
2389 if (hi < lo)
2390 hi = lo;
2391
2392 n = hi - lo;
2393
2394 for (i = n; i >= 0; --i)
2395 {
2396 Scheme_Object *str = scheme_make_byte_string(
2397 (char *)ml_get_buf(buf->buf, (linenr_T)(lo+i), FALSE));
2398 raise_if_error();
2399
2400 // Set the list item
2401 list = scheme_make_pair(str, list);
2402 MZ_GC_CHECK();
2403 }
2404 MZ_GC_UNREG();
2405 return list;
2406 }
2407
2408 /*
2409 * (set-buff-line {linenr} {string/#f} [buffer])
2410 *
2411 * Replace a line in the specified buffer. The line number is
2412 * in Vim format (1-based). The replacement line is given as
2413 * an MzScheme string object. The object is checked for validity
2414 * and correct format. An exception is thrown if the values are not
2415 * the correct format.
2416 *
2417 * It returns a Scheme Object that indicates the length of the
2418 * string changed.
2419 */
2420 static Scheme_Object *
set_buffer_line(void * data,int argc,Scheme_Object ** argv)2421 set_buffer_line(void *data, int argc, Scheme_Object **argv)
2422 {
2423 // First of all, we check the value of the supplied MzScheme object.
2424 // There are three cases:
2425 // 1. #f - this is a deletion.
2426 // 2. A string - this is a replacement.
2427 // 3. Anything else - this is an error.
2428 Vim_Prim *prim = (Vim_Prim *)data;
2429 vim_mz_buffer *buf;
2430 Scheme_Object *line = NULL;
2431 char *save;
2432 int n;
2433
2434 MZ_GC_DECL_REG(1);
2435 MZ_GC_VAR_IN_REG(0, line);
2436 MZ_GC_REG();
2437
2438 #ifdef HAVE_SANDBOX
2439 sandbox_check();
2440 #endif
2441 n = SCHEME_INT_VAL(GUARANTEE_INTEGER(prim->name, 0));
2442 if (!SCHEME_STRINGP(argv[1]) && !SCHEME_FALSEP(argv[1]))
2443 scheme_wrong_type(prim->name, "string or #f", 1, argc, argv);
2444 line = argv[1];
2445 buf = get_buffer_arg(prim->name, 2, argc, argv);
2446
2447 check_line_range(n, buf->buf);
2448
2449 if (SCHEME_FALSEP(line))
2450 {
2451 buf_T *savebuf = curbuf;
2452
2453 curbuf = buf->buf;
2454
2455 if (u_savedel((linenr_T)n, 1L) == FAIL)
2456 {
2457 curbuf = savebuf;
2458 raise_vim_exn(_("cannot save undo information"));
2459 }
2460 else if (ml_delete((linenr_T)n) == FAIL)
2461 {
2462 curbuf = savebuf;
2463 raise_vim_exn(_("cannot delete line"));
2464 }
2465 if (buf->buf == curwin->w_buffer)
2466 mz_fix_cursor(n, n + 1, -1);
2467 deleted_lines_mark((linenr_T)n, 1L);
2468
2469 curbuf = savebuf;
2470
2471 MZ_GC_UNREG();
2472 raise_if_error();
2473 return scheme_void;
2474 }
2475 else
2476 {
2477 // Otherwise it's a line
2478 buf_T *savebuf = curbuf;
2479
2480 save = string_to_line(line);
2481
2482 curbuf = buf->buf;
2483
2484 if (u_savesub((linenr_T)n) == FAIL)
2485 {
2486 curbuf = savebuf;
2487 vim_free(save);
2488 raise_vim_exn(_("cannot save undo information"));
2489 }
2490 else if (ml_replace((linenr_T)n, (char_u *)save, TRUE) == FAIL)
2491 {
2492 curbuf = savebuf;
2493 vim_free(save);
2494 raise_vim_exn(_("cannot replace line"));
2495 }
2496 else
2497 {
2498 vim_free(save);
2499 changed_bytes((linenr_T)n, 0);
2500 }
2501
2502 curbuf = savebuf;
2503
2504 // Check that the cursor is not beyond the end of the line now.
2505 if (buf->buf == curwin->w_buffer)
2506 check_cursor_col();
2507
2508 MZ_GC_UNREG();
2509 raise_if_error();
2510 return scheme_void;
2511 }
2512 }
2513
2514 static void
free_array(char ** array)2515 free_array(char **array)
2516 {
2517 char **curr = array;
2518 while (*curr != NULL)
2519 vim_free(*curr++);
2520 vim_free(array);
2521 }
2522
2523 /*
2524 * (set-buff-line-list {start} {end} {string-list/#f/null} [buffer])
2525 *
2526 * Replace a range of lines in the specified buffer. The line numbers are in
2527 * Vim format (1-based). The range is from lo up to, but not including, hi.
2528 * The replacement lines are given as a Scheme list of string objects. The
2529 * list is checked for validity and correct format.
2530 *
2531 * Errors are returned as a value of FAIL. The return value is OK on success.
2532 * If OK is returned and len_change is not NULL, *len_change is set to the
2533 * change in the buffer length.
2534 */
2535 static Scheme_Object *
set_buffer_line_list(void * data,int argc,Scheme_Object ** argv)2536 set_buffer_line_list(void *data, int argc, Scheme_Object **argv)
2537 {
2538 // First of all, we check the type of the supplied MzScheme object.
2539 // There are three cases:
2540 // 1. #f - this is a deletion.
2541 // 2. A list - this is a replacement.
2542 // 3. Anything else - this is an error.
2543 Vim_Prim *prim = (Vim_Prim *)data;
2544 vim_mz_buffer *buf = NULL;
2545 Scheme_Object *line_list = NULL;
2546 int i, old_len, new_len, hi, lo;
2547 long extra;
2548
2549 MZ_GC_DECL_REG(1);
2550 MZ_GC_VAR_IN_REG(0, line_list);
2551 MZ_GC_REG();
2552
2553 #ifdef HAVE_SANDBOX
2554 sandbox_check();
2555 #endif
2556 lo = SCHEME_INT_VAL(GUARANTEE_INTEGER(prim->name, 0));
2557 hi = SCHEME_INT_VAL(GUARANTEE_INTEGER(prim->name, 1));
2558 if (!SCHEME_PAIRP(argv[2])
2559 && !SCHEME_FALSEP(argv[2]) && !SCHEME_NULLP(argv[2]))
2560 scheme_wrong_type(prim->name, "list or #f", 2, argc, argv);
2561 line_list = argv[2];
2562 buf = get_buffer_arg(prim->name, 3, argc, argv);
2563 old_len = hi - lo;
2564 if (old_len < 0) // process inverse values wisely
2565 {
2566 i = lo;
2567 lo = hi;
2568 hi = i;
2569 old_len = -old_len;
2570 }
2571 extra = 0;
2572
2573 check_line_range(lo, buf->buf); // inclusive
2574 check_line_range(hi - 1, buf->buf); // exclusive
2575
2576 if (SCHEME_FALSEP(line_list) || SCHEME_NULLP(line_list))
2577 {
2578 buf_T *savebuf = curbuf;
2579 curbuf = buf->buf;
2580
2581 if (u_savedel((linenr_T)lo, (long)old_len) == FAIL)
2582 {
2583 curbuf = savebuf;
2584 raise_vim_exn(_("cannot save undo information"));
2585 }
2586 else
2587 {
2588 for (i = 0; i < old_len; i++)
2589 if (ml_delete((linenr_T)lo) == FAIL)
2590 {
2591 curbuf = savebuf;
2592 raise_vim_exn(_("cannot delete line"));
2593 }
2594 if (buf->buf == curwin->w_buffer)
2595 mz_fix_cursor(lo, hi, -old_len);
2596 deleted_lines_mark((linenr_T)lo, (long)old_len);
2597 }
2598
2599 curbuf = savebuf;
2600
2601 MZ_GC_UNREG();
2602 raise_if_error();
2603 return scheme_void;
2604 }
2605 else
2606 {
2607 buf_T *savebuf = curbuf;
2608
2609 // List
2610 new_len = scheme_proper_list_length(line_list);
2611 MZ_GC_CHECK();
2612 if (new_len < 0) // improper or cyclic list
2613 scheme_wrong_type(prim->name, "proper list",
2614 2, argc, argv);
2615 else
2616 {
2617 char **array = NULL;
2618 Scheme_Object *line = NULL;
2619 Scheme_Object *rest = NULL;
2620
2621 MZ_GC_DECL_REG(2);
2622 MZ_GC_VAR_IN_REG(0, line);
2623 MZ_GC_VAR_IN_REG(1, rest);
2624 MZ_GC_REG();
2625
2626 array = ALLOC_CLEAR_MULT(char *, new_len + 1);
2627
2628 rest = line_list;
2629 for (i = 0; i < new_len; ++i)
2630 {
2631 line = SCHEME_CAR(rest);
2632 rest = SCHEME_CDR(rest);
2633 if (!SCHEME_STRINGP(line))
2634 {
2635 free_array(array);
2636 scheme_wrong_type(prim->name, "string-list", 2, argc, argv);
2637 }
2638 array[i] = string_to_line(line);
2639 }
2640
2641 curbuf = buf->buf;
2642
2643 if (u_save((linenr_T)(lo-1), (linenr_T)hi) == FAIL)
2644 {
2645 curbuf = savebuf;
2646 free_array(array);
2647 raise_vim_exn(_("cannot save undo information"));
2648 }
2649
2650 /*
2651 * If the size of the range is reducing (ie, new_len < old_len) we
2652 * need to delete some old_len. We do this at the start, by
2653 * repeatedly deleting line "lo".
2654 */
2655 for (i = 0; i < old_len - new_len; ++i)
2656 {
2657 if (ml_delete((linenr_T)lo) == FAIL)
2658 {
2659 curbuf = savebuf;
2660 free_array(array);
2661 raise_vim_exn(_("cannot delete line"));
2662 }
2663 extra--;
2664 }
2665
2666 /*
2667 * For as long as possible, replace the existing old_len with the
2668 * new old_len. This is a more efficient operation, as it requires
2669 * less memory allocation and freeing.
2670 */
2671 for (i = 0; i < old_len && i < new_len; i++)
2672 if (ml_replace((linenr_T)(lo+i), (char_u *)array[i], TRUE) == FAIL)
2673 {
2674 curbuf = savebuf;
2675 free_array(array);
2676 raise_vim_exn(_("cannot replace line"));
2677 }
2678
2679 /*
2680 * Now we may need to insert the remaining new_len. We don't need to
2681 * free the string passed back because MzScheme has control of that
2682 * memory.
2683 */
2684 while (i < new_len)
2685 {
2686 if (ml_append((linenr_T)(lo + i - 1),
2687 (char_u *)array[i], 0, FALSE) == FAIL)
2688 {
2689 curbuf = savebuf;
2690 free_array(array);
2691 raise_vim_exn(_("cannot insert line"));
2692 }
2693 ++i;
2694 ++extra;
2695 }
2696 MZ_GC_UNREG();
2697 free_array(array);
2698 }
2699
2700 /*
2701 * Adjust marks. Invalidate any which lie in the
2702 * changed range, and move any in the remainder of the buffer.
2703 */
2704 mark_adjust((linenr_T)lo, (linenr_T)(hi - 1),
2705 (long)MAXLNUM, (long)extra);
2706 changed_lines((linenr_T)lo, 0, (linenr_T)hi, (long)extra);
2707
2708 if (buf->buf == curwin->w_buffer)
2709 mz_fix_cursor(lo, hi, extra);
2710 curbuf = savebuf;
2711
2712 MZ_GC_UNREG();
2713 raise_if_error();
2714 return scheme_void;
2715 }
2716 }
2717
2718 /*
2719 * (insert-buff-line-list {linenr} {string/string-list} [buffer])
2720 *
2721 * Insert a number of lines into the specified buffer after the specified line.
2722 * The line number is in Vim format (1-based). The lines to be inserted are
2723 * given as an MzScheme list of string objects or as a single string. The lines
2724 * to be added are checked for validity and correct format. Errors are
2725 * returned as a value of FAIL. The return value is OK on success.
2726 * If OK is returned and len_change is not NULL, *len_change
2727 * is set to the change in the buffer length.
2728 */
2729 static Scheme_Object *
insert_buffer_line_list(void * data,int argc,Scheme_Object ** argv)2730 insert_buffer_line_list(void *data, int argc, Scheme_Object **argv)
2731 {
2732 Vim_Prim *prim = (Vim_Prim *)data;
2733 vim_mz_buffer *buf = NULL;
2734 Scheme_Object *list = NULL;
2735 char *str = NULL;
2736 int i, n, size;
2737
2738 MZ_GC_DECL_REG(1);
2739 MZ_GC_VAR_IN_REG(0, list);
2740 MZ_GC_REG();
2741
2742 #ifdef HAVE_SANDBOX
2743 sandbox_check();
2744 #endif
2745 /*
2746 * First of all, we check the type of the supplied MzScheme object.
2747 * It must be a string or a list, or the call is in error.
2748 */
2749 n = SCHEME_INT_VAL(GUARANTEE_INTEGER(prim->name, 0));
2750 list = argv[1];
2751
2752 if (!SCHEME_STRINGP(list) && !SCHEME_PAIRP(list))
2753 scheme_wrong_type(prim->name, "string or list", 1, argc, argv);
2754 buf = get_buffer_arg(prim->name, 2, argc, argv);
2755
2756 if (n != 0) // 0 can be used in insert
2757 check_line_range(n, buf->buf);
2758 if (SCHEME_STRINGP(list))
2759 {
2760 buf_T *savebuf = curbuf;
2761
2762 str = string_to_line(list);
2763 curbuf = buf->buf;
2764
2765 if (u_save((linenr_T)n, (linenr_T)(n+1)) == FAIL)
2766 {
2767 curbuf = savebuf;
2768 vim_free(str);
2769 raise_vim_exn(_("cannot save undo information"));
2770 }
2771 else if (ml_append((linenr_T)n, (char_u *)str, 0, FALSE) == FAIL)
2772 {
2773 curbuf = savebuf;
2774 vim_free(str);
2775 raise_vim_exn(_("cannot insert line"));
2776 }
2777 else
2778 {
2779 vim_free(str);
2780 appended_lines_mark((linenr_T)n, 1L);
2781 }
2782
2783 curbuf = savebuf;
2784 update_screen(VALID);
2785
2786 MZ_GC_UNREG();
2787 raise_if_error();
2788 return scheme_void;
2789 }
2790
2791 // List
2792 size = scheme_proper_list_length(list);
2793 MZ_GC_CHECK();
2794 if (size < 0) // improper or cyclic list
2795 scheme_wrong_type(prim->name, "proper list",
2796 2, argc, argv);
2797 else
2798 {
2799 Scheme_Object *line = NULL;
2800 Scheme_Object *rest = NULL;
2801 char **array;
2802 buf_T *savebuf = curbuf;
2803
2804 MZ_GC_DECL_REG(2);
2805 MZ_GC_VAR_IN_REG(0, line);
2806 MZ_GC_VAR_IN_REG(1, rest);
2807 MZ_GC_REG();
2808
2809 array = ALLOC_CLEAR_MULT(char *, size + 1);
2810
2811 rest = list;
2812 for (i = 0; i < size; ++i)
2813 {
2814 line = SCHEME_CAR(rest);
2815 rest = SCHEME_CDR(rest);
2816 array[i] = string_to_line(line);
2817 }
2818
2819 curbuf = buf->buf;
2820
2821 if (u_save((linenr_T)n, (linenr_T)(n + 1)) == FAIL)
2822 {
2823 curbuf = savebuf;
2824 free_array(array);
2825 raise_vim_exn(_("cannot save undo information"));
2826 }
2827 else
2828 {
2829 for (i = 0; i < size; ++i)
2830 if (ml_append((linenr_T)(n + i), (char_u *)array[i],
2831 0, FALSE) == FAIL)
2832 {
2833 curbuf = savebuf;
2834 free_array(array);
2835 raise_vim_exn(_("cannot insert line"));
2836 }
2837
2838 if (i > 0)
2839 appended_lines_mark((linenr_T)n, (long)i);
2840 }
2841 free_array(array);
2842 MZ_GC_UNREG();
2843 curbuf = savebuf;
2844 update_screen(VALID);
2845 }
2846
2847 MZ_GC_UNREG();
2848 raise_if_error();
2849 return scheme_void;
2850 }
2851
2852 /*
2853 * Predicates
2854 */
2855 /*
2856 * (buff? obj)
2857 */
2858 static Scheme_Object *
vim_bufferp(void * data UNUSED,int argc UNUSED,Scheme_Object ** argv)2859 vim_bufferp(void *data UNUSED, int argc UNUSED, Scheme_Object **argv)
2860 {
2861 if (SCHEME_VIMBUFFERP(argv[0]))
2862 return scheme_true;
2863 else
2864 return scheme_false;
2865 }
2866
2867 /*
2868 * (win? obj)
2869 */
2870 static Scheme_Object *
vim_windowp(void * data UNUSED,int argc UNUSED,Scheme_Object ** argv)2871 vim_windowp(void *data UNUSED, int argc UNUSED, Scheme_Object **argv)
2872 {
2873 if (SCHEME_VIMWINDOWP(argv[0]))
2874 return scheme_true;
2875 else
2876 return scheme_false;
2877 }
2878
2879 /*
2880 * (buff-valid? obj)
2881 */
2882 static Scheme_Object *
vim_buffer_validp(void * data UNUSED,int argc UNUSED,Scheme_Object ** argv)2883 vim_buffer_validp(void *data UNUSED, int argc UNUSED, Scheme_Object **argv)
2884 {
2885 if (SCHEME_VIMBUFFERP(argv[0])
2886 && ((vim_mz_buffer *)argv[0])->buf != INVALID_BUFFER_VALUE)
2887 return scheme_true;
2888 else
2889 return scheme_false;
2890 }
2891
2892 /*
2893 * (win-valid? obj)
2894 */
2895 static Scheme_Object *
vim_window_validp(void * data UNUSED,int argc UNUSED,Scheme_Object ** argv)2896 vim_window_validp(void *data UNUSED, int argc UNUSED, Scheme_Object **argv)
2897 {
2898 if (SCHEME_VIMWINDOWP(argv[0])
2899 && ((vim_mz_window *)argv[0])->win != INVALID_WINDOW_VALUE)
2900 return scheme_true;
2901 else
2902 return scheme_false;
2903 }
2904
2905 /*
2906 *===========================================================================
2907 * Utilities
2908 *===========================================================================
2909 */
2910
2911 /*
2912 * Convert an MzScheme string into a Vim line.
2913 *
2914 * All internal nulls are replaced by newline characters.
2915 * It is an error for the string to contain newline characters.
2916 *
2917 * Returns pointer to Vim allocated memory
2918 */
2919 static char *
string_to_line(Scheme_Object * obj)2920 string_to_line(Scheme_Object *obj)
2921 {
2922 char *scheme_str = NULL;
2923 char *vim_str = NULL;
2924 OUTPUT_LEN_TYPE len;
2925 int i;
2926
2927 scheme_str = scheme_display_to_string(obj, &len);
2928
2929 // Error checking: String must not contain newlines, as we
2930 // are replacing a single line, and we must replace it with
2931 // a single line.
2932 if (memchr(scheme_str, '\n', len))
2933 scheme_signal_error(_("string cannot contain newlines"));
2934
2935 vim_str = alloc(len + 1);
2936
2937 // Create a copy of the string, with internal nulls replaced by
2938 // newline characters, as is the vim convention.
2939 for (i = 0; i < len; ++i)
2940 {
2941 if (scheme_str[i] == '\0')
2942 vim_str[i] = '\n';
2943 else
2944 vim_str[i] = scheme_str[i];
2945 }
2946
2947 vim_str[i] = '\0';
2948
2949 MZ_GC_CHECK();
2950 return vim_str;
2951 }
2952
2953 #ifdef FEAT_EVAL
2954 /*
2955 * Convert Vim value into MzScheme, adopted from if_python.c
2956 */
2957 static Scheme_Object *
vim_to_mzscheme(typval_T * vim_value)2958 vim_to_mzscheme(typval_T *vim_value)
2959 {
2960 Scheme_Object *result = NULL;
2961 // hash table to store visited values to avoid infinite loops
2962 Scheme_Hash_Table *visited = NULL;
2963
2964 MZ_GC_DECL_REG(2);
2965 MZ_GC_VAR_IN_REG(0, result);
2966 MZ_GC_VAR_IN_REG(1, visited);
2967 MZ_GC_REG();
2968
2969 visited = scheme_make_hash_table(SCHEME_hash_ptr);
2970 MZ_GC_CHECK();
2971
2972 result = vim_to_mzscheme_impl(vim_value, 1, visited);
2973
2974 MZ_GC_UNREG();
2975 return result;
2976 }
2977
2978 static Scheme_Object *
vim_to_mzscheme_impl(typval_T * vim_value,int depth,Scheme_Hash_Table * visited)2979 vim_to_mzscheme_impl(typval_T *vim_value, int depth, Scheme_Hash_Table *visited)
2980 {
2981 Scheme_Object *result = NULL;
2982 int new_value = TRUE;
2983
2984 MZ_GC_DECL_REG(2);
2985 MZ_GC_VAR_IN_REG(0, result);
2986 MZ_GC_VAR_IN_REG(1, visited);
2987 MZ_GC_REG();
2988
2989 // Avoid infinite recursion
2990 if (depth > 100)
2991 {
2992 MZ_GC_UNREG();
2993 return scheme_void;
2994 }
2995
2996 // Check if we run into a recursive loop. The item must be in visited
2997 // then and we can use it again.
2998 result = scheme_hash_get(visited, (Scheme_Object *)vim_value);
2999 MZ_GC_CHECK();
3000 if (result != NULL) // found, do nothing
3001 new_value = FALSE;
3002 else if (vim_value->v_type == VAR_STRING)
3003 {
3004 result = scheme_make_byte_string((char *)vim_value->vval.v_string);
3005 MZ_GC_CHECK();
3006 }
3007 else if (vim_value->v_type == VAR_NUMBER)
3008 {
3009 result = scheme_make_integer((long)vim_value->vval.v_number);
3010 MZ_GC_CHECK();
3011 }
3012 # ifdef FEAT_FLOAT
3013 else if (vim_value->v_type == VAR_FLOAT)
3014 {
3015 result = scheme_make_double((double)vim_value->vval.v_float);
3016 MZ_GC_CHECK();
3017 }
3018 # endif
3019 else if (vim_value->v_type == VAR_LIST)
3020 {
3021 list_T *list = vim_value->vval.v_list;
3022 listitem_T *curr;
3023
3024 if (list == NULL || list->lv_first == NULL)
3025 result = scheme_null;
3026 else
3027 {
3028 Scheme_Object *obj = NULL;
3029
3030 MZ_GC_DECL_REG(1);
3031 MZ_GC_VAR_IN_REG(0, obj);
3032 MZ_GC_REG();
3033
3034 curr = list->lv_u.mat.lv_last;
3035 obj = vim_to_mzscheme_impl(&curr->li_tv, depth + 1, visited);
3036 result = scheme_make_pair(obj, scheme_null);
3037 MZ_GC_CHECK();
3038
3039 while (curr != list->lv_first)
3040 {
3041 curr = curr->li_prev;
3042 obj = vim_to_mzscheme_impl(&curr->li_tv, depth + 1, visited);
3043 result = scheme_make_pair(obj, result);
3044 MZ_GC_CHECK();
3045 }
3046 }
3047 MZ_GC_UNREG();
3048 }
3049 else if (vim_value->v_type == VAR_DICT)
3050 {
3051 Scheme_Object *key = NULL;
3052 Scheme_Object *obj = NULL;
3053
3054 MZ_GC_DECL_REG(2);
3055 MZ_GC_VAR_IN_REG(0, key);
3056 MZ_GC_VAR_IN_REG(1, obj);
3057 MZ_GC_REG();
3058
3059 result = (Scheme_Object *)scheme_make_hash_table(SCHEME_hash_ptr);
3060 MZ_GC_CHECK();
3061 if (vim_value->vval.v_dict != NULL)
3062 {
3063 hashtab_T *ht = &vim_value->vval.v_dict->dv_hashtab;
3064 long_u todo = ht->ht_used;
3065 hashitem_T *hi;
3066 dictitem_T *di;
3067
3068 for (hi = ht->ht_array; todo > 0; ++hi)
3069 {
3070 if (!HASHITEM_EMPTY(hi))
3071 {
3072 --todo;
3073
3074 di = dict_lookup(hi);
3075 obj = vim_to_mzscheme_impl(&di->di_tv, depth + 1, visited);
3076 key = scheme_make_byte_string((char *)hi->hi_key);
3077 MZ_GC_CHECK();
3078 scheme_hash_set((Scheme_Hash_Table *)result, key, obj);
3079 MZ_GC_CHECK();
3080 }
3081 }
3082 }
3083 MZ_GC_UNREG();
3084 }
3085 else if (vim_value->v_type == VAR_FUNC)
3086 {
3087 Scheme_Object *funcname = NULL;
3088
3089 MZ_GC_DECL_REG(1);
3090 MZ_GC_VAR_IN_REG(0, funcname);
3091 MZ_GC_REG();
3092
3093 // FIXME: func_ref() and func_unref() are needed.
3094 funcname = scheme_make_byte_string((char *)vim_value->vval.v_string);
3095 MZ_GC_CHECK();
3096 result = scheme_make_closed_prim_w_arity(vim_funcref, funcname,
3097 (const char *)BYTE_STRING_VALUE(funcname), 0, -1);
3098 MZ_GC_CHECK();
3099
3100 MZ_GC_UNREG();
3101 }
3102 else if (vim_value->v_type == VAR_PARTIAL)
3103 {
3104 if (vim_value->vval.v_partial == NULL)
3105 result = scheme_null;
3106 else
3107 {
3108 Scheme_Object *funcname = NULL;
3109
3110 MZ_GC_DECL_REG(1);
3111 MZ_GC_VAR_IN_REG(0, funcname);
3112 MZ_GC_REG();
3113
3114 // FIXME: func_ref() and func_unref() are needed.
3115 // TODO: Support pt_dict and pt_argv.
3116 funcname = scheme_make_byte_string(
3117 (char *)partial_name(vim_value->vval.v_partial));
3118 MZ_GC_CHECK();
3119 result = scheme_make_closed_prim_w_arity(vim_funcref, funcname,
3120 (const char *)BYTE_STRING_VALUE(funcname), 0, -1);
3121 MZ_GC_CHECK();
3122
3123 MZ_GC_UNREG();
3124 }
3125 }
3126 else if (vim_value->v_type == VAR_BOOL || vim_value->v_type == VAR_SPECIAL)
3127 {
3128 if (vim_value->vval.v_number <= VVAL_TRUE)
3129 result = scheme_make_integer((long)vim_value->vval.v_number);
3130 else
3131 result = scheme_null;
3132 MZ_GC_CHECK();
3133 }
3134 else
3135 {
3136 result = scheme_void;
3137 new_value = FALSE;
3138 }
3139 if (new_value)
3140 {
3141 scheme_hash_set(visited, (Scheme_Object *)vim_value, result);
3142 MZ_GC_CHECK();
3143 }
3144 MZ_GC_UNREG();
3145 return result;
3146 }
3147
3148 static int
mzscheme_to_vim(Scheme_Object * obj,typval_T * tv)3149 mzscheme_to_vim(Scheme_Object *obj, typval_T *tv)
3150 {
3151 int i, status;
3152 Scheme_Hash_Table *visited = NULL;
3153
3154 MZ_GC_DECL_REG(2);
3155 MZ_GC_VAR_IN_REG(0, obj);
3156 MZ_GC_VAR_IN_REG(1, visited);
3157 MZ_GC_REG();
3158
3159 visited = scheme_make_hash_table(SCHEME_hash_ptr);
3160 MZ_GC_CHECK();
3161
3162 status = mzscheme_to_vim_impl(obj, tv, 1, visited);
3163 for (i = 0; i < visited->size; ++i)
3164 {
3165 // free up remembered objects
3166 if (visited->vals[i] != NULL)
3167 free_tv((typval_T *)visited->vals[i]);
3168 }
3169
3170 MZ_GC_UNREG();
3171 return status;
3172 }
3173 static int
mzscheme_to_vim_impl(Scheme_Object * obj,typval_T * tv,int depth,Scheme_Hash_Table * visited)3174 mzscheme_to_vim_impl(Scheme_Object *obj, typval_T *tv, int depth,
3175 Scheme_Hash_Table *visited)
3176 {
3177 int status = OK;
3178 typval_T *found;
3179
3180 MZ_GC_DECL_REG(2);
3181 MZ_GC_VAR_IN_REG(0, obj);
3182 MZ_GC_VAR_IN_REG(1, visited);
3183 MZ_GC_REG();
3184
3185 MZ_GC_CHECK();
3186 if (depth > 100) // limit the deepest recursion level
3187 {
3188 tv->v_type = VAR_NUMBER;
3189 tv->vval.v_number = 0;
3190 return FAIL;
3191 }
3192
3193 found = (typval_T *)scheme_hash_get(visited, obj);
3194 if (found != NULL)
3195 copy_tv(found, tv);
3196 else if (SCHEME_VOIDP(obj))
3197 {
3198 tv->v_type = VAR_SPECIAL;
3199 tv->vval.v_number = VVAL_NULL;
3200 }
3201 else if (SCHEME_INTP(obj))
3202 {
3203 tv->v_type = VAR_NUMBER;
3204 tv->vval.v_number = SCHEME_INT_VAL(obj);
3205 }
3206 else if (SCHEME_BOOLP(obj))
3207 {
3208 tv->v_type = VAR_BOOL;
3209 tv->vval.v_number = SCHEME_TRUEP(obj);
3210 }
3211 # ifdef FEAT_FLOAT
3212 else if (SCHEME_DBLP(obj))
3213 {
3214 tv->v_type = VAR_FLOAT;
3215 tv->vval.v_float = SCHEME_DBL_VAL(obj);
3216 }
3217 # endif
3218 else if (SCHEME_BYTE_STRINGP(obj))
3219 {
3220 tv->v_type = VAR_STRING;
3221 tv->vval.v_string = vim_strsave(BYTE_STRING_VALUE(obj));
3222 }
3223 # if MZSCHEME_VERSION_MAJOR >= 299
3224 else if (SCHEME_CHAR_STRINGP(obj))
3225 {
3226 Scheme_Object *tmp = NULL;
3227 MZ_GC_DECL_REG(1);
3228 MZ_GC_VAR_IN_REG(0, tmp);
3229 MZ_GC_REG();
3230
3231 tmp = scheme_char_string_to_byte_string(obj);
3232 tv->v_type = VAR_STRING;
3233 tv->vval.v_string = vim_strsave(BYTE_STRING_VALUE(tmp));
3234 MZ_GC_UNREG();
3235 }
3236 #endif
3237 else if (SCHEME_VECTORP(obj) || SCHEME_NULLP(obj)
3238 || SCHEME_PAIRP(obj) || SCHEME_MUTABLE_PAIRP(obj))
3239 {
3240 list_T *list = list_alloc();
3241 if (list == NULL)
3242 status = FAIL;
3243 else
3244 {
3245 int i;
3246 Scheme_Object *curr = NULL;
3247 Scheme_Object *cval = NULL;
3248 // temporary var to hold current element of vectors and pairs
3249 typval_T *v;
3250
3251 MZ_GC_DECL_REG(2);
3252 MZ_GC_VAR_IN_REG(0, curr);
3253 MZ_GC_VAR_IN_REG(1, cval);
3254 MZ_GC_REG();
3255
3256 tv->v_type = VAR_LIST;
3257 tv->vval.v_list = list;
3258 ++list->lv_refcount;
3259
3260 v = ALLOC_ONE(typval_T);
3261 if (v == NULL)
3262 status = FAIL;
3263 else
3264 {
3265 // add the value in advance to allow handling of self-referential
3266 // data structures
3267 typval_T *visited_tv = ALLOC_ONE(typval_T);
3268 copy_tv(tv, visited_tv);
3269 scheme_hash_set(visited, obj, (Scheme_Object *)visited_tv);
3270
3271 if (SCHEME_VECTORP(obj))
3272 {
3273 for (i = 0; i < SCHEME_VEC_SIZE(obj); ++i)
3274 {
3275 cval = SCHEME_VEC_ELS(obj)[i];
3276 status = mzscheme_to_vim_impl(cval, v, depth + 1, visited);
3277 if (status == FAIL)
3278 break;
3279 status = list_append_tv(list, v);
3280 clear_tv(v);
3281 if (status == FAIL)
3282 break;
3283 }
3284 }
3285 else if (SCHEME_PAIRP(obj) || SCHEME_MUTABLE_PAIRP(obj))
3286 {
3287 for (curr = obj;
3288 SCHEME_PAIRP(curr) || SCHEME_MUTABLE_PAIRP(curr);
3289 curr = SCHEME_CDR(curr))
3290 {
3291 cval = SCHEME_CAR(curr);
3292 status = mzscheme_to_vim_impl(cval, v, depth + 1, visited);
3293 if (status == FAIL)
3294 break;
3295 status = list_append_tv(list, v);
3296 clear_tv(v);
3297 if (status == FAIL)
3298 break;
3299 }
3300 // improper list not terminated with null
3301 // need to handle the last element
3302 if (status == OK && !SCHEME_NULLP(curr))
3303 {
3304 status = mzscheme_to_vim_impl(cval, v, depth + 1, visited);
3305 if (status == OK)
3306 {
3307 status = list_append_tv(list, v);
3308 clear_tv(v);
3309 }
3310 }
3311 }
3312 // nothing to do for scheme_null
3313 vim_free(v);
3314 }
3315 MZ_GC_UNREG();
3316 }
3317 }
3318 else if (SCHEME_HASHTP(obj))
3319 {
3320 int i;
3321 dict_T *dict;
3322 Scheme_Object *key = NULL;
3323 Scheme_Object *val = NULL;
3324
3325 MZ_GC_DECL_REG(2);
3326 MZ_GC_VAR_IN_REG(0, key);
3327 MZ_GC_VAR_IN_REG(1, val);
3328 MZ_GC_REG();
3329
3330 dict = dict_alloc();
3331 if (dict == NULL)
3332 status = FAIL;
3333 else
3334 {
3335 typval_T *visited_tv = ALLOC_ONE(typval_T);
3336
3337 tv->v_type = VAR_DICT;
3338 tv->vval.v_dict = dict;
3339 ++dict->dv_refcount;
3340
3341 copy_tv(tv, visited_tv);
3342 scheme_hash_set(visited, obj, (Scheme_Object *)visited_tv);
3343
3344 for (i = 0; i < ((Scheme_Hash_Table *)obj)->size; ++i)
3345 {
3346 if (((Scheme_Hash_Table *) obj)->vals[i] != NULL)
3347 {
3348 // generate item for `display'ed Scheme key
3349 dictitem_T *item = dictitem_alloc((char_u *)string_to_line(
3350 ((Scheme_Hash_Table *) obj)->keys[i]));
3351 // convert Scheme val to Vim and add it to the dict
3352 if (mzscheme_to_vim_impl(((Scheme_Hash_Table *) obj)->vals[i],
3353 &item->di_tv, depth + 1, visited) == FAIL
3354 || dict_add(dict, item) == FAIL)
3355 {
3356 dictitem_free(item);
3357 status = FAIL;
3358 break;
3359 }
3360 }
3361
3362 }
3363 }
3364 MZ_GC_UNREG();
3365 }
3366 else
3367 {
3368 // `display' any other value to string
3369 tv->v_type = VAR_STRING;
3370 tv->vval.v_string = (char_u *)string_to_line(obj);
3371 }
3372 MZ_GC_UNREG();
3373 return status;
3374 }
3375
3376 /*
3377 * Scheme prim procedure wrapping Vim funcref
3378 */
3379 static Scheme_Object *
vim_funcref(void * name,int argc,Scheme_Object ** argv)3380 vim_funcref(void *name, int argc, Scheme_Object **argv)
3381 {
3382 int i;
3383 typval_T args;
3384 int status = OK;
3385 Scheme_Object *result = NULL;
3386 list_T *list = list_alloc();
3387
3388 MZ_GC_DECL_REG(1);
3389 MZ_GC_VAR_IN_REG(0, result);
3390 MZ_GC_REG();
3391
3392 result = scheme_void;
3393 if (list == NULL)
3394 status = FAIL;
3395 else
3396 {
3397 args.v_type = VAR_LIST;
3398 args.vval.v_list = list;
3399 ++list->lv_refcount;
3400 for (i = 0; status == OK && i < argc; ++i)
3401 {
3402 typval_T *v = ALLOC_ONE(typval_T);
3403 if (v == NULL)
3404 status = FAIL;
3405 else
3406 {
3407 status = mzscheme_to_vim(argv[i], v);
3408 if (status == OK)
3409 {
3410 status = list_append_tv(list, v);
3411 clear_tv(v);
3412 }
3413 vim_free(v);
3414 }
3415 }
3416 if (status == OK)
3417 {
3418 typval_T ret;
3419 ret.v_type = VAR_UNKNOWN;
3420
3421 mzscheme_call_vim(BYTE_STRING_VALUE((Scheme_Object *)name), &args, &ret);
3422 MZ_GC_CHECK();
3423 result = vim_to_mzscheme(&ret);
3424 clear_tv(&ret);
3425 MZ_GC_CHECK();
3426 }
3427 }
3428 clear_tv(&args);
3429 MZ_GC_UNREG();
3430 if (status != OK)
3431 raise_vim_exn(_("error converting Scheme values to Vim"));
3432 else
3433 raise_if_error();
3434 return result;
3435 }
3436
3437 void
do_mzeval(char_u * str,typval_T * rettv)3438 do_mzeval(char_u *str, typval_T *rettv)
3439 {
3440 Scheme_Object *ret = NULL;
3441
3442 MZ_GC_DECL_REG(1);
3443 MZ_GC_VAR_IN_REG(0, ret);
3444 MZ_GC_REG();
3445
3446 if (mzscheme_init())
3447 {
3448 MZ_GC_UNREG();
3449 return;
3450 }
3451
3452 MZ_GC_CHECK();
3453 if (eval_with_exn_handling(str, do_eval, &ret) == OK)
3454 mzscheme_to_vim(ret, rettv);
3455
3456 MZ_GC_UNREG();
3457 }
3458 #endif
3459
3460 /*
3461 * Check to see whether a Vim error has been reported, or a keyboard
3462 * interrupt (from vim --> got_int) has been detected.
3463 */
3464 static int
vim_error_check(void)3465 vim_error_check(void)
3466 {
3467 return (got_int || did_emsg);
3468 }
3469
3470 /*
3471 * register Scheme exn:vim
3472 */
3473 static void
register_vim_exn(void)3474 register_vim_exn(void)
3475 {
3476 int nc = 0;
3477 int i;
3478 Scheme_Object *struct_exn = NULL;
3479 Scheme_Object *exn_name = NULL;
3480
3481 MZ_GC_DECL_REG(2);
3482 MZ_GC_VAR_IN_REG(0, struct_exn);
3483 MZ_GC_VAR_IN_REG(1, exn_name);
3484 MZ_GC_REG();
3485
3486 exn_name = scheme_intern_symbol("exn:vim");
3487 MZ_GC_CHECK();
3488 struct_exn = scheme_builtin_value("struct:exn");
3489 MZ_GC_CHECK();
3490
3491 if (vim_exn == NULL)
3492 vim_exn = scheme_make_struct_type(exn_name,
3493 struct_exn, NULL, 0, 0, NULL, NULL
3494 #if MZSCHEME_VERSION_MAJOR >= 299
3495 , NULL
3496 #endif
3497 );
3498
3499
3500 {
3501 Scheme_Object **tmp = NULL;
3502 Scheme_Object *exn_names[5] = {NULL, NULL, NULL, NULL, NULL};
3503 Scheme_Object *exn_values[5] = {NULL, NULL, NULL, NULL, NULL};
3504 MZ_GC_DECL_REG(6);
3505 MZ_GC_ARRAY_VAR_IN_REG(0, exn_names, 5);
3506 MZ_GC_ARRAY_VAR_IN_REG(3, exn_values, 5);
3507 MZ_GC_REG();
3508
3509 tmp = scheme_make_struct_names(exn_name, scheme_null, 0, &nc);
3510 mch_memmove(exn_names, tmp, nc * sizeof(Scheme_Object *));
3511 MZ_GC_CHECK();
3512
3513 tmp = scheme_make_struct_values(vim_exn, exn_names, nc, 0);
3514 mch_memmove(exn_values, tmp, nc * sizeof(Scheme_Object *));
3515 MZ_GC_CHECK();
3516
3517 for (i = 0; i < nc; i++)
3518 {
3519 scheme_add_global_symbol(exn_names[i],
3520 exn_values[i], environment);
3521 MZ_GC_CHECK();
3522 }
3523 MZ_GC_UNREG();
3524 }
3525 MZ_GC_UNREG();
3526 }
3527
3528 /*
3529 * raise exn:vim, may be with additional info string
3530 */
3531 void
raise_vim_exn(const char * add_info)3532 raise_vim_exn(const char *add_info)
3533 {
3534 char *fmt = _("Vim error: ~a");
3535 Scheme_Object *argv[2] = {NULL, NULL};
3536 Scheme_Object *exn = NULL;
3537 Scheme_Object *byte_string = NULL;
3538
3539 MZ_GC_DECL_REG(5);
3540 MZ_GC_ARRAY_VAR_IN_REG(0, argv, 2);
3541 MZ_GC_VAR_IN_REG(3, exn);
3542 MZ_GC_VAR_IN_REG(4, byte_string);
3543 MZ_GC_REG();
3544
3545 if (add_info != NULL)
3546 {
3547 char *c_string = NULL;
3548 Scheme_Object *info = NULL;
3549
3550 MZ_GC_DECL_REG(3);
3551 MZ_GC_VAR_IN_REG(0, c_string);
3552 MZ_GC_VAR_IN_REG(2, info);
3553 MZ_GC_REG();
3554
3555 info = scheme_make_byte_string(add_info);
3556 MZ_GC_CHECK();
3557 c_string = scheme_format_utf8(fmt, (int)STRLEN(fmt), 1, &info, NULL);
3558 MZ_GC_CHECK();
3559 byte_string = scheme_make_byte_string(c_string);
3560 MZ_GC_CHECK();
3561 argv[0] = scheme_byte_string_to_char_string(byte_string);
3562 SCHEME_SET_IMMUTABLE(argv[0]);
3563 MZ_GC_UNREG();
3564 }
3565 else
3566 {
3567 byte_string = scheme_make_byte_string(_("Vim error"));
3568 MZ_GC_CHECK();
3569 argv[0] = scheme_byte_string_to_char_string(byte_string);
3570 MZ_GC_CHECK();
3571 }
3572 MZ_GC_CHECK();
3573
3574 #if MZSCHEME_VERSION_MAJOR < 360
3575 argv[1] = scheme_current_continuation_marks();
3576 MZ_GC_CHECK();
3577 #else
3578 argv[1] = scheme_current_continuation_marks(NULL);
3579 MZ_GC_CHECK();
3580 #endif
3581
3582 exn = scheme_make_struct_instance(vim_exn, 2, argv);
3583 MZ_GC_CHECK();
3584 scheme_raise(exn);
3585 MZ_GC_UNREG();
3586 }
3587
3588 void
raise_if_error(void)3589 raise_if_error(void)
3590 {
3591 if (vim_error_check())
3592 raise_vim_exn(NULL);
3593 }
3594
3595 /*
3596 * get buffer:
3597 * either current
3598 * or passed as argv[argnum] with checks
3599 */
3600 static vim_mz_buffer *
get_buffer_arg(const char * fname,int argnum,int argc,Scheme_Object ** argv)3601 get_buffer_arg(const char *fname, int argnum, int argc, Scheme_Object **argv)
3602 {
3603 vim_mz_buffer *b;
3604
3605 if (argc < argnum + 1)
3606 return get_vim_curr_buffer();
3607 if (!SCHEME_VIMBUFFERP(argv[argnum]))
3608 scheme_wrong_type(fname, "vim-buffer", argnum, argc, argv);
3609 b = (vim_mz_buffer *)argv[argnum];
3610 (void)get_valid_buffer(argv[argnum]);
3611 return b;
3612 }
3613
3614 /*
3615 * get window:
3616 * either current
3617 * or passed as argv[argnum] with checks
3618 */
3619 static vim_mz_window *
get_window_arg(const char * fname,int argnum,int argc,Scheme_Object ** argv)3620 get_window_arg(const char *fname, int argnum, int argc, Scheme_Object **argv)
3621 {
3622 vim_mz_window *w;
3623
3624 if (argc < argnum + 1)
3625 return get_vim_curr_window();
3626 w = (vim_mz_window *)argv[argnum];
3627 if (!SCHEME_VIMWINDOWP(argv[argnum]))
3628 scheme_wrong_type(fname, "vim-window", argnum, argc, argv);
3629 (void)get_valid_window(argv[argnum]);
3630 return w;
3631 }
3632
3633 /*
3634 * get valid Vim buffer from Scheme_Object*
3635 */
get_valid_buffer(void * obj)3636 buf_T *get_valid_buffer(void *obj)
3637 {
3638 buf_T *buf = ((vim_mz_buffer *)obj)->buf;
3639
3640 if (buf == INVALID_BUFFER_VALUE)
3641 scheme_signal_error(_("buffer is invalid"));
3642 return buf;
3643 }
3644
3645 /*
3646 * get valid Vim window from Scheme_Object*
3647 */
get_valid_window(void * obj)3648 win_T *get_valid_window(void *obj)
3649 {
3650 win_T *win = ((vim_mz_window *)obj)->win;
3651 if (win == INVALID_WINDOW_VALUE)
3652 scheme_signal_error(_("window is invalid"));
3653 return win;
3654 }
3655
3656 int
mzthreads_allowed(void)3657 mzthreads_allowed(void)
3658 {
3659 return mz_threads_allow;
3660 }
3661
3662 static int
line_in_range(linenr_T lnum,buf_T * buf)3663 line_in_range(linenr_T lnum, buf_T *buf)
3664 {
3665 return (lnum > 0 && lnum <= buf->b_ml.ml_line_count);
3666 }
3667
3668 static void
check_line_range(linenr_T lnum,buf_T * buf)3669 check_line_range(linenr_T lnum, buf_T *buf)
3670 {
3671 if (!line_in_range(lnum, buf))
3672 scheme_signal_error(_("linenr out of range"));
3673 }
3674
3675 /*
3676 * Check if deleting lines made the cursor position invalid
3677 * (or you'll get msg from Vim about invalid linenr).
3678 * Changed the lines from "lo" to "hi" and added "extra" lines (negative if
3679 * deleted). Got from if_python.c
3680 */
3681 static void
mz_fix_cursor(int lo,int hi,int extra)3682 mz_fix_cursor(int lo, int hi, int extra)
3683 {
3684 if (curwin->w_cursor.lnum >= lo)
3685 {
3686 // Adjust the cursor position if it's in/after the changed
3687 // lines.
3688 if (curwin->w_cursor.lnum >= hi)
3689 {
3690 curwin->w_cursor.lnum += extra;
3691 check_cursor_col();
3692 }
3693 else if (extra < 0)
3694 {
3695 curwin->w_cursor.lnum = lo;
3696 check_cursor();
3697 }
3698 else
3699 check_cursor_col();
3700 changed_cline_bef_curs();
3701 }
3702 invalidate_botline();
3703 }
3704
3705 static Vim_Prim prims[]=
3706 {
3707 /*
3708 * Buffer-related commands
3709 */
3710 {get_buffer_line, "get-buff-line", 1, 2},
3711 {set_buffer_line, "set-buff-line", 2, 3},
3712 {get_buffer_line_list, "get-buff-line-list", 2, 3},
3713 {get_buffer_name, "get-buff-name", 0, 1},
3714 {get_buffer_num, "get-buff-num", 0, 1},
3715 {get_buffer_size, "get-buff-size", 0, 1},
3716 {set_buffer_line_list, "set-buff-line-list", 3, 4},
3717 {insert_buffer_line_list, "insert-buff-line-list", 2, 3},
3718 {get_curr_buffer, "curr-buff", 0, 0},
3719 {get_buffer_count, "buff-count", 0, 0},
3720 {get_next_buffer, "get-next-buff", 0, 1},
3721 {get_prev_buffer, "get-prev-buff", 0, 1},
3722 {mzscheme_open_buffer, "open-buff", 1, 1},
3723 {get_buffer_by_name, "get-buff-by-name", 1, 1},
3724 {get_buffer_by_num, "get-buff-by-num", 1, 1},
3725 /*
3726 * Window-related commands
3727 */
3728 {get_curr_win, "curr-win", 0, 0},
3729 {get_window_count, "win-count", 0, 0},
3730 {get_window_by_num, "get-win-by-num", 1, 1},
3731 {get_window_num, "get-win-num", 0, 1},
3732 {get_window_buffer, "get-win-buffer", 0, 1},
3733 {get_window_height, "get-win-height", 0, 1},
3734 {set_window_height, "set-win-height", 1, 2},
3735 {get_window_width, "get-win-width", 0, 1},
3736 {set_window_width, "set-win-width", 1, 2},
3737 {get_cursor, "get-cursor", 0, 1},
3738 {set_cursor, "set-cursor", 1, 2},
3739 {get_window_list, "get-win-list", 0, 1},
3740 /*
3741 * Vim-related commands
3742 */
3743 {vim_command, "command", 1, 1},
3744 {vim_eval, "eval", 1, 1},
3745 {get_range_start, "range-start", 0, 0},
3746 {get_range_end, "range-end", 0, 0},
3747 {mzscheme_beep, "beep", 0, 0},
3748 {get_option, "get-option", 1, 2},
3749 {set_option, "set-option", 1, 2},
3750 /*
3751 * small utilities
3752 */
3753 {vim_bufferp, "buff?", 1, 1},
3754 {vim_windowp, "win?", 1, 1},
3755 {vim_buffer_validp, "buff-valid?", 1, 1},
3756 {vim_window_validp, "win-valid?", 1, 1}
3757 };
3758
3759 /*
3760 * return MzScheme wrapper for curbuf
3761 */
3762 static vim_mz_buffer *
get_vim_curr_buffer(void)3763 get_vim_curr_buffer(void)
3764 {
3765 if (curbuf->b_mzscheme_ref == NULL)
3766 return (vim_mz_buffer *)buffer_new(curbuf);
3767 else
3768 return BUFFER_REF(curbuf);
3769 }
3770
3771 /*
3772 * return MzScheme wrapper for curwin
3773 */
3774 static vim_mz_window *
get_vim_curr_window(void)3775 get_vim_curr_window(void)
3776 {
3777 if (curwin->w_mzscheme_ref == NULL)
3778 return (vim_mz_window *)window_new(curwin);
3779 else
3780 return WINDOW_REF(curwin);
3781 }
3782
3783 static void
make_modules(void)3784 make_modules(void)
3785 {
3786 int i;
3787 Scheme_Env *mod = NULL;
3788 Scheme_Object *vimext_symbol = NULL;
3789 Scheme_Object *closed_prim = NULL;
3790
3791 MZ_GC_DECL_REG(3);
3792 MZ_GC_VAR_IN_REG(0, mod);
3793 MZ_GC_VAR_IN_REG(1, vimext_symbol);
3794 MZ_GC_VAR_IN_REG(2, closed_prim);
3795 MZ_GC_REG();
3796
3797 vimext_symbol = scheme_intern_symbol("vimext");
3798 MZ_GC_CHECK();
3799 mod = scheme_primitive_module(vimext_symbol, environment);
3800 MZ_GC_CHECK();
3801 // all prims made closed so they can access their own names
3802 for (i = 0; i < (int)ARRAY_LENGTH(prims); i++)
3803 {
3804 Vim_Prim *prim = prims + i;
3805 closed_prim = scheme_make_closed_prim_w_arity(prim->prim, prim, prim->name,
3806 prim->mina, prim->maxa);
3807 scheme_add_global(prim->name, closed_prim, mod);
3808 MZ_GC_CHECK();
3809 }
3810 scheme_finish_primitive_module(mod);
3811 MZ_GC_CHECK();
3812 MZ_GC_UNREG();
3813 }
3814
3815 #ifdef HAVE_SANDBOX
3816 static Scheme_Object *M_write = NULL;
3817 static Scheme_Object *M_read = NULL;
3818 static Scheme_Object *M_execute = NULL;
3819 static Scheme_Object *M_delete = NULL;
3820
3821 static void
sandbox_check(void)3822 sandbox_check(void)
3823 {
3824 if (sandbox)
3825 raise_vim_exn(_("not allowed in the Vim sandbox"));
3826 }
3827
3828 /*
3829 * security guards to force Vim's sandbox restrictions on MzScheme level
3830 */
3831 static Scheme_Object *
sandbox_file_guard(int argc UNUSED,Scheme_Object ** argv)3832 sandbox_file_guard(int argc UNUSED, Scheme_Object **argv)
3833 {
3834 if (sandbox)
3835 {
3836 Scheme_Object *requested_access = argv[2];
3837
3838 if (M_write == NULL)
3839 {
3840 MZ_REGISTER_STATIC(M_write);
3841 M_write = scheme_intern_symbol("write");
3842 MZ_GC_CHECK();
3843 }
3844 if (M_read == NULL)
3845 {
3846 MZ_REGISTER_STATIC(M_read);
3847 M_read = scheme_intern_symbol("read");
3848 MZ_GC_CHECK();
3849 }
3850 if (M_execute == NULL)
3851 {
3852 MZ_REGISTER_STATIC(M_execute);
3853 M_execute = scheme_intern_symbol("execute");
3854 MZ_GC_CHECK();
3855 }
3856 if (M_delete == NULL)
3857 {
3858 MZ_REGISTER_STATIC(M_delete);
3859 M_delete = scheme_intern_symbol("delete");
3860 MZ_GC_CHECK();
3861 }
3862
3863 while (!SCHEME_NULLP(requested_access))
3864 {
3865 Scheme_Object *item = SCHEME_CAR(requested_access);
3866 if (scheme_eq(item, M_write) || scheme_eq(item, M_read)
3867 || scheme_eq(item, M_execute) || scheme_eq(item, M_delete))
3868 raise_vim_exn(_("not allowed in the Vim sandbox"));
3869 requested_access = SCHEME_CDR(requested_access);
3870 }
3871 }
3872 return scheme_void;
3873 }
3874
3875 static Scheme_Object *
sandbox_network_guard(int argc UNUSED,Scheme_Object ** argv UNUSED)3876 sandbox_network_guard(int argc UNUSED, Scheme_Object **argv UNUSED)
3877 {
3878 return scheme_void;
3879 }
3880 #endif
3881
3882 #endif
3883