merge in Winehq changes to reduce noise
[reactos.git] / reactos / tools / widl / server.c
index bd218f7..dca83ac 100644 (file)
@@ -30,7 +30,6 @@
 #include <ctype.h>\r
 #include <signal.h>\r
 \r
-#include "windef.h"\r
 #include "widl.h"\r
 #include "utils.h"\r
 #include "parser.h"\r
@@ -64,7 +63,8 @@ static int print_server(const char *format, ...)
 \r
 static void write_procformatstring(type_t *iface)\r
 {\r
-    func_t *cur = iface->funcs;\r
+    func_t *func = iface->funcs;\r
+    var_t *var;\r
 \r
     print_server("static const MIDL_PROC_FORMAT_STRING __MIDL_ProcFormatString =\n");\r
     print_server("{\n");\r
@@ -73,23 +73,135 @@ static void write_procformatstring(type_t *iface)
     print_server("{\n");\r
     indent++;\r
 \r
-    while (NEXT_LINK(cur)) cur = NEXT_LINK(cur);\r
-    while (cur)\r
+    while (NEXT_LINK(func)) func = NEXT_LINK(func);\r
+    while (func)\r
     {\r
-        var_t *def = cur->def;\r
+        /* emit argument data */\r
+        if (func->args)\r
+        {\r
+            var = func->args;\r
+            while (NEXT_LINK(var)) var = NEXT_LINK(var);\r
+            while (var)\r
+            {\r
+                switch(var->type->type)\r
+                {\r
+                case RPC_FC_BYTE:\r
+                    print_server("0x4e,    /* FC_IN_PARAM_BASETYPE */\n");\r
+                    print_server("0x%02x,    /* FC_BYTE */\n", var->type->type);\r
+                    break;\r
+                case RPC_FC_CHAR:\r
+                    print_server("0x4e,    /* FC_IN_PARAM_BASETYPE */\n");\r
+                    print_server("0x%02x,    /* FC_CHAR */\n", var->type->type);\r
+                    break;\r
+                case RPC_FC_WCHAR:\r
+                    print_server("0x4e,    /* FC_IN_PARAM_BASETYPE */\n");\r
+                    print_server("0x%02x,    /* FC_WCHAR */\n", var->type->type);\r
+                    break;\r
+                case RPC_FC_USHORT:\r
+                    print_server("0x4e,    /* FC_IN_PARAM_BASETYPE */\n");\r
+                    print_server("0x%02x,    /* FC_USHORT */\n", var->type->type);\r
+                    break;\r
+                case RPC_FC_SHORT:\r
+                    print_server("0x4e,    /* FC_IN_PARAM_BASETYPE */\n");\r
+                    print_server("0x%02x,    /* FC_SHORT */\n", var->type->type);\r
+                    break;\r
+                case RPC_FC_ULONG:\r
+                    print_server("0x4e,    /* FC_IN_PARAM_BASETYPE */\n");\r
+                    print_server("0x%02x,    /* FC_ULONG */\n", var->type->type);\r
+                    break;\r
+                case RPC_FC_LONG:\r
+                    print_server("0x4e,    /* FC_IN_PARAM_BASETYPE */\n");\r
+                    print_server("0x%02x,    /* FC_LONG */\n", var->type->type);\r
+                    break;\r
+                case RPC_FC_HYPER:\r
+                    print_server("0x4e,    /* FC_IN_PARAM_BASETYPE */\n");\r
+                    print_server("0x%02x,    /* FC_HYPER */\n", var->type->type);\r
+                    break;\r
+                case RPC_FC_IGNORE:\r
+                    print_server("0x4e,    /* FC_IN_PARAM_BASETYPE */\n");\r
+                    print_server("0x%02x,    /* FC_IGNORE */\n", var->type->type);\r
+                    break;\r
+                case RPC_FC_SMALL:\r
+                    print_server("0x4e,    /* FC_IN_PARAM_BASETYPE */\n");\r
+                    print_server("0x%02x,    /* FC_SMALL */\n", RPC_FC_SMALL);\r
+                    break;\r
+                case RPC_FC_FLOAT:\r
+                    print_server("0x4e,    /* FC_IN_PARAM_BASETYPE */\n");\r
+                    print_server("0x%02x,    /* FC_FLOAT */\n", RPC_FC_FLOAT);\r
+                    break;\r
+                case RPC_FC_DOUBLE:\r
+                    print_server("0x4e,    /* FC_IN_PARAM_BASETYPE */\n");\r
+                    print_server("0x%02x,    /* FC_DOUBLE */\n", RPC_FC_DOUBLE);\r
+                    break;\r
+                default:\r
+                    error("Unknown/unsupported type\n");\r
+                }\r
+\r
+                var = PREV_LINK(var);\r
+            }\r
+        }\r
 \r
-        if (is_void(def->type, NULL))\r
+        /* emit return value data */\r
+        var = func->def;\r
+        if (is_void(var->type, NULL))\r
         {\r
             print_server("0x5b,    /* FC_END */\n");\r
             print_server("0x5c,    /* FC_PAD */\n");\r
         }\r
         else\r
         {\r
-            print_server("0x53,    /* FC_RETURN_PARAM_BASETYPE */\n");\r
-            print_server("0x%02x,    /* <type> */\n", def->type->type);\r
+            switch(var->type->type)\r
+            {\r
+            case RPC_FC_BYTE:\r
+                print_server("0x53,    /* FC_RETURN_PARAM_BASETYPE */\n");\r
+                print_server("0x%02x,    /* FC_BYTE */\n", var->type->type);\r
+                break;\r
+            case RPC_FC_CHAR:\r
+                print_server("0x53,    /* FC_RETURN_PARAM_BASETYPE */\n");\r
+                print_server("0x%02x,    /* FC_CHAR */\n", var->type->type);\r
+                break;\r
+            case RPC_FC_WCHAR:\r
+                print_server("0x53,    /* FC_RETURN_PARAM_BASETYPE */\n");\r
+                print_server("0x%02x,    /* FC_WCHAR */\n", var->type->type);\r
+                break;\r
+            case RPC_FC_USHORT:\r
+                print_server("0x53,    /* FC_RETURN_PARAM_BASETYPE */\n");\r
+                print_server("0x%02x,    /* FC_USHORT */\n", var->type->type);\r
+                break;\r
+            case RPC_FC_SHORT:\r
+                print_server("0x53,    /* FC_RETURN_PARAM_BASETYPE */\n");\r
+                print_server("0x%02x,    /* FC_SHORT */\n", var->type->type);\r
+                break;\r
+            case RPC_FC_ULONG:\r
+                print_server("0x53,    /* FC_RETURN_PARAM_BASETYPE */\n");\r
+                print_server("0x%02x,    /* FC_ULONG */\n", var->type->type);\r
+                break;\r
+            case RPC_FC_LONG:\r
+                print_server("0x53,    /* FC_RETURN_PARAM_BASETYPE */\n");\r
+                print_server("0x%02x,    /* FC_LONG */\n", var->type->type);\r
+                break;\r
+            case RPC_FC_HYPER:\r
+                print_server("0x53,    /* FC_RETURN_PARAM_BASETYPE */\n");\r
+                print_server("0x%02x,    /* FC_HYPER */\n", var->type->type);\r
+                break;\r
+            case RPC_FC_SMALL:\r
+                print_server("0x53,    /* FC_RETURN_PARAM_BASETYPE */\n");\r
+                print_server("0x%02x,    /* FC_SMALL */\n", RPC_FC_SMALL);\r
+                break;\r
+            case RPC_FC_FLOAT:\r
+                print_server("0x53,    /* FC_RETURN_PARAM_BASETYPE */\n");\r
+                print_server("0x%02x,    /* FC_FLOAT */\n", RPC_FC_FLOAT);\r
+                break;\r
+            case RPC_FC_DOUBLE:\r
+                print_server("0x53,    /* FC_RETURN_PARAM_BASETYPE */\n");\r
+                print_server("0x%02x,    /* FC_DOUBLE */\n", RPC_FC_DOUBLE);\r
+                break;\r
+            default:\r
+                error("Unknown/unsupported type\n");\r
+            }\r
         }\r
 \r
-        cur = PREV_LINK(cur);\r
+        func = PREV_LINK(func);\r
     }\r
 \r
     print_server("0x0\n");\r
@@ -119,16 +231,143 @@ static void write_typeformatstring(void)
 }\r
 \r
 \r
+unsigned int get_required_buffer_size(type_t *type)\r
+{\r
+    switch(type->type)\r
+    {\r
+        case RPC_FC_BYTE:\r
+        case RPC_FC_SMALL:\r
+        case RPC_FC_CHAR:\r
+        case RPC_FC_WCHAR:\r
+        case RPC_FC_USHORT:\r
+        case RPC_FC_SHORT:\r
+        case RPC_FC_ULONG:\r
+        case RPC_FC_LONG:\r
+        case RPC_FC_FLOAT:\r
+            return 4;\r
+\r
+        case RPC_FC_HYPER:\r
+        case RPC_FC_DOUBLE:\r
+            return 8;\r
+\r
+        case RPC_FC_IGNORE:\r
+            return 0;\r
+\r
+        default:\r
+            error("Unknown/unsupported type: %s\n", type->name);\r
+            return 0;\r
+    }\r
+}\r
+\r
+\r
+static void unmarshall_arguments(func_t *func)\r
+{\r
+    unsigned int alignment;\r
+    unsigned int size;\r
+    unsigned int last_size = 0;\r
+    var_t *var;\r
+\r
+    if (!func->args)\r
+        return;\r
+\r
+    var = func->args;\r
+    while (NEXT_LINK(var)) var = NEXT_LINK(var);\r
+    while (var)\r
+    {\r
+        alignment = 0;\r
+        switch (var->type->type)\r
+        {\r
+        case RPC_FC_BYTE:\r
+        case RPC_FC_CHAR:\r
+        case RPC_FC_SMALL:\r
+            size = 1;\r
+            alignment = 0;\r
+            break;\r
+\r
+        case RPC_FC_WCHAR:\r
+        case RPC_FC_USHORT:\r
+        case RPC_FC_SHORT:\r
+            size = 2;\r
+            if (last_size != 0 && last_size < 2)\r
+                alignment = (2 - last_size);\r
+            break;\r
+\r
+        case RPC_FC_ULONG:\r
+        case RPC_FC_LONG:\r
+        case RPC_FC_FLOAT:\r
+            size = 4;\r
+            if (last_size != 0 && last_size < 4)\r
+                alignment = (4 - last_size);\r
+            break;\r
+\r
+        case RPC_FC_HYPER:\r
+        case RPC_FC_DOUBLE:\r
+            size = 8;\r
+            if (last_size != 0 && last_size < 4)\r
+                alignment = (4 - last_size);\r
+            break;\r
+\r
+        case RPC_FC_IGNORE:\r
+            size = 0;\r
+            break;\r
+\r
+        default:\r
+            error("Unknown/unsupported type!");\r
+        }\r
+\r
+        if (size != 0)\r
+        {\r
+            if (alignment != 0)\r
+                print_server("_StubMsg.Buffer += %u;\n", alignment);\r
+\r
+            print_server("");\r
+            write_name(server, var);\r
+            fprintf(server, " = *((");\r
+            write_type(server, var->type, var, var->tname);\r
+            fprintf(server, " __RPC_FAR*)_StubMsg.Buffer)++;\n");\r
+            fprintf(server, "\n");\r
+\r
+            last_size = size;\r
+        }\r
+\r
+        var = PREV_LINK(var);\r
+    }\r
+}\r
+\r
+\r
 static void write_function_stubs(type_t *iface)\r
 {\r
-    func_t *cur = iface->funcs;\r
-    while (NEXT_LINK(cur)) cur = NEXT_LINK(cur);\r
-    while (cur)\r
+    int explicit_handle = is_attr(iface->attrs, ATTR_EXPLICIT_HANDLE);\r
+    func_t *func = iface->funcs;\r
+    var_t *var;\r
+    var_t* explicit_handle_var;\r
+    unsigned int proc_offset = 0;\r
+\r
+    while (NEXT_LINK(func)) func = NEXT_LINK(func);\r
+    while (func)\r
     {\r
-        var_t *def = cur->def;\r
+        var_t *def = func->def;\r
 \r
-        write_type(server, def->type, def, def->tname);\r
-        fprintf(server, " __RPC_STUB\n");\r
+        /* check for a defined binding handle */\r
+        explicit_handle_var = get_explicit_handle_var(func);\r
+        if (explicit_handle)\r
+        {\r
+            if (!explicit_handle_var)\r
+            {\r
+                error("%s() does not define an explicit binding handle!\n", def->name);\r
+                return;\r
+            }\r
+        }\r
+        else\r
+        {\r
+            if (explicit_handle_var)\r
+            {\r
+                error("%s() must not define a binding handle!\n", def->name);\r
+                return;\r
+            }\r
+        }\r
+\r
+        fprintf(server, "void __RPC_STUB\n");\r
         fprintf(server, "%s_", iface->name);\r
         write_name(server, def);\r
         fprintf(server, "(\n");\r
@@ -148,9 +387,28 @@ static void write_function_stubs(type_t *iface)
             fprintf(server, " _RetVal;\n");\r
         }\r
 \r
+        /* declare arguments */\r
+        if (func->args)\r
+        {\r
+            var = func->args;\r
+            while (NEXT_LINK(var)) var = NEXT_LINK(var);\r
+            while (var)\r
+            {\r
+                print_server("");\r
+                write_type(server, var->type, var, var->tname);\r
+                fprintf(server, " ");\r
+                write_name(server, var);\r
+                fprintf(server, ";\n");\r
+\r
+                var = PREV_LINK(var);\r
+            }\r
+        }\r
+\r
         print_server("MIDL_STUB_MESSAGE _StubMsg;\n");\r
         print_server("RPC_STATUS _Status;\n");\r
         fprintf(server, "\n");\r
+\r
+\r
         print_server("((void)(_Status));\n");\r
         print_server("NdrServerInitializeNew(\n");\r
         indent++;\r
@@ -160,12 +418,33 @@ static void write_function_stubs(type_t *iface)
         indent--;\r
         fprintf(server, "\n");\r
 \r
+        if (explicit_handle)\r
+        {\r
+            print_server("%s = _pRpcMessage->Handle;\n", explicit_handle_var->name);\r
+            fprintf(server, "\n");\r
+        }\r
+\r
         print_server("RpcTryFinally\n");\r
         print_server("{\n");\r
         indent++;\r
         print_server("RpcTryExcept\n");\r
         print_server("{\n");\r
         indent++;\r
+\r
+        if (func->args)\r
+        {\r
+            print_server("if ((_pRpcMessage->DataRepresentation & 0x0000FFFFUL) != NDR_LOCAL_DATA_REPRESENTATION)\n");\r
+            indent++;\r
+            print_server("NdrConvert(\n");\r
+            indent++;\r
+            print_server("(PMIDL_STUB_MESSAGE)&_StubMsg,\n");\r
+            print_server("(PFORMAT_STRING)&__MIDL_ProcFormatString.Format[%u]);\n", proc_offset);\r
+            indent -= 2;\r
+            fprintf(server, "\n");\r
+\r
+            unmarshall_arguments(func);\r
+        }\r
+\r
         print_server("if (_StubMsg.Buffer > _StubMsg.BufferEnd)\n");\r
         print_server("{\n");\r
         indent++;\r
@@ -191,14 +470,37 @@ static void write_function_stubs(type_t *iface)
             print_server("");\r
         write_name(server, def);\r
 \r
-        /* FIXME: handle argument list */\r
-        fprintf(server, "();\n");\r
+        if (func->args)\r
+        {\r
+            int first_arg = 1;\r
 \r
-        /* FIXME: Marshall the return value */\r
+            fprintf(server, "(\n");\r
+            indent++;\r
+            var = func->args;\r
+            while (NEXT_LINK(var)) var = NEXT_LINK(var);\r
+            while (var)\r
+            {\r
+                if (first_arg)\r
+                    first_arg = 0;\r
+                else\r
+                    fprintf(server, ",\n");\r
+                print_server("");\r
+                write_name(server, var);\r
+                var = PREV_LINK(var);\r
+            }\r
+            fprintf(server, ");\n");\r
+            indent--;\r
+        }\r
+        else\r
+        {\r
+            fprintf(server, "();\n");\r
+        }\r
+\r
+        /* marshall the return value */\r
         if (!is_void(def->type, NULL))\r
         {\r
             fprintf(server, "\n");\r
-            print_server("_StubMsg.BufferLength = %uU;\n", 4); /* FIXME */\r
+            print_server("_StubMsg.BufferLength = %uU;\n", get_required_buffer_size(def->type));\r
             print_server("_pRpcMessage->BufferLength = _StubMsg.BufferLength;\n");\r
             fprintf(server, "\n");\r
             print_server("_Status = I_RpcGetBuffer(_pRpcMessage);\n");\r
@@ -232,7 +534,20 @@ static void write_function_stubs(type_t *iface)
         fprintf(server, "}\n");\r
         fprintf(server, "\n");\r
 \r
-        cur = PREV_LINK(cur);\r
+        /* update proc_offset */\r
+        if (func->args)\r
+        {\r
+            var = func->args;\r
+            while (NEXT_LINK(var)) var = NEXT_LINK(var);\r
+            while (var)\r
+            {\r
+                proc_offset += 2; /* FIXME */\r
+                var = PREV_LINK(var);\r
+            }\r
+        }\r
+        proc_offset += 2;  /* FIXME */\r
+\r
+        func = PREV_LINK(func);\r
     }\r
 }\r
 \r
@@ -287,7 +602,7 @@ static void write_stubdescriptor(type_t *iface)
     print_server("(void __RPC_FAR *)& %s___RpcServerInterface,\n", iface->name);\r
     print_server("MIDL_user_allocate,\n");\r
     print_server("MIDL_user_free,\n");\r
-    print_server("0,\n");\r
+    print_server("{NULL},\n");\r
     print_server("0,\n");\r
     print_server("0,\n");\r
     print_server("0,\n");\r
@@ -354,18 +669,32 @@ static void write_formatdesc( const char *str )
 \r
 static void write_formatstringsdecl(type_t *iface)\r
 {\r
-    func_t *cur;\r
+    func_t *func;\r
+    var_t *var;\r
     int byte_count = 1;\r
 \r
     print_server("#define TYPE_FORMAT_STRING_SIZE %d\n", 3); /* FIXME */\r
 \r
     /* determine the proc format string size */\r
-    cur = iface->funcs;\r
-    while (NEXT_LINK(cur)) cur = NEXT_LINK(cur);\r
-    while (cur)\r
+    func = iface->funcs;\r
+    while (NEXT_LINK(func)) func = NEXT_LINK(func);\r
+    while (func)\r
     {\r
+        /* argument list size */\r
+        if (func->args)\r
+        {\r
+            var = func->args;\r
+            while (NEXT_LINK(var)) var = NEXT_LINK(var);\r
+            while (var)\r
+            {\r
+                byte_count += 2; /* FIXME: determine real size */\r
+                var = PREV_LINK(var);\r
+            }\r
+        }\r
+\r
+        /* return value size */\r
         byte_count += 2; /* FIXME: determine real size */\r
-        cur = PREV_LINK(cur);\r
+        func = PREV_LINK(func);\r
     }\r
     print_server("#define PROC_FORMAT_STRING_SIZE %d\n", byte_count);\r
 \r
@@ -396,38 +725,46 @@ static void init_server(void)
 \r
 void write_server(ifref_t *ifaces)\r
 {\r
-    ifref_t *lcur = ifaces;\r
-    char *file_id = server_token;\r
-    int c;\r
+    ifref_t *iface = ifaces;\r
 \r
     if (!do_server)\r
         return;\r
-    if (!lcur)\r
+    if (!iface)\r
         return;\r
-    END_OF_LIST(lcur);\r
+    END_OF_LIST(iface);\r
 \r
     init_server();\r
     if (!server)\r
         return;\r
 \r
-    write_formatstringsdecl(lcur->iface);\r
-    write_serverinterfacedecl(lcur->iface);\r
-    write_stubdescdecl(lcur->iface);\r
+    while (iface)\r
+    {\r
+        fprintf(server, "/*****************************************************************************\n");\r
+        fprintf(server, " * %s interface\n", iface->iface->name);\r
+        fprintf(server, " */\n");\r
+        fprintf(server, "\n");\r
 \r
-    write_function_stubs(lcur->iface);\r
+        write_formatstringsdecl(iface->iface);\r
+        write_serverinterfacedecl(iface->iface);\r
+        write_stubdescdecl(iface->iface);\r
 \r
-    write_stubdescriptor(lcur->iface);\r
-    write_dispatchtable(lcur->iface);\r
+        write_function_stubs(iface->iface);\r
 \r
-    print_server("#if !defined(__RPC_WIN32__)\n");\r
-    print_server("#error  Invalid build platform for this stub.\n");\r
-    print_server("#endif\n");\r
-    fprintf(server, "\n");\r
+        write_stubdescriptor(iface->iface);\r
+        write_dispatchtable(iface->iface);\r
+\r
+        print_server("#if !defined(__RPC_WIN32__)\n");\r
+        print_server("#error  Invalid build platform for this stub.\n");\r
+        print_server("#endif\n");\r
+        fprintf(server, "\n");\r
 \r
-    write_procformatstring(lcur->iface);\r
-    write_typeformatstring();\r
+        write_procformatstring(iface->iface);\r
+        write_typeformatstring();\r
 \r
-    fprintf(server, "\n");\r
+        fprintf(server, "\n");\r
+\r
+        iface = PREV_LINK(iface);\r
+    }\r
 \r
     fclose(server);\r
 }\r