#define _TRACE_PARSER_HPP_
-#include <cassert>
-
#include <iostream>
+#include <list>
-#include <zlib.h>
-
+#include "trace_file.hpp"
#include "trace_format.hpp"
#include "trace_model.hpp"
namespace Trace {
+struct ParseBookmark
+{
+ File::Offset offset;
+ unsigned next_call_no;
+};
+
+
class Parser
{
protected:
- gzFile file;
+ File *file;
+
+ enum Mode {
+ FULL = 0,
+ SCAN,
+ SKIP
+ };
+
+ typedef std::list<Call *> CallList;
+ CallList calls;
+
+ // Helper template that extends a base signature structure, with additional
+ // parsing information.
+ template< class T >
+ struct SigState : public T {
+ // Offset in the file of where signature was defined. It is used when
+ // reparsing to determine whether the signature definition is to be
+ // expected next or not.
+ File::Offset offset;
+ };
+
+ typedef SigState<FunctionSig> FunctionSigState;
+ typedef SigState<StructSig> StructSigState;
+ typedef SigState<EnumSig> EnumSigState;
+ typedef SigState<BitmaskSig> BitmaskSigState;
+
+ typedef std::vector<FunctionSigState *> FunctionMap;
+ typedef std::vector<StructSigState *> StructMap;
+ typedef std::vector<EnumSigState *> EnumMap;
+ typedef std::vector<BitmaskSigState *> BitmaskMap;
+
+ FunctionMap functions;
+ StructMap structs;
+ EnumMap enums;
+ BitmaskMap bitmasks;
+
+ unsigned next_call_no;
+
public:
- Parser() {
- file = NULL;
- }
-
- ~Parser() {
- close();
- }
-
- bool open(const char *filename) {
- unsigned long long version;
-
- file = gzopen(filename, "rb");
- if (!file) {
- return false;
- }
-
- version = read_uint();
- if (version != TRACE_VERSION) {
- std::cerr << "error: unsupported format version" << version << "\n";
- return false;
- }
-
- return true;
- }
-
- void close(void) {
- if (file) {
- gzclose(file);
- file = NULL;
- }
- }
-
- Call *parse_call(void) {
- Call *call = new Call;
- call->name = read_string();
- do {
- int c = read_byte();
- switch(c) {
- case Trace::CALL_END:
- return call;
- case Trace::CALL_ARG:
- call->args.push_back(parse_arg());
- break;
- case Trace::CALL_RET:
- call->ret = parse_value();
- break;
- default:
- std::cerr << "error: unknown call detail " << c << "\n";
- assert(0);
- /* fallthrough */
- case -1:
- delete call;
- return NULL;
- }
- } while(true);
- }
-
- Arg parse_arg(void) {
- std::string name = read_string();
- Value *value = parse_value();
- return Arg(name, value);
- }
-
- Value *parse_value(void) {
- int c;
- c = read_byte();
- switch(c) {
- case Trace::TYPE_NULL:
- return new Null;
- case Trace::TYPE_FALSE:
- return new Bool(false);
- case Trace::TYPE_TRUE:
- return new Bool(true);
- case Trace::TYPE_SINT:
- return parse_sint();
- case Trace::TYPE_UINT:
- return parse_uint();
- case Trace::TYPE_FLOAT:
- return parse_float();
- case Trace::TYPE_DOUBLE:
- return parse_double();
- case Trace::TYPE_STRING:
- return parse_string();
- case Trace::TYPE_CONST:
- return parse_const();
- case Trace::TYPE_BITMASK:
- return parse_bitmask();
- case Trace::TYPE_ARRAY:
- return parse_array();
- case Trace::TYPE_STRUCT:
- return parse_struct();
- case Trace::TYPE_BLOB:
- return parse_blob();
- case Trace::TYPE_POINTER:
- return parse_pointer();
- case Trace::TYPE_OPAQUE:
- return parse_opaque();
- default:
- std::cerr << "error: unknown type " << c << "\n";
- assert(0);
- return NULL;
- }
- }
-
- Value *parse_sint() {
- return new SInt(-read_uint());
- }
-
- Value *parse_uint() {
- return new UInt(read_uint());
- }
-
- Value *parse_float() {
- float value;
- gzread(file, &value, sizeof value);
- return new Float(value);
- }
-
- Value *parse_double() {
- double value;
- gzread(file, &value, sizeof value);
- return new Float(value);
- }
-
- Value *parse_string() {
- return new String(read_string());
- }
-
- Value *parse_const() {
- std::string name = read_string();
- Value *value = parse_value();
- return new Const(name, value);
- }
-
- Value *parse_bitmask() {
- unsigned long long value = 0;
- int c;
- do {
- c = read_byte();
- switch(c) {
- case Trace::TYPE_SINT:
- value |= -read_uint();
- break;
- case Trace::TYPE_UINT:
- value |= read_uint();
- break;
- case Trace::TYPE_CONST:
- read_string();
- break;
- case Trace::TYPE_NULL:
- goto done;
- default:
- std::cerr << "error: uexpected type " << c << "\n";
- assert(0);
+ unsigned long long version;
+
+ Parser();
+
+ ~Parser();
+
+ bool open(const char *filename);
+
+ void close(void);
+
+ Call *parse_call(void) {
+ return parse_call(FULL);
+ }
+
+ bool supportsOffsets() const
+ {
+ return file->supportsOffsets();
+ }
+
+ void getBookmark(ParseBookmark &bookmark);
+
+ void setBookmark(const ParseBookmark &bookmark);
+
+ int percentRead()
+ {
+ return file->percentRead();
+ }
+
+ Call *scan_call() {
+ return parse_call(SCAN);
+ }
+
+protected:
+ Call *parse_call(Mode mode);
+
+ FunctionSig *parse_function_sig(void);
+ StructSig *parse_struct_sig();
+ EnumSig *parse_enum_sig();
+ BitmaskSig *parse_bitmask_sig();
+
+ Call *parse_Call(Mode mode);
+
+ void parse_enter(Mode mode);
+
+ Call *parse_leave(Mode mode);
+
+ bool parse_call_details(Call *call, Mode mode);
+
+ void parse_arg(Call *call, Mode mode);
+
+ Value *parse_value(void);
+ void scan_value(void);
+ inline Value *parse_value(Mode mode) {
+ if (mode == FULL) {
+ return parse_value();
+ } else {
+ scan_value();
return NULL;
- }
- } while(true);
-done:
- return new UInt(value);
- }
-
- Value *parse_array(void) {
- size_t len = read_uint();
- Array *array = new Array(len);
- for (size_t i = 0; i < len; ++i) {
- array->values[i] = parse_value();
- }
- return array;
- }
-
- Value *parse_blob(void) {
- size_t size = read_uint();
- Blob *blob = new Blob(size);
- if (size) {
- gzread(file, blob->buf, size);
- }
- return blob;
- }
-
- Value *parse_pointer() {
- unsigned long long addr;
- Value *value;
- addr = read_uint();
- value = parse_value();
- if (!value)
- value = new UInt(addr);
- return value;
- }
-
- Value *parse_struct() {
- std::string name;
- /* XXX */
- name = read_string();
- while(name.length()) {
- Value *value = parse_value();
- std::cout << " " << name << " = " << value << "\n";
- name = read_string();
- }
- return NULL;
- }
-
- Value *parse_opaque() {
- unsigned long long addr;
- addr = read_uint();
- /* XXX */
- return new UInt(addr);
- }
-
- std::string read_string(void) {
- size_t len = read_uint();
- if (!len) {
- return std::string();
- }
- char * buf = new char[len];
- gzread(file, buf, len);
- std::string value(buf, len);
- delete [] buf;
-#ifdef TRACE_VERBOSE
- std::cerr << '"' << value << '"' << "\n";
-#endif
- return value;
- }
-
- unsigned long long read_uint(void) {
- unsigned long long value = 0;
- int c;
- unsigned shift = 0;
- do {
- c = gzgetc(file);
- if (c == -1) {
- break;
- }
- value |= (unsigned long long)(c & 0x7f) << shift;
- shift += 7;
- } while(c & 0x80);
-#ifdef TRACE_VERBOSE
- std::cerr << value << "\n";
-#endif
- return value;
- }
-
- int read_byte(void) {
- int c = gzgetc(file);
-#ifdef TRACE_VERBOSE
- if (c < 0)
- std::cerr << "EOF" << "\n";
- else
- std::cerr << "0x" << std::hex << c << "\n";
-#endif
- return c;
- }
+ }
+ }
+
+ Value *parse_sint();
+ void scan_sint();
+
+ Value *parse_uint();
+ void scan_uint();
+
+ Value *parse_float();
+ void scan_float();
+
+ Value *parse_double();
+ void scan_double();
+
+ Value *parse_string();
+ void scan_string();
+
+ Value *parse_enum();
+ void scan_enum();
+
+ Value *parse_bitmask();
+ void scan_bitmask();
+
+ Value *parse_array(void);
+ void scan_array(void);
+
+ Value *parse_blob(void);
+ void scan_blob(void);
+
+ Value *parse_struct();
+ void scan_struct();
+
+ Value *parse_opaque();
+ void scan_opaque();
+
+ const char * read_string(void);
+ void skip_string(void);
+
+ unsigned long long read_uint(void);
+ void skip_uint(void);
+
+ inline int read_byte(void);
+ inline void skip_byte(void);
};