shithub: pdffs

ref: 34238e0feb181a0c120561486d4c86b054d112bc
dir: /object.c/

#include <u.h>
#include <libc.h>
#include <ctype.h>
#include "pdf.h"

Object *
pdfobject(char *p, char **e, int len)
{
	Object *o;
	char *s, *p0;
	int sz, gen;

	o = nil;
	for(; len > 0 && isws(*p); p++, len--);
	if(len < 1){
		werrstr("too short");
		goto err;
	}

	if(*p < 1){
		werrstr("unexpected non-ascii char");
		goto err;
	}

	switch(*p){
	case '<': /* dictionary or a string */
		if(len < 2){
			werrstr("too short");
			goto err;
		}
		if(p[1] == '<'){ /* dictionary */
			o = pdfdict(p, e, len);
			break;
		}
		/* fall through */

	case '(': /* string */
		if((sz = pdfstring(p, e, len)) < 0)
			goto err;
		if((o = malloc(sizeof(*o)+sz+1)) != nil){
			o->type = Ostr;
			o->str = (char*)(o+1);
			strcpy(o->str, p);
		}
		break;

	case '/':
		if((sz = pdfname(p, e, len)) < 0)
			goto err;
		if((o = malloc(sizeof(*o)+sz+1)) != nil){
			o->type = Oname;
			o->str = (char*)(o+1);
			strcpy(o->str, p);
		}
		break;

	case '[':
		o = pdfarray(p, e, len);
		break;

	default:
		if(isdigit(*p)){
			if((o = malloc(sizeof(*o))) != nil){
				o->type = Onum;
				o->num = strtod(p, e);
				sz = len - (*e - p);
				p0 = *e;
				s = p0;
				for(; sz > 0 && isws(*s); s++, sz--);
				if(sz > 0 && isdigit(*s)){
					gen = strtod(s, &p);
					sz -= (p - s);
					s = p;
					for(; sz > 0 && isws(*s); s++, sz--);
					if(sz > 0 && *s == 'R'){ /* indirect object */
						s++;
						o->type = Oindir;
						o->indir.id = o->num;
						o->indir.gen = gen;
						p0 = s;
					}
				}
				*e = p0;
			}
			break;
		}
		werrstr("unexpected char %c", *p);
		goto err;
	}

	if(o != nil)
		return o;
err:
	werrstr("object: %r");
	freeobject(o);
	return nil;
}

void
freeobject(Object *o)
{
	if(o == nil)
		return;
	free(o);
}