first attempts at refactoring this code to make it more readable;

mostly changing to dictionaries for format strings


svn:r450
This commit is contained in:
Niels Provos 2007-09-23 03:49:28 +00:00
parent a4cc3d148a
commit 3c1a6a68d2

View File

@ -85,20 +85,18 @@ class Struct:
print >>file, ' uint8_t %s_set;' % entry.Name() print >>file, ' uint8_t %s_set;' % entry.Name()
print >>file, '};\n' print >>file, '};\n'
print >>file, ( print >>file, \
'struct %s *%s_new(void);\n' % (self._name, self._name) + """struct %(name)s *%(name)s_new(void);
'void %s_free(struct %s *);\n' % (self._name, self._name) + void %(name)s_free(struct %(name)s *);
'void %s_clear(struct %s *);\n' % (self._name, self._name) + void %(name)s_clear(struct %(name)s *);
'void %s_marshal(struct evbuffer *, const struct %s *);\n' % ( void %(name)s_marshal(struct evbuffer *, const struct %(name)s *);
self._name, self._name) + int %(name)s_unmarshal(struct %(name)s *, struct evbuffer *);
'int %s_unmarshal(struct %s *, struct evbuffer *);\n' % ( int %(name)s_complete(struct %(name)s *);
self._name, self._name) + void evtag_marshal_%(name)s(struct evbuffer *, uint8_t,
'int %s_complete(struct %s *);' % (self._name, self._name) const struct %(name)s *);
) int evtag_unmarshal_%(name)s(struct evbuffer *, uint8_t,
print >>file, ('void evtag_marshal_%s(struct evbuffer *, uint8_t, ' struct %(name)s *);""" % { 'name' : self._name }
'const struct %s *);') % ( self._name, self._name)
print >>file, ('int evtag_unmarshal_%s(struct evbuffer *, uint8_t, '
'struct %s *);') % ( self._name, self._name)
# Write a setting function of every variable # Write a setting function of every variable
for entry in self._entries: for entry in self._entries:
@ -118,20 +116,22 @@ class Struct:
' */\n') % self._name ' */\n') % self._name
# Creation # Creation
print >>file, ( 'struct %s *\n' % self._name + print >>file, (
'%s_new(void)\n' % self._name + 'struct %(name)s *\n'
'%(name)s_new(void)\n'
'{\n' '{\n'
' struct %s *tmp;\n' % self._name + ' struct %(name)s *tmp;\n'
' if ((tmp = malloc(sizeof(struct %s))) == NULL) {\n' ' if ((tmp = malloc(sizeof(struct %(name)s))) == NULL) {\n'
' event_warn("%%s: malloc", __func__);\n' ' event_warn("%%s: malloc", __func__);\n'
' return (NULL);\n' % self._name + ' return (NULL);\n'
' }' ' }') % { 'name' : self._name }
)
for entry in self._entries: for entry in self._entries:
self.PrintIdented(file, ' ', entry.CodeNew('tmp')) self.PrintIdented(file, ' ', entry.CodeNew('tmp'))
print >>file, ' tmp->%s_set = 0;\n' % entry.Name() print >>file, ' tmp->%s_set = 0;\n' % entry.Name()
print >>file, (' return (tmp);\n' print >>file, (
' return (tmp);\n'
'}\n') '}\n')
# Adding # Adding
@ -152,10 +152,9 @@ class Struct:
# Clearing # Clearing
print >>file, ( 'void\n' print >>file, ( 'void\n'
'%s_clear(struct %s *tmp)\n' % ( '%(name)s_clear(struct %(name)s *tmp)\n'
self._name, self._name)+
'{' '{'
) ) % { 'name' : self._name }
for entry in self._entries: for entry in self._entries:
self.PrintIdented(file, ' ', entry.CodeClear('tmp')) self.PrintIdented(file, ' ', entry.CodeClear('tmp'))
@ -163,10 +162,10 @@ class Struct:
# Freeing # Freeing
print >>file, ( 'void\n' print >>file, ( 'void\n'
'%s_free(struct %s *tmp)\n' % ( '%(name)s_free(struct %(name)s *tmp)\n'
self._name, self._name)+
'{' '{'
) ) % { 'name' : self._name }
for entry in self._entries: for entry in self._entries:
self.PrintIdented(file, ' ', entry.CodeFree('tmp')) self.PrintIdented(file, ' ', entry.CodeFree('tmp'))
@ -175,9 +174,9 @@ class Struct:
# Marshaling # Marshaling
print >>file, ('void\n' print >>file, ('void\n'
'%s_marshal(struct evbuffer *evbuf, ' '%(name)s_marshal(struct evbuffer *evbuf, '
'const struct %s *tmp)' % (self._name, self._name) + 'const struct %(name)s *tmp)'
'{') '{') % { 'name' : self._name }
for entry in self._entries: for entry in self._entries:
indent = ' ' indent = ' '
# Optional entries do not have to be set # Optional entries do not have to be set
@ -194,16 +193,15 @@ class Struct:
# Unmarshaling # Unmarshaling
print >>file, ('int\n' print >>file, ('int\n'
'%s_unmarshal(struct %s *tmp, ' '%(name)s_unmarshal(struct %(name)s *tmp, '
' struct evbuffer *evbuf)\n' % ( ' struct evbuffer *evbuf)\n'
self._name, self._name) +
'{\n' '{\n'
' uint8_t tag;\n' ' uint8_t tag;\n'
' while (EVBUFFER_LENGTH(evbuf) > 0) {\n' ' while (EVBUFFER_LENGTH(evbuf) > 0) {\n'
' if (evtag_peek(evbuf, &tag) == -1)\n' ' if (evtag_peek(evbuf, &tag) == -1)\n'
' return (-1);\n' ' return (-1);\n'
' switch (tag) {\n' ' switch (tag) {\n'
) ) % { 'name' : self._name }
for entry in self._entries: for entry in self._entries:
print >>file, ' case %s:\n' % self.EntryTagName(entry) print >>file, ' case %s:\n' % self.EntryTagName(entry)
if not entry.Array(): if not entry.Array():
@ -224,8 +222,9 @@ class Struct:
' }\n' ' }\n'
' }\n' ) ' }\n' )
# Check if it was decoded completely # Check if it was decoded completely
print >>file, ( ' if (%s_complete(tmp) == -1)\n' % self._name + print >>file, ( ' if (%(name)s_complete(tmp) == -1)\n'
' return (-1);') ' return (-1);'
) % { 'name' : self._name }
# Successfully decoded # Successfully decoded
print >>file, ( ' return (0);\n' print >>file, ( ' return (0);\n'
@ -234,8 +233,8 @@ class Struct:
# Checking if a structure has all the required data # Checking if a structure has all the required data
print >>file, ( print >>file, (
'int\n' 'int\n'
'%s_complete(struct %s *msg)\n' % (self._name, self._name) + '%(name)s_complete(struct %(name)s *msg)\n'
'{' ) '{' ) % { 'name' : self._name }
for entry in self._entries: for entry in self._entries:
self.PrintIdented( self.PrintIdented(
file, ' ', file, ' ',
@ -247,10 +246,8 @@ class Struct:
# Complete message unmarshaling # Complete message unmarshaling
print >>file, ( print >>file, (
'int\n' 'int\n'
'evtag_unmarshal_%s(struct evbuffer *evbuf, uint8_t need_tag, ' 'evtag_unmarshal_%(name)s(struct evbuffer *evbuf, '
' struct %s *msg)' 'uint8_t need_tag, struct %(name)s *msg)\n'
) % (self._name, self._name)
print >>file, (
'{\n' '{\n'
' uint8_t tag;\n' ' uint8_t tag;\n'
' int res = -1;\n' ' int res = -1;\n'
@ -261,7 +258,7 @@ class Struct:
' || tag != need_tag)\n' ' || tag != need_tag)\n'
' goto error;\n' ' goto error;\n'
'\n' '\n'
' if (%s_unmarshal(msg, tmp) == -1)\n' ' if (%(name)s_unmarshal(msg, tmp) == -1)\n'
' goto error;\n' ' goto error;\n'
'\n' '\n'
' res = 0;\n' ' res = 0;\n'
@ -269,22 +266,22 @@ class Struct:
' error:\n' ' error:\n'
' evbuffer_free(tmp);\n' ' evbuffer_free(tmp);\n'
' return (res);\n' ' return (res);\n'
'}\n' ) % self._name '}\n' ) % { 'name' : self._name }
# Complete message marshaling # Complete message marshaling
print >>file, ( print >>file, (
'void\n' 'void\n'
'evtag_marshal_%s(struct evbuffer *evbuf, uint8_t tag, ' 'evtag_marshal_%(name)s(struct evbuffer *evbuf, uint8_t tag, '
'const struct %s *msg)\n' % (self._name, self._name) + 'const struct %(name)s *msg)\n'
'{\n' '{\n'
' struct evbuffer *_buf = evbuffer_new();\n' ' struct evbuffer *_buf = evbuffer_new();\n'
' assert(_buf != NULL);\n' ' assert(_buf != NULL);\n'
' evbuffer_drain(_buf, -1);\n' ' evbuffer_drain(_buf, -1);\n'
' %s_marshal(_buf, msg);\n' % self._name + ' %(name)s_marshal(_buf, msg);\n'
' evtag_marshal(evbuf, tag, EVBUFFER_DATA(_buf), ' ' evtag_marshal(evbuf, tag, EVBUFFER_DATA(_buf), '
'EVBUFFER_LENGTH(_buf));\n' 'EVBUFFER_LENGTH(_buf));\n'
' evbuffer_free(_buf);\n' ' evbuffer_free(_buf);\n'
'}\n' ) '}\n' ) % { 'name' : self._name }
class Entry: class Entry:
def __init__(self, type, name, tag): def __init__(self, type, name, tag):
@ -297,6 +294,14 @@ class Entry:
self._array = 0 self._array = 0
self._line_count = -1 self._line_count = -1
self._struct = None self._struct = None
self._refname = None
def GetTranslation(self):
return { "parent_name" : self._struct.Name(),
"name" : self._name,
"ctype" : self._ctype,
"refname" : self._refname
}
def SetStruct(self, struct): def SetStruct(self, struct):
self._struct = struct self._struct = struct
@ -338,17 +343,19 @@ class Entry:
return code return code
def CodeGet(self): def CodeGet(self):
code = [ 'int', code = (
'%s_%s_get(struct %s *msg, %s *value)' % ( 'int',
self._struct.Name(), self._name, '%(parent_name)s_%(name)s_get(struct %(parent_name)s *msg, '
self._struct.Name(), self._ctype), '%(ctype)s *value)',
'{', '{',
' if (msg->%s_set != 1)' % self._name, ' if (msg->%(name)s_set != 1)',
' return (-1);', ' return (-1);',
' *value = msg->%s_data;' % self._name, ' *value = msg->%(name)s_data;',
' return (0);', ' return (0);',
'}' ] '}' )
return code code = '\n'.join(code)
code = code % self.GetTranslation()
return code.split('\n')
def AssignFuncName(self): def AssignFuncName(self):
return '%s_%s_assign' % (self._struct.Name(), self._name) return '%s_%s_assign' % (self._struct.Name(), self._name)
@ -363,15 +370,16 @@ class Entry:
def CodeAssign(self): def CodeAssign(self):
code = [ 'int', code = [ 'int',
'%s_%s_assign(struct %s *msg, const %s value)' % ( '%(parent_name)s_%(name)s_assign(struct %(parent_name)s *msg,'
self._struct.Name(), self._name, ' const %(ctype)s value)',
self._struct.Name(), self._ctype),
'{', '{',
' msg->%s_set = 1;' % self._name, ' msg->%(name)s_set = 1;',
' msg->%s_data = value;' % self._name, ' msg->%(name)s_data = value;',
' return (0);', ' return (0);',
'}' ] '}' ]
return code code = '\n'.join(code)
code = code % self.GetTranslation()
return code.split('\n')
def CodeClear(self, structname): def CodeClear(self, structname):
code = [ '%s->%s_set = 0;' % (structname, self.Name()) ] code = [ '%s->%s_set = 0;' % (structname, self.Name()) ]
@ -391,16 +399,19 @@ class Entry:
return [] return []
def CodeNew(self, name): def CodeNew(self, name):
code = [ '%s->%s_assign = %s_%s_assign;' % ( code = [
name, self._name, self._struct.Name(), self._name ), '%(vname)s->%(name)s_assign = %(parent_name)s_%(name)s_assign;',
'%s->%s_get = %s_%s_get;' % ( '%(vname)s->%(name)s_get = %(parent_name)s_%(name)s_get;'
name, self._name, self._struct.Name(), self._name ),
] ]
if self.Array(): if self.Array():
code.append( code.append(
'%s->%s_add = %s_%s_add;' % ( '%(vname)s->%(name)s_add = %(parent_name)s_%(name)s_add;')
name, self._name, self._struct.Name(), self._name ) )
return code code = '\n'.join(code)
trans = self.GetTranslation()
trans['vname'] = name
code = code % trans
return code.split('\n')
def Verify(self): def Verify(self):
if self.Array() and not self._can_be_array: if self.Array() and not self._can_be_array:
@ -543,19 +554,19 @@ class EntryString(Entry):
def CodeAssign(self): def CodeAssign(self):
name = self._name name = self._name
code = [ 'int', code = """int
'%s_%s_assign(struct %s *msg, const %s value)' % ( %(parent_name)s_%(name)s_assign(struct %(parent_name)s *msg,
self._struct.Name(), name, const %(ctype)s value)
self._struct.Name(), self._ctype), {
'{', if (msg->%(name)s_data != NULL)
' if (msg->%s_data != NULL)' % name, free(msg->%(name)s_data);
' free(msg->%s_data);' % name, if ((msg->%(name)s_data = strdup(value)) == NULL)
' if ((msg->%s_data = strdup(value)) == NULL)' % name, return (-1);
' return (-1);', msg->%(name)s_set = 1;
' msg->%s_set = 1;' % name, return (0);
' return (0);', }""" % self.GetTranslation()
'}' ]
return code return code.split('\n')
def CodeUnmarshal(self, buf, tag_name, var_name): def CodeUnmarshal(self, buf, tag_name, var_name):
code = ['if (evtag_unmarshal_string(%s, %s, &%s->%s_data) == -1) {' % ( code = ['if (evtag_unmarshal_string(%s, %s, &%s->%s_data) == -1) {' % (
@ -605,22 +616,12 @@ class EntryStruct(Entry):
self._can_be_array = 1 self._can_be_array = 1
self._refname = refname self._refname = refname
self._ctype = 'struct %s' % refname self._ctype = 'struct %s*' % refname
def GetDeclaration(self, funcname):
code = [ 'int %s(struct %s *, %s **);' % (
funcname, self._struct.Name(), self._ctype ) ]
return code
def AssignDeclaration(self, funcname):
code = [ 'int %s(struct %s *, const %s *);' % (
funcname, self._struct.Name(), self._ctype ) ]
return code
def CodeGet(self): def CodeGet(self):
name = self._name name = self._name
code = [ 'int', code = [ 'int',
'%s_%s_get(struct %s *msg, %s **value)' % ( '%s_%s_get(struct %s *msg, %s *value)' % (
self._struct.Name(), name, self._struct.Name(), name,
self._struct.Name(), self._ctype), self._struct.Name(), self._ctype),
'{', '{',
@ -637,47 +638,43 @@ class EntryStruct(Entry):
def CodeAssign(self): def CodeAssign(self):
name = self._name name = self._name
code = [ 'int', code = """int
'%s_%s_assign(struct %s *msg, const %s *value)' % ( %(parent_name)s_%(name)s_assign(struct %(parent_name)s *msg,
self._struct.Name(), name, const %(ctype)s value)
self._struct.Name(), self._ctype), {
'{', struct evbuffer *tmp = NULL;
' struct evbuffer *tmp = NULL;', if (msg->%(name)s_set) {
' if (msg->%s_set) {' % name, %(refname)s_clear(msg->%(name)s_data);
' %s_clear(msg->%s_data);' % (self._refname, name), msg->%(name)s_set = 0;
' msg->%s_set = 0;' % name, } else {
' } else {', msg->%(name)s_data = %(refname)s_new();
' msg->%s_data = %s_new();' % (name, self._refname), if (msg->%(name)s_data == NULL) {
' if (msg->%s_data == NULL) {' % name, event_warn("%%s: %(refname)s_new()", __func__);
' event_warn("%%s: %s_new()", __func__);' % ( goto error;
self._refname), }
' goto error;', }
' }', if ((tmp = evbuffer_new()) == NULL) {
' }', event_warn("%%s: evbuffer_new()", __func__);
' if ((tmp = evbuffer_new()) == NULL) {', goto error;
' event_warn("%s: evbuffer_new()", __func__);', }
' goto error;', %(refname)s_marshal(tmp, value);
' }', if (%(refname)s_unmarshal(msg->%(name)s_data, tmp) == -1) {
' %s_marshal(tmp, value); ' % self._refname, event_warnx("%%s: %(refname)s_unmarshal", __func__);
' if (%s_unmarshal(msg->%s_data, tmp) == -1) {' % ( goto error;
self._refname, name ), }
' event_warnx("%%s: %s_unmarshal", __func__);' % ( msg->%(name)s_set = 1;
self._refname), evbuffer_free(tmp);
' goto error;', return (0);
' }', error:
' msg->%s_set = 1;' % name, if (tmp != NULL)
' evbuffer_free(tmp);', evbuffer_free(tmp);
' return (0);', if (msg->%(name)s_data != NULL) {
' error:', %(refname)s_free(msg->%(name)s_data);
' if (tmp != NULL)', msg->%(name)s_data = NULL;
' evbuffer_free(tmp);', }
' if (msg->%s_data != NULL) {' % name, return (-1);
' %s_free(msg->%s_data);' % (self._refname, name), }""" % self.GetTranslation()
' msg->%s_data = NULL;' % name, return code.split('\n')
' }',
' return (-1);',
'}' ]
return code
def CodeComplete(self, structname): def CodeComplete(self, structname):
if self.Optional(): if self.Optional():
@ -735,7 +732,7 @@ class EntryStruct(Entry):
return code return code
def Declaration(self): def Declaration(self):
dcl = ['struct %s *%s_data;' % (self._refname, self._name)] dcl = ['%s %s_data;' % (self._ctype, self._name)]
return dcl return dcl
@ -853,145 +850,132 @@ class EntryArray(Entry):
self._entry = entry self._entry = entry
self._refname = entry._refname self._refname = entry._refname
self._ctype = 'struct %s' % self._refname self._ctype = 'struct %s *' % self._refname
def GetDeclaration(self, funcname): def GetDeclaration(self, funcname):
"""Allows direct access to elements of the array.""" """Allows direct access to elements of the array."""
code = [ 'int %s(struct %s *, int, %s **);' % ( translate = self.GetTranslation()
funcname, self._struct.Name(), self._ctype ) ] translate["funcname"] = funcname
code = [
'int %(funcname)s(struct %(parent_name)s *, int, %(ctype)s *);' %
translate ]
return code return code
def AssignDeclaration(self, funcname): def AssignDeclaration(self, funcname):
code = [ 'int %s(struct %s *, int, const %s *);' % ( code = [ 'int %s(struct %s *, int, const %s);' % (
funcname, self._struct.Name(), self._ctype ) ] funcname, self._struct.Name(), self._ctype ) ]
return code return code
def AddDeclaration(self, funcname): def AddDeclaration(self, funcname):
code = [ '%s *%s(struct %s *);' % ( code = [ '%s %s(struct %s *);' % (
self._ctype, funcname, self._struct.Name() ) ] self._ctype, funcname, self._struct.Name() ) ]
return code return code
def CodeGet(self): def CodeGet(self):
name = self._name code = """int
code = [ 'int', %(parent_name)s_%(name)s_get(struct %(parent_name)s *msg, int offset,
'%s_%s_get(struct %s *msg, int offset, %s **value)' % ( %(ctype)s *value)
self._struct.Name(), name, {
self._struct.Name(), self._ctype), if (!msg->%(name)s_set || offset < 0 || offset >= msg->%(name)s_length)
'{', return (-1);
' if (msg->%s_set != 1)' % name, *value = msg->%(name)s_data[offset];
' return (-1);', return (0);
' if (offset >= msg->%s_length)' % name, }""" % self.GetTranslation()
' return (-1);',
' *value = msg->%s_data[offset];' % name, return code.split('\n')
' return (0);',
'}' ]
return code
def CodeAssign(self): def CodeAssign(self):
name = self._name code = """int
code = [ 'int', %(parent_name)s_%(name)s_assign(struct %(parent_name)s *msg, int off,
'%s_%s_assign(struct %s *msg, int off, const %s *value)' % ( const %(ctype)s value)
self._struct.Name(), name, {
self._struct.Name(), self._ctype), struct evbuffer *tmp = NULL;
'{', if (!msg->%(name)s_set || off < 0 || off >= msg->%(name)s_length)
' struct evbuffer *tmp = NULL;', return (-1);
' if (msg->%s_set != 1)' % name, %(refname)s_clear(msg->%(name)s_data[off]);
' return (-1);', if ((tmp = evbuffer_new()) == NULL) {
' if (off >= msg->%s_length)' % name, event_warn("%%s: evbuffer_new()", __func__);
' return (-1);', goto error;
'', }
' %s_clear(msg->%s_data[off]);' % (self._refname, name), %(refname)s_marshal(tmp, value);
' if ((tmp = evbuffer_new()) == NULL) {', if (%(refname)s_unmarshal(msg->%(name)s_data[off], tmp) == -1) {
' event_warn("%s: evbuffer_new()", __func__);', event_warnx("%%s: %(refname)s_unmarshal", __func__);
' goto error;', goto error;
' }', }
' %s_marshal(tmp, value); ' % self._refname, evbuffer_free(tmp);
' if (%s_unmarshal(msg->%s_data[off], tmp) == -1) {' % ( return (0);
self._refname, name ), error:
' event_warnx("%%s: %s_unmarshal", __func__);' % ( if (tmp != NULL)
self._refname), evbuffer_free(tmp);
' goto error;', %(refname)s_clear(msg->%(name)s_data[off]);
' }', return (-1);
' evbuffer_free(tmp);', }""" % self.GetTranslation()
' return (0);',
' error:', return code.split('\n')
' if (tmp != NULL)',
' evbuffer_free(tmp);',
' %s_clear(msg->%s_data[off]);' % (self._refname, name),
' return (-1);',
'}' ]
return code
def CodeAdd(self): def CodeAdd(self):
name = self._name code = \
code = [ """%(ctype)s
'%s *' % self._ctype, %(parent_name)s_%(name)s_add(struct %(parent_name)s *msg)
'%s_%s_add(struct %s *msg)' % ( {
self._struct.Name(), name, self._struct.Name()), if (++msg->%(name)s_length >= msg->%(name)s_num_allocated) {
'{', int tobe_allocated = msg->%(name)s_num_allocated;
' msg->%s_length++;' % name, %(ctype)s* new_data = NULL;
' if (msg->%s_length >= msg->%s_num_allocated) { ' % (name, name), tobe_allocated = !tobe_allocated ? 1 : tobe_allocated << 1;
' struct %s** new_data = NULL; ' % self._refname, new_data = (%(ctype)s*) realloc(msg->%(name)s_data,
' int tobe_allocated = msg->%s_num_allocated; ' % name, tobe_allocated * sizeof(%(ctype)s));
' if (!tobe_allocated) ', if (new_data == NULL)
' tobe_allocated = 1; ', goto error;
' else ', msg->%(name)s_data = new_data;
' tobe_allocated <<= 1; ', msg->%(name)s_num_allocated = tobe_allocated;
' new_data = (struct %s**)realloc(msg->%s_data, ' }
'tobe_allocated * sizeof(struct %s*));' % ( msg->%(name)s_data[msg->%(name)s_length - 1] = %(refname)s_new();
self._refname, name, self._refname ), if (msg->%(name)s_data[msg->%(name)s_length - 1] == NULL)
' if (new_data == NULL) {', goto error;
' --msg->%s_length;' % name, msg->%(name)s_set = 1;
' return (NULL);', return (msg->%(name)s_data[msg->%(name)s_length - 1]);
' }', error:
' msg->%s_data = new_data;' % name, --msg->%(name)s_length;
' msg->%s_num_allocated = tobe_allocated;' % name, return (NULL);
' }', }
' msg->%s_data[msg->%s_length - 1] = %s_new();' % ( """ % self.GetTranslation()
name, name, self._refname),
' if (msg->%s_data[msg->%s_length - 1] == NULL) {' % (name, name), return code.split('\n')
' --msg->%s_length; ' % name,
' return (NULL);',
' }',
' msg->%s_set = 1;' % name,
' return (msg->%s_data[msg->%s_length - 1]);' % (name, name),
'}'
]
return code
def CodeComplete(self, structname): def CodeComplete(self, structname):
code = [] code = []
if self.Optional(): translate = self.GetTranslation()
code.append( 'if (%s->%s_set)' % (structname, self.Name()))
code.extend(['{', if self.Optional():
' int i;', code.append( 'if (%(structname)s->%(name)s_set)' % translate)
' for (i = 0; i < %s->%s_length; ++i) {' % (
structname, self.Name()), translate["structname"] = structname
' if (%s_complete(%s->%s_data[i]) == -1)' % ( tmp = """{
self._refname, structname, self.Name()), int i;
' return (-1);', for (i = 0; i < %(structname)s->%(name)s_length; ++i) {
' }', if (%(refname)s_complete(%(structname)s->%(name)s_data[i]) == -1)
'}' return (-1);
]) }
}""" % translate
code.extend(tmp.split('\n'))
return code return code
def CodeUnmarshal(self, buf, tag_name, var_name): def CodeUnmarshal(self, buf, tag_name, var_name):
code = ['if (%s_%s_add(%s) == NULL)' % ( translate = self.GetTranslation()
self._struct.Name(), self._name, var_name), translate["var_name"] = var_name
' return (-1);', translate["buf"] = buf
'if (evtag_unmarshal_%s(%s, %s, ' translate["tag_name"] = tag_name
'%s->%s_data[%s->%s_length - 1]) == -1) {' % ( code = """if (%(parent_name)s_%(name)s_add(%(var_name)s) == NULL)
self._refname, buf, tag_name, var_name, self._name, return (-1);
var_name, self._name), if (evtag_unmarshal_%(refname)s(%(buf)s, %(tag_name)s,
' %s->%s_length--; ' % (var_name, self._name), %(var_name)s->%(name)s_data[%(var_name)s->%(name)s_length - 1]) == -1) {
' event_warnx("%%s: failed to unmarshal %s", __func__);' % ( --%(var_name)s->%(name)s_length;
self._name ), event_warnx("%%s: failed to unmarshal %(name)s", __func__);
' return (-1);', return (-1);
'}' }""" % translate
]
return code return code.split('\n')
def CodeMarshal(self, buf, tag_name, var_name): def CodeMarshal(self, buf, tag_name, var_name):
code = ['{', code = ['{',