aljaz/lightning
1
0
Fork 0
mirror of https://github.com/aljazceru/lightning.git synced 2026-02-23 06:54:30 +01:00
Code Issues Packages Projects Releases Wiki Activity

tools/generate-wire.py: handle marshal/unmarshal of enum types.

Browse Source 
Signed-off-by: Rusty Russell <rusty@rustcorp.com.au>
This commit is contained in:
Rusty Russell
2017-06-20 15:26:03 +09:30
parent 060a440548
commit 9868e896d9
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tools/generate-wire.py
View File

@@ -47,8 +47,8 @@ class FieldType(object):
def _typesize(typename):
if typename in type2size:
return type2size[typename]
elif typename.startswith('struct '):
# We allow unknown structures, for extensiblity (can only happen
elif typename.startswith('struct ') or typename.startswith('enum '):
# We allow unknown structures/enums, for extensiblity (can only happen
# if explicitly specified in csv)
return 0
else:
@@ -275,6 +275,8 @@ class Message(object):
basetype=f.fieldtype.name
if f.fieldtype.name.startswith('struct '):
basetype=f.fieldtype.name[7:]
elif f.fieldtype.name.startswith('enum '):
basetype=f.fieldtype.name[5:]
for c in f.comments:
subcalls.append('\t/*{} */'.format(c))
@@ -353,6 +355,8 @@ class Message(object):
basetype=f.fieldtype.name
if basetype.startswith('struct '):
basetype=basetype[7:]
elif basetype.startswith('enum '):
basetype=basetype[5:]
for c in f.comments:
subcalls.append('\t/*{} */'.format(c))