1 /*
2 * Copyright (c) 2021 Huawei Device Co., Ltd.
3 *
4 * HDF is dual licensed: you can use it either under the terms of
5 * the GPL, or the BSD license, at your option.
6 * See the LICENSE file in the root of this repository for complete details.
7 */
8
9 #include "codegen/cpp_interface_code_emitter.h"
10 #include "util/file.h"
11 #include "util/logger.h"
12
13 namespace OHOS {
14 namespace HDI {
ResolveDirectory(const std::string & targetDirectory)15 bool CppInterfaceCodeEmitter::ResolveDirectory(const std::string &targetDirectory)
16 {
17 if (ast_->GetASTFileType() == ASTFileType::AST_IFACE || ast_->GetASTFileType() == ASTFileType::AST_ICALLBACK) {
18 directory_ = GetFileParentPath(targetDirectory);
19 } else {
20 return false;
21 }
22
23 if (!File::CreateParentDir(directory_)) {
24 Logger::E("CppInterfaceCodeEmitter", "Create '%s' failed!", directory_.c_str());
25 return false;
26 }
27
28 return true;
29 }
30
EmitCode()31 void CppInterfaceCodeEmitter::EmitCode()
32 {
33 switch (mode_) {
34 case GenMode::PASSTHROUGH:
35 case GenMode::IPC: {
36 EmitInterfaceHeaderFile();
37 break;
38 }
39 default:
40 break;
41 }
42 }
43
EmitInterfaceHeaderFile()44 void CppInterfaceCodeEmitter::EmitInterfaceHeaderFile()
45 {
46 std::string filePath =
47 File::AdapterPath(StringHelper::Format("%s/%s.h", directory_.c_str(), FileName(interfaceName_).c_str()));
48 File file(filePath, File::WRITE);
49 StringBuilder sb;
50
51 EmitLicense(sb);
52 EmitHeadMacro(sb, interfaceFullName_);
53 sb.Append("\n");
54 EmitInterfaceInclusions(sb);
55 if (!Options::GetInstance().DoPassthrough()) {
56 sb.Append("\n");
57 EmitInterfaceBuffSizeMacro(sb);
58 }
59 sb.Append("\n");
60 EmitBeginNamespace(sb);
61 EmitUsingNamespace(sb);
62 if (!Options::GetInstance().DoPassthrough()) {
63 sb.Append("\n");
64 if (interface_->GetExtendsInterface() == nullptr) {
65 EmitInterfaceMethodCommands(sb, "");
66 } else {
67 EmitInterfaceMethodCommandsWithExtends(sb, "");
68 }
69 }
70 sb.Append("\n");
71 EmitInterfaceDefinition(sb);
72 EmitEndNamespace(sb);
73 sb.Append("\n");
74 EmitTailMacro(sb, interfaceFullName_);
75
76 std::string data = sb.ToString();
77 file.WriteData(data.c_str(), data.size());
78 file.Flush();
79 file.Close();
80 }
81
EmitInterfaceInclusions(StringBuilder & sb)82 void CppInterfaceCodeEmitter::EmitInterfaceInclusions(StringBuilder &sb)
83 {
84 HeaderFile::HeaderFileSet headerFiles;
85
86 GetStdlibInclusions(headerFiles);
87 GetImportInclusions(headerFiles);
88 GetHeaderOtherLibInclusions(headerFiles);
89
90 for (const auto &file : headerFiles) {
91 sb.AppendFormat("%s\n", file.ToString().c_str());
92 }
93 }
94
GetHeaderOtherLibInclusions(HeaderFile::HeaderFileSet & headerFiles) const95 void CppInterfaceCodeEmitter::GetHeaderOtherLibInclusions(HeaderFile::HeaderFileSet &headerFiles) const
96 {
97 headerFiles.emplace(HeaderFileType::OTHER_MODULES_HEADER_FILE, "hdf_base");
98 headerFiles.emplace(HeaderFileType::C_STD_HEADER_FILE, "stdint");
99 headerFiles.emplace(HeaderFileType::OTHER_MODULES_HEADER_FILE, "hdi_base");
100 headerFiles.emplace(HeaderFileType::OTHER_MODULES_HEADER_FILE, "hdf_base");
101 }
102
EmitInterfaceVersionMacro(StringBuilder & sb) const103 void CppInterfaceCodeEmitter::EmitInterfaceVersionMacro(StringBuilder &sb) const
104 {
105 sb.AppendFormat("#define %s %u\n", majorVerName_.c_str(), ast_->GetMajorVer());
106 sb.AppendFormat("#define %s %u\n", minorVerName_.c_str(), ast_->GetMinorVer());
107 }
108
EmitInterfaceDefinition(StringBuilder & sb)109 void CppInterfaceCodeEmitter::EmitInterfaceDefinition(StringBuilder &sb)
110 {
111 AutoPtr<ASTInterfaceType> interface = interface_->GetExtendsInterface();
112 if (interface != nullptr) {
113 sb.AppendFormat("class %s : public %s {\n", interfaceName_.c_str(),
114 EmitDefinitionByInterface(interface, interfaceName_).c_str());
115 } else {
116 sb.AppendFormat("class %s : public HdiBase {\n", interfaceName_.c_str());
117 }
118 sb.Append("public:\n");
119 EmitInterfaceDescriptor(sb, TAB);
120 sb.Append("\n");
121 EmitInterfaceDestruction(sb, TAB);
122 sb.Append("\n");
123 if (!interface_->IsSerializable()) {
124 EmitGetMethodDecl(sb, TAB);
125 sb.Append("\n");
126 }
127 if (interface_->GetExtendsInterface() != nullptr) {
128 EmitCastFromDecl(sb, TAB);
129 sb.Append("\n");
130 }
131 EmitInterfaceMethodsDecl(sb, TAB);
132 sb.Append("\n");
133 EmitGetDescMethod(sb, TAB);
134 sb.Append("};\n");
135 }
136
EmitGetDescMethod(StringBuilder & sb,const std::string & prefix) const137 void CppInterfaceCodeEmitter::EmitGetDescMethod(StringBuilder &sb, const std::string &prefix) const
138 {
139 AutoPtr<ASTInterfaceType> interface = interface_->GetExtendsInterface();
140 if (interface == nullptr) {
141 sb.Append(prefix).Append("virtual const std::u16string GetDesc()");
142 } else {
143 sb.Append(prefix).Append("const std::u16string GetDesc() override");
144 }
145 sb.Append("\n");
146 sb.Append(prefix).Append("{").Append("\n");
147 sb.Append(prefix + TAB).Append("return metaDescriptor_;\n");
148 sb.Append(prefix).Append("}\n");
149 }
150
EmitInterfaceDescriptor(StringBuilder & sb,const std::string & prefix) const151 void CppInterfaceCodeEmitter::EmitInterfaceDescriptor(StringBuilder &sb, const std::string &prefix) const
152 {
153 (void)prefix;
154 sb.Append(TAB).AppendFormat("DECLARE_HDI_DESCRIPTOR(u\"%s\");\n", interfaceFullName_.c_str());
155 }
156
EmitCastFromDecl(StringBuilder & sb,const std::string & prefix) const157 void CppInterfaceCodeEmitter::EmitCastFromDecl(StringBuilder &sb, const std::string &prefix) const
158 {
159 std::string currentInterface = EmitDefinitionByInterface(interface_, interfaceName_);
160
161 AutoPtr<ASTInterfaceType> interface = interface_->GetExtendsInterface();
162 while (interface != nullptr) {
163 std::string parentInterface = EmitDefinitionByInterface(interface, interfaceName_);
164 sb.Append(prefix).AppendFormat("static sptr<%s> CastFrom(const sptr<%s> &parent);\n",
165 currentInterface.c_str(), parentInterface.c_str());
166 interface = interface->GetExtendsInterface();
167 }
168 }
169
EmitGetMethodDecl(StringBuilder & sb,const std::string & prefix) const170 void CppInterfaceCodeEmitter::EmitGetMethodDecl(StringBuilder &sb, const std::string &prefix) const
171 {
172 sb.Append(prefix).AppendFormat("static %s Get(bool isStub = false);\n", interface_->EmitCppType().c_str());
173 sb.Append(prefix).AppendFormat(
174 "static %s Get(const std::string &serviceName, bool isStub = false);\n", interface_->EmitCppType().c_str());
175 }
176
EmitInterfaceDestruction(StringBuilder & sb,const std::string & prefix) const177 void CppInterfaceCodeEmitter::EmitInterfaceDestruction(StringBuilder &sb, const std::string &prefix) const
178 {
179 sb.Append(prefix).AppendFormat("virtual ~%s() = default;\n", interface_->GetName().c_str());
180 }
181
EmitInterfaceMethodsDecl(StringBuilder & sb,const std::string & prefix)182 void CppInterfaceCodeEmitter::EmitInterfaceMethodsDecl(StringBuilder &sb, const std::string &prefix)
183 {
184 for (const auto &method : interface_->GetMethodsBySystem(Options::GetInstance().GetSystemLevel())) {
185 EmitInterfaceMethodDecl(method, sb, prefix);
186 sb.Append("\n");
187 }
188
189 EmitInterfaceGetVersionMethod(sb, prefix);
190 if (interface_->GetExtendsInterface() == nullptr) {
191 sb.Append("\n");
192 EmitInterfaceIsProxyMethod(sb, prefix);
193 }
194 }
195
EmitInterfaceMethodDecl(const AutoPtr<ASTMethod> & method,StringBuilder & sb,const std::string & prefix) const196 void CppInterfaceCodeEmitter::EmitInterfaceMethodDecl(
197 const AutoPtr<ASTMethod> &method, StringBuilder &sb, const std::string &prefix) const
198 {
199 if (method->GetParameterNumber() == 0) {
200 sb.Append(prefix).AppendFormat("virtual int32_t %s() = 0;\n", method->GetName().c_str());
201 } else {
202 StringBuilder paramStr;
203 paramStr.Append(prefix).AppendFormat("virtual int32_t %s(", method->GetName().c_str());
204 for (size_t i = 0; i < method->GetParameterNumber(); i++) {
205 AutoPtr<ASTParameter> param = method->GetParameter(i);
206 EmitInterfaceMethodParameter(param, paramStr, "");
207 if (i + 1 < method->GetParameterNumber()) {
208 paramStr.Append(", ");
209 }
210 }
211
212 paramStr.Append(") = 0;");
213 sb.Append(SpecificationParam(paramStr, prefix + TAB));
214 sb.Append("\n");
215 }
216 }
217
EmitInterfaceGetVersionMethod(StringBuilder & sb,const std::string & prefix) const218 void CppInterfaceCodeEmitter::EmitInterfaceGetVersionMethod(StringBuilder &sb, const std::string &prefix) const
219 {
220 AutoPtr<ASTMethod> method = interface_->GetVersionMethod();
221 if (interface_->GetExtendsInterface() == nullptr) {
222 sb.Append(prefix).AppendFormat("virtual int32_t %s(", method->GetName().c_str());
223 } else {
224 sb.Append(prefix).AppendFormat("int32_t %s(", method->GetName().c_str());
225 }
226 for (size_t i = 0; i < method->GetParameterNumber(); i++) {
227 AutoPtr<ASTParameter> param = method->GetParameter(i);
228 EmitInterfaceMethodParameter(param, sb, "");
229 if (i + 1 < method->GetParameterNumber()) {
230 sb.Append(", ");
231 }
232 }
233 sb.Append(")");
234 if (interface_->GetExtendsInterface() != nullptr) {
235 sb.Append(" override");
236 }
237 sb.Append("\n");
238 sb.Append(prefix).Append("{\n");
239
240 AutoPtr<ASTParameter> majorParam = method->GetParameter(0);
241 sb.Append(prefix + TAB).AppendFormat("%s = %d;\n", majorParam->GetName().c_str(), ast_->GetMajorVer());
242 AutoPtr<ASTParameter> minorParam = method->GetParameter(1);
243 sb.Append(prefix + TAB).AppendFormat("%s = %d;\n", minorParam->GetName().c_str(), ast_->GetMinorVer());
244
245 sb.Append(prefix + TAB).Append("return HDF_SUCCESS;\n");
246 sb.Append(prefix).Append("}\n");
247 }
248
EmitInterfaceIsProxyMethod(StringBuilder & sb,const std::string & prefix) const249 void CppInterfaceCodeEmitter::EmitInterfaceIsProxyMethod(StringBuilder &sb, const std::string &prefix) const
250 {
251 sb.Append(prefix).AppendFormat("virtual bool %s(", "IsProxy");
252 sb.Append(")\n");
253 sb.Append(prefix).Append("{\n");
254 sb.Append(prefix + TAB).Append("return false;\n");
255 sb.Append(prefix).Append("}\n");
256 }
257
EmitInterfaceMethodParameter(const AutoPtr<ASTParameter> & param,StringBuilder & sb,const std::string & prefix) const258 void CppInterfaceCodeEmitter::EmitInterfaceMethodParameter(
259 const AutoPtr<ASTParameter> ¶m, StringBuilder &sb, const std::string &prefix) const
260 {
261 sb.Append(prefix).Append(param->EmitCppParameter());
262 }
263
EmitInterfaceMethodCommandsWithExtends(StringBuilder & sb,const std::string & prefix)264 void CppInterfaceCodeEmitter::EmitInterfaceMethodCommandsWithExtends(StringBuilder &sb, const std::string &prefix)
265 {
266 size_t extendMethods = 0;
267 AutoPtr<ASTInterfaceType> interface = interface_->GetExtendsInterface();
268 while (interface != nullptr) {
269 extendMethods += interface->GetMethodNumber();
270 interface = interface->GetExtendsInterface();
271 }
272
273 sb.Append(prefix).AppendFormat("enum {\n");
274 for (size_t i = 0; i < interface_->GetMethodNumber(); i++) {
275 AutoPtr<ASTMethod> method = interface_->GetMethod(i);
276 sb.Append(prefix + TAB)
277 .Append(EmitMethodCmdID(method))
278 .AppendFormat(" = %d", extendMethods + i + 1)
279 .Append(",\n");
280 }
281 sb.Append(prefix).Append("};\n");
282 }
283 } // namespace HDI
284 } // namespace OHOS