xref: /plan9/sys/src/cmd/aux/antiword/drawfile.c (revision 25b329d522281a8cdd35da0dcc08c3fc621059a9)
1 /*
2  * drawfile.c
3  * Copyright (C) 2005 A.J. van Os; Released under GPL
4  *
5  * Description:
6  * Functions to process with the Draw diagram
7  */
8 
9 #include <string.h>
10 #include "DeskLib:Error.h"
11 #include "DeskLib:SWI.h"
12 #include "drawfile.h"
13 #include "antiword.h"
14 
15 #define DRAWFILE_OBJECT_TOO_SMALL	200
16 #define DRAWFILE_NO_TEXT		201
17 #define DRAWFILE_BAD_CHARACTER		202
18 #define DRAWFILE_SMALL_MEMORY		203
19 #define DRAWFILE_PATH_WITHOUT_LINES	204
20 #define DRAWFILE_BAD_PATH_TYPE		205
21 #define DRAWFILE_PATH_WITHOUT_END	206
22 #define DRAWFILE_BAD_SPRITE_SIZE	207
23 #define DRAWFILE_BAD_JPEG_SIZE		208
24 #define DRAWFILE_TOO_SMALL		209
25 #define DRAWFILE_NOT_A_DRAWFILE		210
26 #define DRAWFILE_OBJECT_SIZE		211
27 #define DRAWFILE_MANY_FONTTABLES	212
28 #define DRAWFILE_TEXT_NO_FONT		213
29 #define DRAWFILE_OBJECT_UNEXPECTED	214
30 #define DRAWFILE_SIZE_ERROR		215
31 
32 typedef struct drawfile_error_tag {
33 	int		iErrorNumber;
34 	const char	*szErrorText;
35 } drawfile_error_type;
36 
37 static const drawfile_error_type atErrors[] = {
38 	{ DRAWFILE_OBJECT_TOO_SMALL, "Object too small"},
39 	{ DRAWFILE_NO_TEXT, "Text object without text"},
40 	{ DRAWFILE_BAD_CHARACTER, "Bad character in string"},
41 	{ DRAWFILE_SMALL_MEMORY, "Not enough memory reserved"},
42 	{ DRAWFILE_PATH_WITHOUT_LINES, "This path has no lines"},
43 	{ DRAWFILE_BAD_PATH_TYPE, "Bad path-type in path"},
44 	{ DRAWFILE_PATH_WITHOUT_END, "No end of path seen"},
45 	{ DRAWFILE_BAD_SPRITE_SIZE, "Bad sprite size"},
46 	{ DRAWFILE_BAD_JPEG_SIZE, "Bad jpeg size"},
47 	{ DRAWFILE_TOO_SMALL, "Too small to be a drawfile"},
48 	{ DRAWFILE_NOT_A_DRAWFILE, "Not a drawfile"},
49 	{ DRAWFILE_OBJECT_SIZE, "Object with incorrect size"},
50 	{ DRAWFILE_MANY_FONTTABLES, "More than one font table"},
51 	{ DRAWFILE_TEXT_NO_FONT, "Text, but no font table seen"},
52 	{ DRAWFILE_OBJECT_UNEXPECTED, "Unexpected object type"},
53 	{ DRAWFILE_SIZE_ERROR, "Sizes don't match"},
54 };
55 
56 
57 /*
58  * pFillError - error number to error struct
59  */
60 static os_error *
pFillError(int iErrorNumber)61 pFillError(int iErrorNumber)
62 {
63 	static os_error		tError;
64 	const drawfile_error_type	*pTmp;
65 	const char	*szErrorText;
66 
67 	szErrorText = "Unknown error";
68 	for (pTmp = atErrors; pTmp < atErrors + elementsof(atErrors); pTmp++) {
69 		if (iErrorNumber == pTmp->iErrorNumber) {
70 			szErrorText = pTmp->szErrorText;
71 			break;
72 		}
73 	}
74 	tError.errnum = iErrorNumber;
75 	strncpy(tError.errmess, szErrorText, sizeof(tError.errmess) - 1);
76 	tError.errmess[sizeof(tError.errmess) - 1] = '\0';
77 	DBG_DEC(tError.errnum);
78 	DBG_MSG(tError.errmess);
79 	return &tError;
80 } /* end of pFillError */
81 
82 /*
83  * Drawfile_BBox - Find the bounding box of a diagram
84  */
85 os_error *
Drawfile_Bbox(drawfile_bbox_flags flags,drawfile_diagram const * diagram,int size,os_trfm const * trfm,wimp_box * bbox)86 Drawfile_Bbox(drawfile_bbox_flags flags,
87 	drawfile_diagram const *diagram,
88 	int size,
89 	os_trfm const *trfm,
90 	wimp_box *bbox)
91 {
92 	return SWI(5, 0, DrawFile_BBox | XOS_Bit,
93 		flags, diagram, size, trfm, bbox);
94 } /* end of Drawfile_Bbox */
95 
96 /*
97  * Drawfile_CreateDiagram - create an empty drawfile diagram
98  */
99 os_error *
Drawfile_CreateDiagram(drawfile_info * pInfo,size_t tMemorySize,const char * szCreator,wimp_box tBbox)100 Drawfile_CreateDiagram(drawfile_info *pInfo, size_t tMemorySize,
101 	const char *szCreator, wimp_box tBbox)
102 {
103 	drawfile_diagram	*pDiag;
104 
105 	if (tMemorySize < offsetof(drawfile_diagram, objects)) {
106 		return pFillError(DRAWFILE_SMALL_MEMORY);
107 	}
108 	pDiag = (drawfile_diagram *)pInfo->data;
109 	strncpy(pDiag->tag, "Draw", 4);
110 	pDiag->major_version = 201;
111 	pDiag->minor_version = 0;
112 	strncpy(pDiag->source, szCreator, sizeof(pDiag->source));
113 	pDiag->bbox = tBbox;
114 	/* Memory in use */
115 	pInfo->length = offsetof(drawfile_diagram, objects);
116 	return NULL;
117 } /* end of Drawfile_CreateDiagram */
118 
119 /*
120  * Drawfile_AppendObject - append an object to a diagram
121  */
122 os_error *
Drawfile_AppendObject(drawfile_info * pInfo,size_t tMemorySize,const drawfile_object * pObject,BOOL bRebind)123 Drawfile_AppendObject(drawfile_info *pInfo, size_t tMemorySize,
124 	const drawfile_object *pObject, BOOL bRebind)
125 {
126 	wimp_box	*pMainBbox;
127 	const wimp_box	*pBbox;
128 	byte		*pAfter;
129 
130 	if (tMemorySize < pInfo->length + pObject->size) {
131 		return pFillError(DRAWFILE_OBJECT_TOO_SMALL);
132 	}
133 	/* After the last object */
134 	pAfter = (byte *)pInfo->data + pInfo->length;
135 	/* Copy in the new data */
136 	memcpy(pAfter, pObject, pObject->size);
137 	/* Rebind if needed */
138 	if (bRebind) {
139 		pMainBbox = &((drawfile_diagram *)pInfo->data)->bbox;
140 		switch (pObject->type) {
141 		case drawfile_TYPE_FONT_TABLE:
142 			pBbox = NULL;
143 			break;
144 		case drawfile_TYPE_TEXT:
145 			pBbox = &pObject->data.text.bbox;
146 			break;
147 		case drawfile_TYPE_PATH:
148 			pBbox = &pObject->data.path.bbox;
149 			break;
150 		case drawfile_TYPE_SPRITE:
151 			pBbox = &pObject->data.sprite.bbox;
152 			break;
153 		case drawfile_TYPE_GROUP:
154 			pBbox = &pObject->data.group.bbox;
155 			break;
156 		case drawfile_TYPE_TAGGED:
157 			pBbox = &pObject->data.tagged.bbox;
158 			break;
159 		case drawfile_TYPE_TEXT_AREA:
160 			pBbox = &pObject->data.text_area.bbox;
161 			break;
162 		case drawfile_TYPE_TEXT_COLUMN:
163 			pBbox = NULL;
164 			break;
165 		case drawfile_TYPE_OPTIONS:
166 			pBbox = &pObject->data.options.bbox;
167 			break;
168 		case drawfile_TYPE_TRFM_TEXT:
169 			pBbox = &pObject->data.trfm_text.bbox;
170 			break;
171 		case drawfile_TYPE_TRFM_SPRITE:
172 			pBbox = &pObject->data.trfm_sprite.bbox;
173 			break;
174 		case drawfile_TYPE_JPEG:
175 			pBbox = &pObject->data.jpeg.bbox;
176 			break;
177 		default:
178 			pBbox = NULL;
179 			break;
180 		}
181 		if (pBbox != NULL) {
182 			if (pBbox->min.x < pMainBbox->min.x) {
183 				pMainBbox->min.x = pBbox->min.x;
184 			}
185 			if (pBbox->min.y < pMainBbox->min.y) {
186 				pMainBbox->min.y = pBbox->min.y;
187 			}
188 			if (pBbox->max.x > pMainBbox->max.x) {
189 				pMainBbox->max.x = pBbox->max.x;
190 			}
191 			if (pBbox->max.y > pMainBbox->max.y) {
192 				pMainBbox->max.y = pBbox->max.y;
193 			}
194 		}
195 	}
196 	/* Memory in use */
197 	pInfo->length += pObject->size;
198 	return NULL;
199 } /* end of Drawfile_AppendObject */
200 
201 /*
202  * Replaces the draw_render_diag function from RISC_OSLib
203  */
204 os_error *
Drawfile_RenderDiagram(drawfile_info * pInfo,window_redrawblock * pRedraw,double dScale)205 Drawfile_RenderDiagram(drawfile_info *pInfo, window_redrawblock *pRedraw,
206 	double dScale)
207 {
208 	int	aiTransform[6];
209 
210 	fail(pInfo == NULL);
211 	fail(pInfo->data == NULL);
212 	fail(pRedraw == NULL);
213 	fail(dScale < 0.01);
214 
215 	aiTransform[0] = (int)(dScale * 0x10000);
216 	aiTransform[1] = 0;
217 	aiTransform[2] = 0;
218 	aiTransform[3] = (int)(dScale * 0x10000);
219 	aiTransform[4] = (pRedraw->rect.min.x - pRedraw->scroll.x) * 256;
220 	aiTransform[5] = (pRedraw->rect.max.y - pRedraw->scroll.y) * 256;
221 
222 	return SWI(6, 0, DrawFile_Render | XOS_Bit,
223 		0, pInfo->data, pInfo->length, aiTransform, &pRedraw->rect, 0);
224 } /* end of Drawfile_RenderDiagram */
225 
226 /*
227  * pVerifyText - verify a text object
228  */
229 static os_error *
pVerifyText(const drawfile_text * pText)230 pVerifyText(const drawfile_text *pText)
231 {
232 	const unsigned char	*pucTmp;
233 
234 	if (pText->text[0] == '\0') {
235 		return pFillError(DRAWFILE_NO_TEXT);
236 	}
237 	pucTmp = (const unsigned char *)pText->text;
238 	while (*pucTmp != '\0') {
239 		if (*pucTmp < 0x20 || *pucTmp == 0x7f) {
240 			return pFillError(DRAWFILE_BAD_CHARACTER);
241 		}
242 		pucTmp++;
243 	}
244 	return NULL;
245 } /* end of pVerifyText */
246 
247 /*
248  * pVerifyPath - verify a path object
249  */
250 static os_error *
pVerifyPath(const drawfile_path * pPath,int iSize)251 pVerifyPath(const drawfile_path *pPath, int iSize)
252 {
253 	const int	*piTmp;
254 	int	iElements;
255 	BOOL	bLine;
256 
257 	bLine = FALSE;
258 	iElements = (iSize - offsetof(drawfile_path, path)) / 4;
259 
260 	for (piTmp = pPath->path; piTmp < pPath->path + iElements; piTmp++) {
261 		switch(*piTmp) {
262 		case drawfile_PATH_END_PATH:
263 			if (bLine) {
264 				return NULL;
265 			}
266 			return pFillError(DRAWFILE_PATH_WITHOUT_LINES);
267 		case drawfile_PATH_LINE_TO:
268 			bLine = TRUE;
269 			piTmp += 2;
270 			break;
271 		case drawfile_PATH_MOVE_TO:
272 			piTmp += 2;
273 			break;
274 		case drawfile_PATH_CLOSE_LINE:
275 			bLine = TRUE;
276 			break;
277 		default:
278 			return pFillError(DRAWFILE_BAD_PATH_TYPE);
279 		}
280 	}
281 	return pFillError(DRAWFILE_PATH_WITHOUT_END);
282 } /* end of pVerifyPath */
283 
284 /*
285  * pVerifySprite - verify a sprite object
286  */
287 static os_error *
pVerifySprite(const drawfile_sprite * pSprite,int iSize)288 pVerifySprite(const drawfile_sprite *pSprite, int iSize)
289 {
290 	iSize -= offsetof(drawfile_sprite, header);
291 	if (iSize < pSprite->header.offset_next) {
292 		DBG_DEC(iSize);
293 		DBG_DEC(pSprite->header.offset_next);
294 		return pFillError(DRAWFILE_BAD_SPRITE_SIZE);
295 	}
296 	return NULL;
297 } /* end of pVerifySprite */
298 
299 /*
300  * pVerifyJpeg - verify a jpeg object
301  */
302 static os_error *
pVerifyJpeg(const drawfile_jpeg * pJpeg,int iSize)303 pVerifyJpeg(const drawfile_jpeg *pJpeg, int iSize)
304 {
305 	iSize -= offsetof(drawfile_jpeg, data);
306 	if (iSize < pJpeg->len) {
307 		DBG_DEC(iSize);
308 		DBG_DEC(pJpeg->len);
309 		return pFillError(DRAWFILE_BAD_JPEG_SIZE);
310 	}
311 	return NULL;
312 } /* end of pVerifyJpeg */
313 
314 /*
315  * Drawfile_VerifyDiagram - Verify the diagram generated from the Word file
316  *
317  * returns NULL if the diagram is correct
318  */
319 os_error *
Drawfile_VerifyDiagram(drawfile_info * pInfo)320 Drawfile_VerifyDiagram(drawfile_info *pInfo)
321 {
322 	drawfile_diagram	*pDiag;
323 	drawfile_object	*pObj;
324 	os_error	*pError;
325 	const char	*pcTmp;
326 	int		iToGo, iFontTables;
327 	BOOL		bTypeFontTable;
328 
329 	TRACE_MSG("Drawfile_VerifyDiagram");
330 
331 	fail(pInfo == NULL);
332 
333   	if (pInfo->length < offsetof(drawfile_diagram, objects)) {
334 		return pFillError(DRAWFILE_TOO_SMALL);
335   	}
336 
337 	pDiag = (drawfile_diagram *)pInfo->data;
338 	if (strncmp(pDiag->tag, "Draw", 4) != 0 ||
339 	    pDiag->major_version != 201 ||
340 	    pDiag->minor_version != 0) {
341 		return pFillError(DRAWFILE_NOT_A_DRAWFILE);
342 	}
343 
344 	iToGo = pInfo->length - offsetof(drawfile_diagram, objects);
345 	pcTmp = (const char *)pInfo->data + offsetof(drawfile_diagram, objects);
346 	iFontTables = 0;
347 	bTypeFontTable = FALSE;
348 
349 	while (iToGo > 0) {
350 		pObj = (drawfile_object *)pcTmp;
351 		if (pObj->size < 0 || pObj->size % 4 != 0) {
352 			return pFillError(DRAWFILE_OBJECT_SIZE);
353 		}
354 		switch (pObj->type) {
355 		case drawfile_TYPE_FONT_TABLE:
356 			if (bTypeFontTable) {
357 				return pFillError(DRAWFILE_MANY_FONTTABLES);
358 			}
359 			bTypeFontTable = TRUE;
360 			break;
361 		case drawfile_TYPE_TEXT:
362 			if (pObj->data.text.style.font_ref != 0 &&
363 			    !bTypeFontTable) {
364 				return pFillError(DRAWFILE_TEXT_NO_FONT);
365 			}
366 			pError = pVerifyText(&pObj->data.text);
367 			if (pError != NULL) {
368 				return pError;
369 			}
370 			break;
371 		case drawfile_TYPE_PATH:
372 			pError = pVerifyPath(&pObj->data.path,
373 				pObj->size - offsetof(drawfile_object, data));
374 			if (pError != NULL) {
375 				return pError;
376 			}
377 			break;
378 		case drawfile_TYPE_SPRITE:
379 			pError = pVerifySprite(&pObj->data.sprite,
380 				pObj->size - offsetof(drawfile_object, data));
381 			if (pError != NULL) {
382 				return pError;
383 			}
384 			break;
385 		case drawfile_TYPE_JPEG:
386 			pError = pVerifyJpeg(&pObj->data.jpeg,
387 				pObj->size - offsetof(drawfile_object, data));
388 			if (pError != NULL) {
389 				return pError;
390 			}
391 			break;
392 		default:
393 			DBG_DEC(pObj->type);
394 			return pFillError(DRAWFILE_OBJECT_UNEXPECTED);
395 		}
396 		pcTmp += pObj->size;
397 		iToGo -= pObj->size;
398 	}
399 	if (iToGo < 0) {
400 		return pFillError(DRAWFILE_SIZE_ERROR);
401 	}
402 	return NULL;
403 } /* end of Drawfile_VerifyDiagram */
404 
405 /*
406  * Drawfile_QueryBox - Find the bounding box of a diagram
407  */
408 void
Drawfile_QueryBox(drawfile_info * pInfo,wimp_box * pRect,BOOL bScreenUnits)409 Drawfile_QueryBox(drawfile_info *pInfo, wimp_box *pRect, BOOL bScreenUnits)
410 {
411 	fail(pInfo == NULL);
412 	fail(pRect == NULL);
413 
414 	Error_CheckFatal(Drawfile_Bbox(0,
415 		pInfo->data, pInfo->length, NULL, pRect));
416 	if (bScreenUnits) {
417 		pRect->min.x = Drawfile_DrawToScreen(pRect->min.x);
418 		pRect->min.y = Drawfile_DrawToScreen(pRect->min.y);
419 		pRect->max.x = Drawfile_DrawToScreen(pRect->max.x);
420 		pRect->max.y = Drawfile_DrawToScreen(pRect->max.y);
421 	}
422 } /* end of Drawfile_QueryBox */
423