OLD | NEW |
| (Empty) |
1 /* | |
2 * Copyright 2008 The Native Client Authors. All rights reserved. | |
3 * Use of this source code is governed by a BSD-style license that can | |
4 * be found in the LICENSE file. | |
5 */ | |
6 | |
7 // Lookup table types for method dispatching. | |
8 | |
9 #ifndef NATIVE_CLIENT_SRC_TRUSTED_PLUGIN_SRPC_METHOD_MAP_H | |
10 #define NATIVE_CLIENT_SRC_TRUSTED_PLUGIN_SRPC_METHOD_MAP_H | |
11 | |
12 #include "native_client/src/include/nacl_macros.h" | |
13 #include "native_client/src/include/portability_string.h" | |
14 #include "native_client/src/shared/srpc/nacl_srpc.h" | |
15 | |
16 #include <limits.h> | |
17 #include <map> | |
18 | |
19 namespace plugin { | |
20 | |
21 class Plugin; | |
22 | |
23 bool InitSrpcArgArray(NaClSrpcArg* arr, int size); | |
24 void FreeSrpcArg(NaClSrpcArg* arg); | |
25 | |
26 // A utility class that builds and deletes parameter vectors used in rpcs. | |
27 class SrpcParams { | |
28 public: | |
29 SrpcParams() : exception_string_(NULL) { | |
30 memset(ins_, 0, sizeof(ins_)); | |
31 memset(outs_, 0, sizeof(outs_)); | |
32 } | |
33 | |
34 SrpcParams(const char* in_types, const char* out_types) | |
35 : exception_string_(NULL) { | |
36 if (!Init(in_types, out_types)) { | |
37 FreeAll(); | |
38 } | |
39 } | |
40 | |
41 ~SrpcParams() { | |
42 FreeAll(); | |
43 free(exception_string_); | |
44 } | |
45 | |
46 bool Init(const char* in_types, const char* out_types); | |
47 uint32_t SignatureLength() const; | |
48 uint32_t OutputLength() const; | |
49 | |
50 NaClSrpcArg** ins() const { return const_cast<NaClSrpcArg**>(ins_); } | |
51 NaClSrpcArg** outs() const { return const_cast<NaClSrpcArg**>(outs_); } | |
52 | |
53 char* exception_string() const { return exception_string_; } | |
54 void set_exception_string(const char* msg) { | |
55 exception_string_ = STRDUP(msg); | |
56 } | |
57 | |
58 private: | |
59 NACL_DISALLOW_COPY_AND_ASSIGN(SrpcParams); | |
60 void FreeAll(); | |
61 bool FillVec(NaClSrpcArg* vec[], const char* types); | |
62 void FreeArguments(NaClSrpcArg* vec[]); | |
63 // The ins_ and outs_ arrays contain one more element, to hold a NULL pointer | |
64 // to indicate the end of the list. | |
65 NaClSrpcArg* ins_[NACL_SRPC_MAX_ARGS + 1]; | |
66 NaClSrpcArg* outs_[NACL_SRPC_MAX_ARGS + 1]; | |
67 char* exception_string_; | |
68 }; | |
69 | |
70 typedef bool (*RpcFunction)(void* obj, SrpcParams* params); | |
71 | |
72 // MethodInfo records the method names and type signatures of an SRPC server. | |
73 class MethodInfo { | |
74 public: | |
75 // statically defined method - called through a pointer | |
76 MethodInfo(const RpcFunction function_ptr, | |
77 const char* name, | |
78 const char* ins, | |
79 const char* outs, | |
80 // index is set to UINT_MAX for methods implemented by the plugin, | |
81 // All methods implemented by nacl modules have indexes | |
82 // that are lower than UINT_MAX. | |
83 const uint32_t index = UINT_MAX) : | |
84 function_ptr_(function_ptr), | |
85 name_(STRDUP(name)), | |
86 ins_(STRDUP(ins)), | |
87 outs_(STRDUP(outs)), | |
88 index_(index) { } | |
89 ~MethodInfo(); | |
90 | |
91 RpcFunction function_ptr() const { return function_ptr_; } | |
92 char* name() const { return name_; } | |
93 char* ins() const { return ins_; } | |
94 char* outs() const { return outs_; } | |
95 uint32_t index() const { return index_; } | |
96 | |
97 private: | |
98 NACL_DISALLOW_COPY_AND_ASSIGN(MethodInfo); | |
99 RpcFunction function_ptr_; | |
100 char* name_; | |
101 char* ins_; | |
102 char* outs_; | |
103 uint32_t index_; | |
104 }; | |
105 | |
106 class MethodMap { | |
107 public: | |
108 MethodMap() {} | |
109 ~MethodMap(); | |
110 MethodInfo* GetMethod(uintptr_t method_id); | |
111 void AddMethod(uintptr_t method_id, MethodInfo* info); | |
112 | |
113 private: | |
114 NACL_DISALLOW_COPY_AND_ASSIGN(MethodMap); | |
115 typedef std::map<uintptr_t, MethodInfo*> MethodMapStorage; | |
116 MethodMapStorage method_map_; | |
117 }; | |
118 | |
119 } // namespace plugin | |
120 | |
121 #endif // NATIVE_CLIENT_SRC_TRUSTED_PLUGIN_SRPC_METHOD_MAP_H | |
OLD | NEW |