1 /* $Id: eqn.c,v 1.38 2011/07/25 15:37:00 kristaps Exp $ */
2 /*
3 * Copyright (c) 2011 Kristaps Dzonsons <kristaps@bsd.lv>
4 *
5 * Permission to use, copy, modify, and distribute this software for any
6 * purpose with or without fee is hereby granted, provided that the above
7 * copyright notice and this permission notice appear in all copies.
8 *
9 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
16 */
17 #ifdef HAVE_CONFIG_H
18 #include "config.h"
19 #endif
20
21 #include <assert.h>
22 #include <limits.h>
23 #include <stdio.h>
24 #include <stdlib.h>
25 #include <string.h>
26 #include <time.h>
27
28 #include "mandoc.h"
29 #include "libmandoc.h"
30 #include "libroff.h"
31
32 #define EQN_NEST_MAX 128 /* maximum nesting of defines */
33 #define EQN_MSG(t, x) mandoc_msg((t), (x)->parse, (x)->eqn.ln, (x)->eqn.pos, NULL)
34
35 enum eqn_rest {
36 EQN_DESCOPE,
37 EQN_ERR,
38 EQN_OK,
39 EQN_EOF
40 };
41
42 enum eqn_symt {
43 EQNSYM_alpha,
44 EQNSYM_beta,
45 EQNSYM_chi,
46 EQNSYM_delta,
47 EQNSYM_epsilon,
48 EQNSYM_eta,
49 EQNSYM_gamma,
50 EQNSYM_iota,
51 EQNSYM_kappa,
52 EQNSYM_lambda,
53 EQNSYM_mu,
54 EQNSYM_nu,
55 EQNSYM_omega,
56 EQNSYM_omicron,
57 EQNSYM_phi,
58 EQNSYM_pi,
59 EQNSYM_ps,
60 EQNSYM_rho,
61 EQNSYM_sigma,
62 EQNSYM_tau,
63 EQNSYM_theta,
64 EQNSYM_upsilon,
65 EQNSYM_xi,
66 EQNSYM_zeta,
67 EQNSYM_DELTA,
68 EQNSYM_GAMMA,
69 EQNSYM_LAMBDA,
70 EQNSYM_OMEGA,
71 EQNSYM_PHI,
72 EQNSYM_PI,
73 EQNSYM_PSI,
74 EQNSYM_SIGMA,
75 EQNSYM_THETA,
76 EQNSYM_UPSILON,
77 EQNSYM_XI,
78 EQNSYM_inter,
79 EQNSYM_union,
80 EQNSYM_prod,
81 EQNSYM_int,
82 EQNSYM_sum,
83 EQNSYM_grad,
84 EQNSYM_del,
85 EQNSYM_times,
86 EQNSYM_cdot,
87 EQNSYM_nothing,
88 EQNSYM_approx,
89 EQNSYM_prime,
90 EQNSYM_half,
91 EQNSYM_partial,
92 EQNSYM_inf,
93 EQNSYM_muchgreat,
94 EQNSYM_muchless,
95 EQNSYM_larrow,
96 EQNSYM_rarrow,
97 EQNSYM_pm,
98 EQNSYM_nequal,
99 EQNSYM_equiv,
100 EQNSYM_lessequal,
101 EQNSYM_moreequal,
102 EQNSYM__MAX
103 };
104
105 enum eqnpartt {
106 EQN_DEFINE = 0,
107 EQN_NDEFINE,
108 EQN_TDEFINE,
109 EQN_SET,
110 EQN_UNDEF,
111 EQN_GFONT,
112 EQN_GSIZE,
113 EQN_BACK,
114 EQN_FWD,
115 EQN_UP,
116 EQN_DOWN,
117 EQN__MAX
118 };
119
120 struct eqnstr {
121 const char *name;
122 size_t sz;
123 };
124
125 #define STRNEQ(p1, sz1, p2, sz2) \
126 ((sz1) == (sz2) && 0 == strncmp((p1), (p2), (sz1)))
127 #define EQNSTREQ(x, p, sz) \
128 STRNEQ((x)->name, (x)->sz, (p), (sz))
129
130 struct eqnpart {
131 struct eqnstr str;
132 int (*fp)(struct eqn_node *);
133 };
134
135 struct eqnsym {
136 struct eqnstr str;
137 const char *sym;
138 };
139
140
141 static enum eqn_rest eqn_box(struct eqn_node *, struct eqn_box *);
142 static struct eqn_box *eqn_box_alloc(struct eqn_node *,
143 struct eqn_box *);
144 static void eqn_box_free(struct eqn_box *);
145 static struct eqn_def *eqn_def_find(struct eqn_node *,
146 const char *, size_t);
147 static int eqn_do_gfont(struct eqn_node *);
148 static int eqn_do_gsize(struct eqn_node *);
149 static int eqn_do_define(struct eqn_node *);
150 static int eqn_do_ign1(struct eqn_node *);
151 static int eqn_do_ign2(struct eqn_node *);
152 static int eqn_do_tdefine(struct eqn_node *);
153 static int eqn_do_undef(struct eqn_node *);
154 static enum eqn_rest eqn_eqn(struct eqn_node *, struct eqn_box *);
155 static enum eqn_rest eqn_list(struct eqn_node *, struct eqn_box *);
156 static enum eqn_rest eqn_matrix(struct eqn_node *, struct eqn_box *);
157 static const char *eqn_nexttok(struct eqn_node *, size_t *);
158 static const char *eqn_nextrawtok(struct eqn_node *, size_t *);
159 static const char *eqn_next(struct eqn_node *,
160 char, size_t *, int);
161 static void eqn_rewind(struct eqn_node *);
162
163 static const struct eqnpart eqnparts[EQN__MAX] = {
164 { { "define", 6 }, eqn_do_define }, /* EQN_DEFINE */
165 { { "ndefine", 7 }, eqn_do_define }, /* EQN_NDEFINE */
166 { { "tdefine", 7 }, eqn_do_tdefine }, /* EQN_TDEFINE */
167 { { "set", 3 }, eqn_do_ign2 }, /* EQN_SET */
168 { { "undef", 5 }, eqn_do_undef }, /* EQN_UNDEF */
169 { { "gfont", 5 }, eqn_do_gfont }, /* EQN_GFONT */
170 { { "gsize", 5 }, eqn_do_gsize }, /* EQN_GSIZE */
171 { { "back", 4 }, eqn_do_ign1 }, /* EQN_BACK */
172 { { "fwd", 3 }, eqn_do_ign1 }, /* EQN_FWD */
173 { { "up", 2 }, eqn_do_ign1 }, /* EQN_UP */
174 { { "down", 4 }, eqn_do_ign1 }, /* EQN_DOWN */
175 };
176
177 static const struct eqnstr eqnmarks[EQNMARK__MAX] = {
178 { "", 0 }, /* EQNMARK_NONE */
179 { "dot", 3 }, /* EQNMARK_DOT */
180 { "dotdot", 6 }, /* EQNMARK_DOTDOT */
181 { "hat", 3 }, /* EQNMARK_HAT */
182 { "tilde", 5 }, /* EQNMARK_TILDE */
183 { "vec", 3 }, /* EQNMARK_VEC */
184 { "dyad", 4 }, /* EQNMARK_DYAD */
185 { "bar", 3 }, /* EQNMARK_BAR */
186 { "under", 5 }, /* EQNMARK_UNDER */
187 };
188
189 static const struct eqnstr eqnfonts[EQNFONT__MAX] = {
190 { "", 0 }, /* EQNFONT_NONE */
191 { "roman", 5 }, /* EQNFONT_ROMAN */
192 { "bold", 4 }, /* EQNFONT_BOLD */
193 { "fat", 3 }, /* EQNFONT_FAT */
194 { "italic", 6 }, /* EQNFONT_ITALIC */
195 };
196
197 static const struct eqnstr eqnposs[EQNPOS__MAX] = {
198 { "", 0 }, /* EQNPOS_NONE */
199 { "over", 4 }, /* EQNPOS_OVER */
200 { "sup", 3 }, /* EQNPOS_SUP */
201 { "sub", 3 }, /* EQNPOS_SUB */
202 { "to", 2 }, /* EQNPOS_TO */
203 { "from", 4 }, /* EQNPOS_FROM */
204 };
205
206 static const struct eqnstr eqnpiles[EQNPILE__MAX] = {
207 { "", 0 }, /* EQNPILE_NONE */
208 { "pile", 4 }, /* EQNPILE_PILE */
209 { "cpile", 5 }, /* EQNPILE_CPILE */
210 { "rpile", 5 }, /* EQNPILE_RPILE */
211 { "lpile", 5 }, /* EQNPILE_LPILE */
212 { "col", 3 }, /* EQNPILE_COL */
213 { "ccol", 4 }, /* EQNPILE_CCOL */
214 { "rcol", 4 }, /* EQNPILE_RCOL */
215 { "lcol", 4 }, /* EQNPILE_LCOL */
216 };
217
218 static const struct eqnsym eqnsyms[EQNSYM__MAX] = {
219 { { "alpha", 5 }, "*a" }, /* EQNSYM_alpha */
220 { { "beta", 4 }, "*b" }, /* EQNSYM_beta */
221 { { "chi", 3 }, "*x" }, /* EQNSYM_chi */
222 { { "delta", 5 }, "*d" }, /* EQNSYM_delta */
223 { { "epsilon", 7 }, "*e" }, /* EQNSYM_epsilon */
224 { { "eta", 3 }, "*y" }, /* EQNSYM_eta */
225 { { "gamma", 5 }, "*g" }, /* EQNSYM_gamma */
226 { { "iota", 4 }, "*i" }, /* EQNSYM_iota */
227 { { "kappa", 5 }, "*k" }, /* EQNSYM_kappa */
228 { { "lambda", 6 }, "*l" }, /* EQNSYM_lambda */
229 { { "mu", 2 }, "*m" }, /* EQNSYM_mu */
230 { { "nu", 2 }, "*n" }, /* EQNSYM_nu */
231 { { "omega", 5 }, "*w" }, /* EQNSYM_omega */
232 { { "omicron", 7 }, "*o" }, /* EQNSYM_omicron */
233 { { "phi", 3 }, "*f" }, /* EQNSYM_phi */
234 { { "pi", 2 }, "*p" }, /* EQNSYM_pi */
235 { { "psi", 2 }, "*q" }, /* EQNSYM_psi */
236 { { "rho", 3 }, "*r" }, /* EQNSYM_rho */
237 { { "sigma", 5 }, "*s" }, /* EQNSYM_sigma */
238 { { "tau", 3 }, "*t" }, /* EQNSYM_tau */
239 { { "theta", 5 }, "*h" }, /* EQNSYM_theta */
240 { { "upsilon", 7 }, "*u" }, /* EQNSYM_upsilon */
241 { { "xi", 2 }, "*c" }, /* EQNSYM_xi */
242 { { "zeta", 4 }, "*z" }, /* EQNSYM_zeta */
243 { { "DELTA", 5 }, "*D" }, /* EQNSYM_DELTA */
244 { { "GAMMA", 5 }, "*G" }, /* EQNSYM_GAMMA */
245 { { "LAMBDA", 6 }, "*L" }, /* EQNSYM_LAMBDA */
246 { { "OMEGA", 5 }, "*W" }, /* EQNSYM_OMEGA */
247 { { "PHI", 3 }, "*F" }, /* EQNSYM_PHI */
248 { { "PI", 2 }, "*P" }, /* EQNSYM_PI */
249 { { "PSI", 3 }, "*Q" }, /* EQNSYM_PSI */
250 { { "SIGMA", 5 }, "*S" }, /* EQNSYM_SIGMA */
251 { { "THETA", 5 }, "*H" }, /* EQNSYM_THETA */
252 { { "UPSILON", 7 }, "*U" }, /* EQNSYM_UPSILON */
253 { { "XI", 2 }, "*C" }, /* EQNSYM_XI */
254 { { "inter", 5 }, "ca" }, /* EQNSYM_inter */
255 { { "union", 5 }, "cu" }, /* EQNSYM_union */
256 { { "prod", 4 }, "product" }, /* EQNSYM_prod */
257 { { "int", 3 }, "integral" }, /* EQNSYM_int */
258 { { "sum", 3 }, "sum" }, /* EQNSYM_sum */
259 { { "grad", 4 }, "gr" }, /* EQNSYM_grad */
260 { { "del", 3 }, "gr" }, /* EQNSYM_del */
261 { { "times", 5 }, "mu" }, /* EQNSYM_times */
262 { { "cdot", 4 }, "pc" }, /* EQNSYM_cdot */
263 { { "nothing", 7 }, "&" }, /* EQNSYM_nothing */
264 { { "approx", 6 }, "~~" }, /* EQNSYM_approx */
265 { { "prime", 5 }, "aq" }, /* EQNSYM_prime */
266 { { "half", 4 }, "12" }, /* EQNSYM_half */
267 { { "partial", 7 }, "pd" }, /* EQNSYM_partial */
268 { { "inf", 3 }, "if" }, /* EQNSYM_inf */
269 { { ">>", 2 }, ">>" }, /* EQNSYM_muchgreat */
270 { { "<<", 2 }, "<<" }, /* EQNSYM_muchless */
271 { { "<-", 2 }, "<-" }, /* EQNSYM_larrow */
272 { { "->", 2 }, "->" }, /* EQNSYM_rarrow */
273 { { "+-", 2 }, "+-" }, /* EQNSYM_pm */
274 { { "!=", 2 }, "!=" }, /* EQNSYM_nequal */
275 { { "==", 2 }, "==" }, /* EQNSYM_equiv */
276 { { "<=", 2 }, "<=" }, /* EQNSYM_lessequal */
277 { { ">=", 2 }, ">=" }, /* EQNSYM_moreequal */
278 };
279
280 /* ARGSUSED */
281 enum rofferr
282 eqn_read(struct eqn_node **epp, int ln,
283 const char *p, int pos, int *offs)
284 {
285 size_t sz;
286 struct eqn_node *ep;
287 enum rofferr er;
288
289 ep = *epp;
290
291 /*
292 * If we're the terminating mark, unset our equation status and
293 * validate the full equation.
294 */
295
296 if (0 == strncmp(p, ".EN", 3)) {
297 er = eqn_end(epp);
298 p += 3;
299 while (' ' == *p || '\t' == *p)
300 p++;
301 if ('\0' == *p)
302 return(er);
303 mandoc_msg(MANDOCERR_ARGSLOST, ep->parse, ln, pos, NULL);
304 return(er);
305 }
306
307 /*
308 * Build up the full string, replacing all newlines with regular
309 * whitespace.
310 */
311
312 sz = strlen(p + pos) + 1;
313 ep->data = mandoc_realloc(ep->data, ep->sz + sz + 1);
314
315 /* First invocation: nil terminate the string. */
316
317 if (0 == ep->sz)
318 *ep->data = '\0';
319
320 ep->sz += sz;
321 strlcat(ep->data, p + pos, ep->sz + 1);
322 strlcat(ep->data, " ", ep->sz + 1);
323 return(ROFF_IGN);
324 }
325
326 struct eqn_node *
327 eqn_alloc(const char *name, int pos, int line, struct mparse *parse)
328 {
329 struct eqn_node *p;
330 size_t sz;
331 const char *end;
332
333 p = mandoc_calloc(1, sizeof(struct eqn_node));
334
335 if (name && '\0' != *name) {
336 sz = strlen(name);
337 assert(sz);
338 do {
339 sz--;
340 end = name + (int)sz;
341 } while (' ' == *end || '\t' == *end);
342 p->eqn.name = mandoc_strndup(name, sz + 1);
343 }
344
345 p->parse = parse;
346 p->eqn.ln = line;
347 p->eqn.pos = pos;
348 p->gsize = EQN_DEFSIZE;
349
350 return(p);
351 }
352
353 enum rofferr
354 eqn_end(struct eqn_node **epp)
355 {
356 struct eqn_node *ep;
357 struct eqn_box *root;
358 enum eqn_rest c;
359
360 ep = *epp;
361 *epp = NULL;
362
363 ep->eqn.root = mandoc_calloc(1, sizeof(struct eqn_box));
364
365 root = ep->eqn.root;
366 root->type = EQN_ROOT;
367
368 if (0 == ep->sz)
369 return(ROFF_IGN);
370
371 if (EQN_DESCOPE == (c = eqn_eqn(ep, root))) {
372 EQN_MSG(MANDOCERR_EQNNSCOPE, ep);
373 c = EQN_ERR;
374 }
375
376 return(EQN_EOF == c ? ROFF_EQN : ROFF_IGN);
377 }
378
379 static enum eqn_rest
380 eqn_eqn(struct eqn_node *ep, struct eqn_box *last)
381 {
382 struct eqn_box *bp;
383 enum eqn_rest c;
384
385 bp = eqn_box_alloc(ep, last);
386 bp->type = EQN_SUBEXPR;
387
388 while (EQN_OK == (c = eqn_box(ep, bp)))
389 /* Spin! */ ;
390
391 return(c);
392 }
393
394 static enum eqn_rest
395 eqn_matrix(struct eqn_node *ep, struct eqn_box *last)
396 {
397 struct eqn_box *bp;
398 const char *start;
399 size_t sz;
400 enum eqn_rest c;
401
402 bp = eqn_box_alloc(ep, last);
403 bp->type = EQN_MATRIX;
404
405 if (NULL == (start = eqn_nexttok(ep, &sz))) {
406 EQN_MSG(MANDOCERR_EQNEOF, ep);
407 return(EQN_ERR);
408 }
409 if ( ! STRNEQ(start, sz, "{", 1)) {
410 EQN_MSG(MANDOCERR_EQNSYNT, ep);
411 return(EQN_ERR);
412 }
413
414 while (EQN_OK == (c = eqn_box(ep, bp)))
415 switch (bp->last->pile) {
416 case (EQNPILE_LCOL):
417 /* FALLTHROUGH */
418 case (EQNPILE_CCOL):
419 /* FALLTHROUGH */
420 case (EQNPILE_RCOL):
421 continue;
422 default:
423 EQN_MSG(MANDOCERR_EQNSYNT, ep);
424 return(EQN_ERR);
425 };
426
427 if (EQN_DESCOPE != c) {
428 if (EQN_EOF == c)
429 EQN_MSG(MANDOCERR_EQNEOF, ep);
430 return(EQN_ERR);
431 }
432
433 eqn_rewind(ep);
434 start = eqn_nexttok(ep, &sz);
435 assert(start);
436 if (STRNEQ(start, sz, "}", 1))
437 return(EQN_OK);
438
439 EQN_MSG(MANDOCERR_EQNBADSCOPE, ep);
440 return(EQN_ERR);
441 }
442
443 static enum eqn_rest
444 eqn_list(struct eqn_node *ep, struct eqn_box *last)
445 {
446 struct eqn_box *bp;
447 const char *start;
448 size_t sz;
449 enum eqn_rest c;
450
451 bp = eqn_box_alloc(ep, last);
452 bp->type = EQN_LIST;
453
454 if (NULL == (start = eqn_nexttok(ep, &sz))) {
455 EQN_MSG(MANDOCERR_EQNEOF, ep);
456 return(EQN_ERR);
457 }
458 if ( ! STRNEQ(start, sz, "{", 1)) {
459 EQN_MSG(MANDOCERR_EQNSYNT, ep);
460 return(EQN_ERR);
461 }
462
463 while (EQN_DESCOPE == (c = eqn_eqn(ep, bp))) {
464 eqn_rewind(ep);
465 start = eqn_nexttok(ep, &sz);
466 assert(start);
467 if ( ! STRNEQ(start, sz, "above", 5))
468 break;
469 }
470
471 if (EQN_DESCOPE != c) {
472 if (EQN_ERR != c)
473 EQN_MSG(MANDOCERR_EQNSCOPE, ep);
474 return(EQN_ERR);
475 }
476
477 eqn_rewind(ep);
478 start = eqn_nexttok(ep, &sz);
479 assert(start);
480 if (STRNEQ(start, sz, "}", 1))
481 return(EQN_OK);
482
483 EQN_MSG(MANDOCERR_EQNBADSCOPE, ep);
484 return(EQN_ERR);
485 }
486
487 static enum eqn_rest
488 eqn_box(struct eqn_node *ep, struct eqn_box *last)
489 {
490 size_t sz;
491 const char *start;
492 char *left;
493 char sym[64];
494 enum eqn_rest c;
495 int i, size;
496 struct eqn_box *bp;
497
498 if (NULL == (start = eqn_nexttok(ep, &sz)))
499 return(EQN_EOF);
500
501 if (STRNEQ(start, sz, "}", 1))
502 return(EQN_DESCOPE);
503 else if (STRNEQ(start, sz, "right", 5))
504 return(EQN_DESCOPE);
505 else if (STRNEQ(start, sz, "above", 5))
506 return(EQN_DESCOPE);
507 else if (STRNEQ(start, sz, "mark", 4))
508 return(EQN_OK);
509 else if (STRNEQ(start, sz, "lineup", 6))
510 return(EQN_OK);
511
512 for (i = 0; i < (int)EQN__MAX; i++) {
513 if ( ! EQNSTREQ(&eqnparts[i].str, start, sz))
514 continue;
515 return((*eqnparts[i].fp)(ep) ?
516 EQN_OK : EQN_ERR);
517 }
518
519 if (STRNEQ(start, sz, "{", 1)) {
520 if (EQN_DESCOPE != (c = eqn_eqn(ep, last))) {
521 if (EQN_ERR != c)
522 EQN_MSG(MANDOCERR_EQNSCOPE, ep);
523 return(EQN_ERR);
524 }
525 eqn_rewind(ep);
526 start = eqn_nexttok(ep, &sz);
527 assert(start);
528 if (STRNEQ(start, sz, "}", 1))
529 return(EQN_OK);
530 EQN_MSG(MANDOCERR_EQNBADSCOPE, ep);
531 return(EQN_ERR);
532 }
533
534 for (i = 0; i < (int)EQNPILE__MAX; i++) {
535 if ( ! EQNSTREQ(&eqnpiles[i], start, sz))
536 continue;
537 if (EQN_OK == (c = eqn_list(ep, last)))
538 last->last->pile = (enum eqn_pilet)i;
539 return(c);
540 }
541
542 if (STRNEQ(start, sz, "matrix", 6))
543 return(eqn_matrix(ep, last));
544
545 if (STRNEQ(start, sz, "left", 4)) {
546 if (NULL == (start = eqn_nexttok(ep, &sz))) {
547 EQN_MSG(MANDOCERR_EQNEOF, ep);
548 return(EQN_ERR);
549 }
550 left = mandoc_strndup(start, sz);
551 c = eqn_eqn(ep, last);
552 if (last->last)
553 last->last->left = left;
554 else
555 free(left);
556 if (EQN_DESCOPE != c)
557 return(c);
558 assert(last->last);
559 eqn_rewind(ep);
560 start = eqn_nexttok(ep, &sz);
561 assert(start);
562 if ( ! STRNEQ(start, sz, "right", 5))
563 return(EQN_DESCOPE);
564 if (NULL == (start = eqn_nexttok(ep, &sz))) {
565 EQN_MSG(MANDOCERR_EQNEOF, ep);
566 return(EQN_ERR);
567 }
568 last->last->right = mandoc_strndup(start, sz);
569 return(EQN_OK);
570 }
571
572 for (i = 0; i < (int)EQNPOS__MAX; i++) {
573 if ( ! EQNSTREQ(&eqnposs[i], start, sz))
574 continue;
575 if (NULL == last->last) {
576 EQN_MSG(MANDOCERR_EQNSYNT, ep);
577 return(EQN_ERR);
578 }
579 last->last->pos = (enum eqn_post)i;
580 if (EQN_EOF == (c = eqn_box(ep, last))) {
581 EQN_MSG(MANDOCERR_EQNEOF, ep);
582 return(EQN_ERR);
583 }
584 return(c);
585 }
586
587 for (i = 0; i < (int)EQNMARK__MAX; i++) {
588 if ( ! EQNSTREQ(&eqnmarks[i], start, sz))
589 continue;
590 if (NULL == last->last) {
591 EQN_MSG(MANDOCERR_EQNSYNT, ep);
592 return(EQN_ERR);
593 }
594 last->last->mark = (enum eqn_markt)i;
595 if (EQN_EOF == (c = eqn_box(ep, last))) {
596 EQN_MSG(MANDOCERR_EQNEOF, ep);
597 return(EQN_ERR);
598 }
599 return(c);
600 }
601
602 for (i = 0; i < (int)EQNFONT__MAX; i++) {
603 if ( ! EQNSTREQ(&eqnfonts[i], start, sz))
604 continue;
605 if (EQN_EOF == (c = eqn_box(ep, last))) {
606 EQN_MSG(MANDOCERR_EQNEOF, ep);
607 return(EQN_ERR);
608 } else if (EQN_OK == c)
609 last->last->font = (enum eqn_fontt)i;
610 return(c);
611 }
612
613 if (STRNEQ(start, sz, "size", 4)) {
614 if (NULL == (start = eqn_nexttok(ep, &sz))) {
615 EQN_MSG(MANDOCERR_EQNEOF, ep);
616 return(EQN_ERR);
617 }
618 size = mandoc_strntoi(start, sz, 10);
619 if (EQN_EOF == (c = eqn_box(ep, last))) {
620 EQN_MSG(MANDOCERR_EQNEOF, ep);
621 return(EQN_ERR);
622 } else if (EQN_OK != c)
623 return(c);
624 last->last->size = size;
625 }
626
627 bp = eqn_box_alloc(ep, last);
628 bp->type = EQN_TEXT;
629 for (i = 0; i < (int)EQNSYM__MAX; i++)
630 if (EQNSTREQ(&eqnsyms[i].str, start, sz)) {
631 sym[63] = '\0';
632 snprintf(sym, 62, "\\[%s]", eqnsyms[i].sym);
633 bp->text = mandoc_strdup(sym);
634 return(EQN_OK);
635 }
636
637 bp->text = mandoc_strndup(start, sz);
638 return(EQN_OK);
639 }
640
641 void
642 eqn_free(struct eqn_node *p)
643 {
644 int i;
645
646 eqn_box_free(p->eqn.root);
647
648 for (i = 0; i < (int)p->defsz; i++) {
649 free(p->defs[i].key);
650 free(p->defs[i].val);
651 }
652
653 free(p->eqn.name);
654 free(p->data);
655 free(p->defs);
656 free(p);
657 }
658
659 static struct eqn_box *
660 eqn_box_alloc(struct eqn_node *ep, struct eqn_box *parent)
661 {
662 struct eqn_box *bp;
663
664 bp = mandoc_calloc(1, sizeof(struct eqn_box));
665 bp->parent = parent;
666 bp->size = ep->gsize;
667
668 if (NULL == parent->first)
669 parent->first = bp;
670 else
671 parent->last->next = bp;
672
673 parent->last = bp;
674 return(bp);
675 }
676
677 static void
678 eqn_box_free(struct eqn_box *bp)
679 {
680
681 if (bp->first)
682 eqn_box_free(bp->first);
683 if (bp->next)
684 eqn_box_free(bp->next);
685
686 free(bp->text);
687 free(bp->left);
688 free(bp->right);
689 free(bp);
690 }
691
692 static const char *
693 eqn_nextrawtok(struct eqn_node *ep, size_t *sz)
694 {
695
696 return(eqn_next(ep, '"', sz, 0));
697 }
698
699 static const char *
700 eqn_nexttok(struct eqn_node *ep, size_t *sz)
701 {
702
703 return(eqn_next(ep, '"', sz, 1));
704 }
705
706 static void
707 eqn_rewind(struct eqn_node *ep)
708 {
709
710 ep->cur = ep->rew;
711 }
712
713 static const char *
714 eqn_next(struct eqn_node *ep, char quote, size_t *sz, int repl)
715 {
716 char *start, *next;
717 int q, diff, lim;
718 size_t ssz, dummy;
719 struct eqn_def *def;
720
721 if (NULL == sz)
722 sz = &dummy;
723
724 lim = 0;
725 ep->rew = ep->cur;
726 again:
727 /* Prevent self-definitions. */
728
729 if (lim >= EQN_NEST_MAX) {
730 EQN_MSG(MANDOCERR_ROFFLOOP, ep);
731 return(NULL);
732 }
733
734 ep->cur = ep->rew;
735 start = &ep->data[(int)ep->cur];
736 q = 0;
737
738 if ('\0' == *start)
739 return(NULL);
740
741 if (quote == *start) {
742 ep->cur++;
743 q = 1;
744 }
745
746 start = &ep->data[(int)ep->cur];
747
748 if ( ! q) {
749 if ('{' == *start || '}' == *start)
750 ssz = 1;
751 else
752 ssz = strcspn(start + 1, " ^~\"{}\t") + 1;
753 next = start + (int)ssz;
754 if ('\0' == *next)
755 next = NULL;
756 } else
757 next = strchr(start, quote);
758
759 if (NULL != next) {
760 *sz = (size_t)(next - start);
761 ep->cur += *sz;
762 if (q)
763 ep->cur++;
764 while (' ' == ep->data[(int)ep->cur] ||
765 '\t' == ep->data[(int)ep->cur] ||
766 '^' == ep->data[(int)ep->cur] ||
767 '~' == ep->data[(int)ep->cur])
768 ep->cur++;
769 } else {
770 if (q)
771 EQN_MSG(MANDOCERR_BADQUOTE, ep);
772 next = strchr(start, '\0');
773 *sz = (size_t)(next - start);
774 ep->cur += *sz;
775 }
776
777 /* Quotes aren't expanded for values. */
778
779 if (q || ! repl)
780 return(start);
781
782 if (NULL != (def = eqn_def_find(ep, start, *sz))) {
783 diff = def->valsz - *sz;
784
785 if (def->valsz > *sz) {
786 ep->sz += diff;
787 ep->data = mandoc_realloc(ep->data, ep->sz + 1);
788 ep->data[ep->sz] = '\0';
789 start = &ep->data[(int)ep->rew];
790 }
791
792 diff = def->valsz - *sz;
793 memmove(start + *sz + diff, start + *sz,
794 (strlen(start) - *sz) + 1);
795 memcpy(start, def->val, def->valsz);
796 goto again;
797 }
798
799 return(start);
800 }
801
802 static int
803 eqn_do_ign1(struct eqn_node *ep)
804 {
805
806 if (NULL == eqn_nextrawtok(ep, NULL))
807 EQN_MSG(MANDOCERR_EQNEOF, ep);
808 else
809 return(1);
810
811 return(0);
812 }
813
814 static int
815 eqn_do_ign2(struct eqn_node *ep)
816 {
817
818 if (NULL == eqn_nextrawtok(ep, NULL))
819 EQN_MSG(MANDOCERR_EQNEOF, ep);
820 else if (NULL == eqn_nextrawtok(ep, NULL))
821 EQN_MSG(MANDOCERR_EQNEOF, ep);
822 else
823 return(1);
824
825 return(0);
826 }
827
828 static int
829 eqn_do_tdefine(struct eqn_node *ep)
830 {
831
832 if (NULL == eqn_nextrawtok(ep, NULL))
833 EQN_MSG(MANDOCERR_EQNEOF, ep);
834 else if (NULL == eqn_next(ep, ep->data[(int)ep->cur], NULL, 0))
835 EQN_MSG(MANDOCERR_EQNEOF, ep);
836 else
837 return(1);
838
839 return(0);
840 }
841
842 static int
843 eqn_do_define(struct eqn_node *ep)
844 {
845 const char *start;
846 size_t sz;
847 struct eqn_def *def;
848 int i;
849
850 if (NULL == (start = eqn_nextrawtok(ep, &sz))) {
851 EQN_MSG(MANDOCERR_EQNEOF, ep);
852 return(0);
853 }
854
855 /*
856 * Search for a key that already exists.
857 * Create a new key if none is found.
858 */
859
860 if (NULL == (def = eqn_def_find(ep, start, sz))) {
861 /* Find holes in string array. */
862 for (i = 0; i < (int)ep->defsz; i++)
863 if (0 == ep->defs[i].keysz)
864 break;
865
866 if (i == (int)ep->defsz) {
867 ep->defsz++;
868 ep->defs = mandoc_realloc
869 (ep->defs, ep->defsz *
870 sizeof(struct eqn_def));
871 ep->defs[i].key = ep->defs[i].val = NULL;
872 }
873
874 ep->defs[i].keysz = sz;
875 ep->defs[i].key = mandoc_realloc
876 (ep->defs[i].key, sz + 1);
877
878 memcpy(ep->defs[i].key, start, sz);
879 ep->defs[i].key[(int)sz] = '\0';
880 def = &ep->defs[i];
881 }
882
883 start = eqn_next(ep, ep->data[(int)ep->cur], &sz, 0);
884
885 if (NULL == start) {
886 EQN_MSG(MANDOCERR_EQNEOF, ep);
887 return(0);
888 }
889
890 def->valsz = sz;
891 def->val = mandoc_realloc(def->val, sz + 1);
892 memcpy(def->val, start, sz);
893 def->val[(int)sz] = '\0';
894 return(1);
895 }
896
897 static int
898 eqn_do_gfont(struct eqn_node *ep)
899 {
900
901 if (NULL == eqn_nextrawtok(ep, NULL)) {
902 EQN_MSG(MANDOCERR_EQNEOF, ep);
903 return(0);
904 }
905 return(1);
906 }
907
908 static int
909 eqn_do_gsize(struct eqn_node *ep)
910 {
911 const char *start;
912 size_t sz;
913
914 if (NULL == (start = eqn_nextrawtok(ep, &sz))) {
915 EQN_MSG(MANDOCERR_EQNEOF, ep);
916 return(0);
917 }
918 ep->gsize = mandoc_strntoi(start, sz, 10);
919 return(1);
920 }
921
922 static int
923 eqn_do_undef(struct eqn_node *ep)
924 {
925 const char *start;
926 struct eqn_def *def;
927 size_t sz;
928
929 if (NULL == (start = eqn_nextrawtok(ep, &sz))) {
930 EQN_MSG(MANDOCERR_EQNEOF, ep);
931 return(0);
932 } else if (NULL != (def = eqn_def_find(ep, start, sz)))
933 def->keysz = 0;
934
935 return(1);
936 }
937
938 static struct eqn_def *
939 eqn_def_find(struct eqn_node *ep, const char *key, size_t sz)
940 {
941 int i;
942
943 for (i = 0; i < (int)ep->defsz; i++)
944 if (ep->defs[i].keysz && STRNEQ(ep->defs[i].key,
945 ep->defs[i].keysz, key, sz))
946 return(&ep->defs[i]);
947
948 return(NULL);
949 }