]> git.proxmox.com Git - mirror_edk2.git/blame - Tools/Source/MigrationTools/org/tianocore/migration/SourceFileReplacer.java
REMODEL OF SourceFileReplacer.java
[mirror_edk2.git] / Tools / Source / MigrationTools / org / tianocore / migration / SourceFileReplacer.java
CommitLineData
b0282412 1/** @file\r
2 \r
3 Copyright (c) 2006, Intel Corporation\r
4 All rights reserved. This program and the accompanying materials\r
5 are licensed and made available under the terms and conditions of the BSD License\r
6 which accompanies this distribution. The full text of the license may be found at\r
7 http://opensource.org/licenses/bsd-license.php\r
8 \r
9 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
10 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
11 \r
12 **/\r
0dc8c589 13package org.tianocore.migration;\r
14\r
15import java.io.*;\r
16import java.util.*;\r
17import java.util.regex.Matcher;\r
18import java.util.regex.Pattern;\r
19\r
16b7eeef 20public final class SourceFileReplacer implements Common.ForDoAll {\r
21 private static final SourceFileReplacer SFReplacer = new SourceFileReplacer();\r
22 private ModuleInfo mi;\r
23 private static boolean showdetails = true; // set this as default now, may be changed in the future\r
0dc8c589 24 \r
eee63a7b 25 private static class r8tor9 {\r
0dc8c589 26 r8tor9(String r8, String r9) {\r
27 r8thing = r8;\r
28 r9thing = r9;\r
29 }\r
30 public String r8thing;\r
31 public String r9thing;\r
32 }\r
33 \r
34 // these sets are used only for printing log of the changes in current file\r
eee63a7b 35 private static final Set<r8tor9> filefunc = new HashSet<r8tor9>();\r
36 private static final Set<r8tor9> filemacro = new HashSet<r8tor9>();\r
37 private static final Set<r8tor9> fileguid = new HashSet<r8tor9>();\r
38 private static final Set<r8tor9> fileppi = new HashSet<r8tor9>();\r
39 private static final Set<r8tor9> fileprotocol = new HashSet<r8tor9>();\r
40 private static final Set<String> filer8only = new HashSet<String>();\r
0dc8c589 41 \r
9481cdac 42 private static final String addincludefile(String wholeline, String hfile) {\r
43 return wholeline.replaceFirst("(\\*/\\s)", "$1\n#include " + hfile + "\n");\r
44 }\r
45 \r
16b7eeef 46 private final String convertdxs(String wholeline) {\r
9481cdac 47 if (mi.getModuleType().equals("PEIM")) {\r
48 return addincludefile(wholeline, "\\<PeimDepex.h\\>");\r
49 } else {\r
50 return addincludefile(wholeline, "\\<DxeDepex.h\\>");\r
51 }\r
52 }\r
53 \r
16b7eeef 54 private final void addr8only() throws Exception {\r
0dc8c589 55 String paragraph = null;\r
482407d3 56 String line = Common.file2string(MigrationTool.db.DatabasePath + File.separator + "R8Lib.c");\r
5f4eb6b6 57 PrintWriter outfile1 = new PrintWriter(new BufferedWriter(new FileWriter(MigrationTool.ModuleInfoMap.get(mi) + File.separator + "Migration_" + mi.modulename + File.separator + "R8Lib.c")));\r
58 PrintWriter outfile2 = new PrintWriter(new BufferedWriter(new FileWriter(MigrationTool.ModuleInfoMap.get(mi) + File.separator + "Migration_" + mi.modulename + File.separator + "R8Lib.h")));\r
0dc8c589 59 Pattern ptnr8only = Pattern.compile("////#?(\\w*)?.*?R8_(\\w*).*?////~", Pattern.DOTALL);\r
60 Matcher mtrr8only = ptnr8only.matcher(line);\r
61 Matcher mtrr8onlyhead;\r
62 while (mtrr8only.find()) {\r
63 if (mi.hashr8only.contains(mtrr8only.group(2))) {\r
64 paragraph = mtrr8only.group();\r
65 outfile1.append(paragraph + "\n\n");\r
66 if (mtrr8only.group(1).length() != 0) {\r
67 mi.hashrequiredr9libs.add(mtrr8only.group(1));\r
68 }\r
69 //generate R8lib.h\r
70 while ((mtrr8onlyhead = Func.ptnbrace.matcher(paragraph)).find()) {\r
71 paragraph = mtrr8onlyhead.replaceAll(";");\r
72 }\r
73 outfile2.append(paragraph + "\n\n");\r
74 }\r
75 }\r
76 outfile1.flush();\r
77 outfile1.close();\r
78 outfile2.flush();\r
79 outfile2.close();\r
e6a5df3b 80 \r
81 mi.localmodulesources.add("R8Lib.h");\r
82 mi.localmodulesources.add("R8Lib.c");\r
0dc8c589 83 }\r
84 \r
0dc8c589 85 // Caution : if there is @ in file , it will be replaced with \n , so is you use Doxygen ... God Bless you!\r
16b7eeef 86 private final String sourcefilereplace(String wholeline) throws Exception {\r
0dc8c589 87 boolean addr8 = false;\r
88\r
89 Pattern pat = Pattern.compile("g?(BS|RT)(\\s*->\\s*)([a-zA-Z_]\\w*)", Pattern.MULTILINE); // ! only two level () bracket allowed !\r
90 //Pattern ptnpei = Pattern.compile("\\(\\*\\*?PeiServices\\)[.-][>]?\\s*(\\w*[#$]*)(\\s*\\(([^\\(\\)]*(\\([^\\(\\)]*\\))?[^\\(\\)]*)*\\))", Pattern.MULTILINE);\r
91\r
0dc8c589 92 // replace BS -> gBS , RT -> gRT\r
9481cdac 93 Matcher mat = pat.matcher(wholeline);\r
0dc8c589 94 if (mat.find()) { // add a library here\r
482407d3 95 MigrationTool.ui.println("Converting all BS->gBS, RT->gRT");\r
9481cdac 96 wholeline = mat.replaceAll("g$1$2$3"); //unknown correctiveness\r
0dc8c589 97 }\r
98 mat.reset();\r
99 while (mat.find()) {\r
100 if (mat.group(1).matches("BS")) {\r
101 mi.hashrequiredr9libs.add("UefiBootServicesTableLib");\r
102 }\r
103 if (mat.group(1).matches("RT")) {\r
104 mi.hashrequiredr9libs.add("UefiRuntimeServicesTableLib");\r
105 }\r
106 }\r
107 /*\r
108 // remove EFI_DRIVER_ENTRY_POINT\r
109 Pattern patentrypoint = Pattern.compile("EFI_DRIVER_ENTRY_POINT[^\\}]*\\}");\r
110 Matcher matentrypoint = patentrypoint.matcher(line);\r
111 if (matentrypoint.find()) {\r
482407d3 112 MigrationTool.ui.println("Deleting Entry_Point");\r
0dc8c589 113 line = matentrypoint.replaceAll("");\r
114 }\r
115 */\r
116 // start replacing names\r
6f55be9f 117 String r8thing;\r
118 String r9thing;\r
0dc8c589 119 Iterator<String> it;\r
120 // Converting non-locla function\r
121 it = mi.hashnonlocalfunc.iterator();\r
122 while (it.hasNext()) {\r
123 r8thing = it.next();\r
124 if (r8thing.matches("EfiInitializeDriverLib")) { //s\r
125 mi.hashrequiredr9libs.add("UefiBootServicesTableLib"); //p\r
126 mi.hashrequiredr9libs.add("UefiRuntimeServicesTableLib"); //e\r
127 } else if (r8thing.matches("DxeInitializeDriverLib")) { //c\r
128 mi.hashrequiredr9libs.add("UefiBootServicesTableLib"); //i\r
129 mi.hashrequiredr9libs.add("UefiRuntimeServicesTableLib"); //a\r
130 mi.hashrequiredr9libs.add("DxeServicesTableLib"); //l\r
131 } else { //\r
482407d3 132 mi.hashrequiredr9libs.add(MigrationTool.db.getR9Lib(r8thing)); // add a library here\r
0dc8c589 133 }\r
6f55be9f 134\r
135 r8tor9 temp;\r
482407d3 136 if ((r9thing = MigrationTool.db.getR9Func(r8thing)) != null) {\r
0dc8c589 137 if (!r8thing.equals(r9thing)) {\r
9481cdac 138 if (wholeline.contains(r8thing)) {\r
139 wholeline = wholeline.replaceAll(r8thing, r9thing);\r
0dc8c589 140 filefunc.add(new r8tor9(r8thing, r9thing));\r
141 Iterator<r8tor9> rt = filefunc.iterator();\r
142 while (rt.hasNext()) {\r
143 temp = rt.next();\r
482407d3 144 if (MigrationTool.db.r8only.contains(temp.r8thing)) {\r
0dc8c589 145 filer8only.add(r8thing);\r
146 mi.hashr8only.add(r8thing);\r
147 addr8 = true;\r
148 }\r
149 }\r
150 }\r
151 }\r
152 }\r
153 } //is any of the guids changed?\r
154 if (addr8 == true) {\r
9481cdac 155 wholeline = addincludefile(wholeline, "\"R8Lib.h\"");\r
0dc8c589 156 }\r
157 \r
158 // Converting macro\r
159 it = mi.hashnonlocalmacro.iterator();\r
160 while (it.hasNext()) { //macros are all assumed MdePkg currently\r
161 r8thing = it.next();\r
482407d3 162 //mi.hashrequiredr9libs.add(MigrationTool.db.getR9Lib(r8thing)); \r
163 if ((r9thing = MigrationTool.db.getR9Macro(r8thing)) != null) {\r
9481cdac 164 if (wholeline.contains(r8thing)) {\r
165 wholeline = wholeline.replaceAll(r8thing, r9thing);\r
0dc8c589 166 filemacro.add(new r8tor9(r8thing, r9thing));\r
167 }\r
168 }\r
169 }\r
170\r
171 // Converting guid\r
9481cdac 172 replaceGuid(wholeline, mi.guid, "guid", fileguid);\r
173 replaceGuid(wholeline, mi.ppi, "ppi", fileppi);\r
174 replaceGuid(wholeline, mi.protocol, "protocol", fileprotocol);\r
0dc8c589 175\r
176 // Converting Pei\r
177 // First , find all (**PeiServices)-> or (*PeiServices). with arg "PeiServices" , change name and add #%\r
178 Pattern ptnpei = Pattern.compile("\\(\\*\\*?PeiServices\\)[.-][>]?\\s*(\\w*)(\\s*\\(\\s*PeiServices\\s*,\\s*)", Pattern.MULTILINE);\r
179 if (mi.moduletype.contains("PEIM")) {\r
9481cdac 180 Matcher mtrpei = ptnpei.matcher(wholeline);\r
0dc8c589 181 while (mtrpei.find()) { // ! add a library here !\r
9481cdac 182 wholeline = mtrpei.replaceAll("PeiServices$1#%$2");\r
0dc8c589 183 mi.hashrequiredr9libs.add("PeiServicesLib");\r
184 }\r
185 mtrpei.reset();\r
9481cdac 186 if (wholeline.contains("PeiServicesCopyMem")) {\r
187 wholeline = wholeline.replaceAll("PeiServicesCopyMem#%", "CopyMem");\r
0dc8c589 188 mi.hashrequiredr9libs.add("BaseMemoryLib");\r
189 }\r
9481cdac 190 if (wholeline.contains("PeiServicesSetMem")) {\r
191 wholeline = wholeline.replaceAll("PeiServicesSetMem#%", "SetMem");\r
0dc8c589 192 mi.hashrequiredr9libs.add("BaseMemoryLib");\r
193 }\r
194\r
195 // Second , find all #% to drop the arg "PeiServices"\r
196 Pattern ptnpeiarg = Pattern.compile("#%+(\\s*\\(+\\s*)PeiServices\\s*,\\s*", Pattern.MULTILINE);\r
9481cdac 197 Matcher mtrpeiarg = ptnpeiarg.matcher(wholeline);\r
0dc8c589 198 while (mtrpeiarg.find()) {\r
9481cdac 199 wholeline = mtrpeiarg.replaceAll("$1");\r
0dc8c589 200 }\r
201 }\r
202 \r
203 Matcher mtrmac;\r
9481cdac 204 mtrmac = Pattern.compile("EFI_IDIV_ROUND\\((.*), (.*)\\)").matcher(wholeline);\r
0dc8c589 205 if (mtrmac.find()) {\r
9481cdac 206 wholeline = mtrmac.replaceAll("\\($1 \\/ $2 \\+ \\(\\(\\(2 \\* \\($1 \\% $2\\)\\) \\< $2\\) \\? 0 \\: 1\\)\\)");\r
0dc8c589 207 }\r
9481cdac 208 mtrmac = Pattern.compile("EFI_MIN\\((.*), (.*)\\)").matcher(wholeline);\r
0dc8c589 209 if (mtrmac.find()) {\r
9481cdac 210 wholeline = mtrmac.replaceAll("\\(\\($1 \\< $2\\) \\? $1 \\: $2\\)");\r
0dc8c589 211 }\r
9481cdac 212 mtrmac = Pattern.compile("EFI_MAX\\((.*), (.*)\\)").matcher(wholeline);\r
0dc8c589 213 if (mtrmac.find()) {\r
9481cdac 214 wholeline = mtrmac.replaceAll("\\(\\($1 \\> $2\\) \\? $1 \\: $2\\)");\r
0dc8c589 215 }\r
9481cdac 216 mtrmac = Pattern.compile("EFI_UINTN_ALIGNED\\((.*)\\)").matcher(wholeline);\r
0dc8c589 217 if (mtrmac.find()) {\r
9481cdac 218 wholeline = mtrmac.replaceAll("\\(\\(\\(UINTN\\) $1\\) \\& \\(sizeof \\(UINTN\\) \\- 1\\)\\)");\r
0dc8c589 219 }\r
9481cdac 220 if (wholeline.contains("EFI_UINTN_ALIGN_MASK")) {\r
221 wholeline = wholeline.replaceAll("EFI_UINTN_ALIGN_MASK", "(sizeof (UINTN) - 1)");\r
0dc8c589 222 }\r
223\r
224 show(filefunc, "function");\r
225 show(filemacro, "macro");\r
226 show(fileguid, "guid");\r
227 show(fileppi, "ppi");\r
228 show(fileprotocol, "protocol");\r
229 if (!filer8only.isEmpty()) {\r
482407d3 230 MigrationTool.ui.println("Converting r8only : " + filer8only);\r
0dc8c589 231 }\r
232\r
233 filefunc.clear();\r
234 filemacro.clear();\r
235 fileguid.clear();\r
236 fileppi.clear();\r
237 fileprotocol.clear();\r
238 filer8only.clear();\r
e6a5df3b 239\r
9481cdac 240 return wholeline;\r
0dc8c589 241 }\r
242 \r
eee63a7b 243 private static final void show(Set<r8tor9> hash, String sh) {\r
0dc8c589 244 Iterator<r8tor9> it = hash.iterator();\r
245 r8tor9 temp;\r
246 if (showdetails) {\r
247 if (!hash.isEmpty()) {\r
482407d3 248 MigrationTool.ui.print("Converting " + sh + " : ");\r
0dc8c589 249 while (it.hasNext()) {\r
250 temp = it.next();\r
482407d3 251 MigrationTool.ui.print("[" + temp.r8thing + "->" + temp.r9thing + "] ");\r
0dc8c589 252 }\r
482407d3 253 MigrationTool.ui.println("");\r
0dc8c589 254 }\r
255 }\r
256 }\r
257 \r
eee63a7b 258 private static final void replaceGuid(String line, Set<String> hash, String kind, Set<r8tor9> filehash) {\r
0dc8c589 259 Iterator<String> it;\r
260 String r8thing;\r
261 String r9thing;\r
262 it = hash.iterator();\r
263 while (it.hasNext()) {\r
264 r8thing = it.next();\r
482407d3 265 if ((r9thing = MigrationTool.db.getR9Guidname(r8thing)) != null) {\r
0dc8c589 266 if (!r8thing.equals(r9thing)) {\r
267 if (line.contains(r8thing)) {\r
268 line = line.replaceAll(r8thing, r9thing);\r
269 filehash.add(new r8tor9(r8thing, r9thing));\r
270 }\r
271 }\r
272 }\r
273 }\r
274 }\r
16b7eeef 275 \r
276 //-----------------------------------ForDoAll-----------------------------------//\r
277 public void run(String filepath) throws Exception {\r
278 String outname = null;\r
279 String inname = filepath.replace(mi.modulepath + File.separator, "");\r
280 String tempinpath = mi.modulepath + File.separator + "temp" + File.separator;\r
281 String tempoutpath = MigrationTool.ModuleInfoMap.get(mi) + File.separator + "Migration_" + mi.modulename + File.separator;\r
282\r
283 if (inname.contains(".c") || inname.contains(".C")) {\r
284 if (inname.contains(".C")) {\r
285 outname = inname.replaceFirst(".C", ".c");\r
286 } else {\r
287 outname = inname;\r
288 }\r
289 MigrationTool.ui.println("\nModifying file: " + inname);\r
290 Common.string2file(sourcefilereplace(Common.file2string(tempinpath + inname)), tempoutpath + outname);\r
291 } else if (inname.contains(".h") || inname.contains(".H") || inname.contains(".uni")) {\r
292 if (inname.contains(".H")) {\r
293 outname = inname.replaceFirst(".H", ".h");\r
294 } else {\r
295 outname = inname;\r
296 }\r
297 MigrationTool.ui.println("\nCopying file: " + inname);\r
298 Common.string2file(Common.file2string(tempinpath + inname), tempoutpath + outname);\r
299 } else if (inname.contains(".dxs")) {\r
300 outname = inname;\r
301 MigrationTool.ui.println("\nModifying file: " + inname);\r
302 Common.string2file(convertdxs(Common.file2string(tempinpath + inname)), tempoutpath + outname);\r
303 }\r
304 }\r
305 \r
306 public boolean dirFilter(String filepath) {\r
307 return true;\r
308 }\r
309 \r
310 public boolean fileFilter(String filepath) {\r
311 return true;\r
312 }\r
313 //-----------------------------------ForDoAll-----------------------------------//\r
314 \r
315 private final void setModuleInfo(ModuleInfo moduleinfo) {\r
316 mi = moduleinfo;\r
317 }\r
318 \r
319 private final void start() throws Exception {\r
320 Common.toDoAll(mi.localmodulesources, this);\r
321 \r
322 if (!mi.hashr8only.isEmpty()) {\r
323 addr8only();\r
324 }\r
325 }\r
326 \r
327 public static final void fireAt(ModuleInfo moduleinfo) throws Exception {\r
328 SFReplacer.setModuleInfo(moduleinfo);\r
329 SFReplacer.start();\r
330 }\r
0dc8c589 331}\r