Function handle_method_cpp

lsteamclient/gen_wrapper.py:1032–1157

(method, classname, out, wow64)

def handle_method_cpp(method, classname, out, wow64):
    returns_void = method.result_type.kind == TypeKind.VOID
    returns_record = method.result_type.get_canonical().kind == TypeKind.RECORD
    prefix = "wow64_" if wow64 else ""

    names = [p.spelling if p.spelling != "" else f'_{chr(0x61 + i)}'
             for i, p in enumerate(method.get_arguments())]
    path_conv_wtou = PATH_CONV_METHODS_WTOU.get(f'{klass.name}_{method.spelling}', {})
    outstr_param = OUTSTR_PARAMS.get(method.name, None)

    need_convert = {n: p for n, p in zip(names, method.get_arguments())
                    if param_needs_conversion(p, wow64) and n != outstr_param
                    and n not in path_conv_wtou}
    manual_convert = {n: p for n, p in zip(names, method.get_arguments())
                      if underlying_type(p).spelling in MANUAL_TYPES
                      or p.spelling in MANUAL_PARAMS}

    names = ['u_iface'] + names

    out(f'NTSTATUS {prefix}{method.full_name}( void *args )\n')
    out(u'{\n')
    out(f'    struct {prefix}{method.full_name}_params *params = (struct {prefix}{method.full_name}_params *)args;\n')
    out(f'    struct u_{klass.full_name} *iface = (struct u_{klass.full_name} *)params->u_iface;\n')
    if method.name in OUTSTR_PARAMS and OUTSTR_PARAMS[method.name] in names:
        out(u'    char *u_str;\n')

    params = list(zip(names[1:], method.get_arguments()))
    for i, (name, param) in enumerate(params[:-1]):
        if underlying_type(param).kind != TypeKind.RECORD:
            continue
        next_name, next_param = params[i + 1]
        if not any(w in next_name.lower() for w in ('count', 'len', 'size', 'num')):
            continue
        assert underlying_typename(param) in SIZED_STRUCTS | EXEMPT_STRUCTS

    for i, (name, param) in enumerate(params[1:]):
        if underlying_type(param).kind != TypeKind.RECORD:
            continue
        prev_name, prev_param = params[i - 1]
        if not any(w in prev_name.lower() for w in ('count', 'len', 'size', 'num')):
            continue
        if underlying_typename(param) not in SIZED_STRUCTS | EXEMPT_STRUCTS:
            print('Warning:', underlying_typename(param), name, 'following', prev_name)

    for name, conv in filter(lambda x: x[0] in names, path_conv_wtou.items()):
        if conv['array']:
            out(f'    const char **u_{name} = {prefix}steamclient_dos_to_unix_path_array( params->{name} );\n')
        else:
            out(f'    char *u_{name} = steamclient_dos_to_unix_path( params->{name}, {int(conv["url"])} );\n')

    wow64_convert = {}
    need_output = {}

    for name, param in sorted(need_convert.items()):
        if param.type.kind != TypeKind.POINTER:
            out(f'    {declspec(param.type, f"u_{name}", "u_")} = params->{name};\n')
            continue