diff options
| author | Luke T. Shumaker <lukeshu@lukeshu.com> | 2025-03-18 10:53:12 -0600 |
|---|---|---|
| committer | Luke T. Shumaker <lukeshu@lukeshu.com> | 2025-03-26 22:17:22 -0600 |
| commit | e7f3db0679e5d37970a06f428208c3f5b51db5d2 (patch) | |
| tree | 607b6716d89fb1615dda1bc27e582bed23043f68 /lib9p/protogen/c_unmarshal.py | |
| parent | 865bb702f828784a0225b5eae9ed8803094140d5 (diff) | |
Diffstat (limited to 'lib9p/protogen/c_unmarshal.py')
| -rw-r--r-- | lib9p/protogen/c_unmarshal.py | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/lib9p/protogen/c_unmarshal.py b/lib9p/protogen/c_unmarshal.py index 018d750..f14fb55 100644 --- a/lib9p/protogen/c_unmarshal.py +++ b/lib9p/protogen/c_unmarshal.py @@ -95,7 +95,19 @@ def gen_c_unmarshal(versions: set[str], typs: list[idl.UserType]) -> str: if child.cnt: cnt_path = path.parent().add(child.cnt) if child.typ.static_size == 1: # SPECIAL (zerocopy) - ret += f"{'\t'*indent_lvl()}UNMARSHAL_BYTES(ctx, {path.c_str('out->')[:-3]}, {cnt_path.c_str('out->')});\n" + if ( + c9util.typename(child.typ, child) + == f"struct {c9util.ident('_iovec')}" + ): # SPECIAL(zerocopy) + ret += ( + f"{'\t'*indent_lvl()}out->{child.membname}.iov = ctx->extra;\n" + ) + ret += f"{'\t'*indent_lvl()}ctx->extra += sizeof(out->{child.membname}.iov[0]);\n" + ret += f"{'\t'*indent_lvl()}out->{child.membname}.iov->iov_base = &ctx->net_bytes[ctx->net_offset];\n" + ret += f"{'\t'*indent_lvl()}out->{child.membname}.iov->iov_len = out->{child.cnt.membname};\n" + ret += f"{'\t'*indent_lvl()}out->{child.membname}.iovcnt = 1;\n" + else: + ret += f"{'\t'*indent_lvl()}UNMARSHAL_BYTES(ctx, {path.c_str('out->')[:-3]}, {cnt_path.c_str('out->')});\n" return idlutil.WalkCmd.KEEP_GOING, pop ret += f"{'\t'*indent_lvl()}{path.c_str('out->')[:-3]} = extra;\n" ret += f"{'\t'*indent_lvl()}extra += sizeof({path.c_str('out->')[:-3]}[0]) * {cnt_path.c_str('out->')};\n" |