3 Copyright (c) 2006, Intel Corporation
4 All rights reserved. This program and the accompanying materials
5 are licensed and made available under the terms and conditions of the BSD License
6 which accompanies this distribution. The full text of the license may be found at
7 http://opensource.org/licenses/bsd-license.php
9 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
10 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
13 package org
.tianocore
.migration
;
17 import java
.util
.regex
.Matcher
;
18 import java
.util
.regex
.Pattern
;
20 import org
.tianocore
.UsageTypes
;
22 public final class SourceFileReplacer
implements Common
.ForDoAll
{
23 private static final SourceFileReplacer SFReplacer
= new SourceFileReplacer();
24 private ModuleInfo mi
;
25 private static final Set
<Common
.Laplace
> Laplaces
= new HashSet
<Common
.Laplace
>();
27 // these sets are used only for printing log of the changes in current file
28 private static final Set
<r8tor9
> filefunc
= new HashSet
<r8tor9
>();
29 private static final Set
<r8tor9
> filemacro
= new HashSet
<r8tor9
>();
30 private static final Set
<r8tor9
> fileguid
= new HashSet
<r8tor9
>();
31 private static final Set
<r8tor9
> fileppi
= new HashSet
<r8tor9
>();
32 private static final Set
<r8tor9
> fileprotocol
= new HashSet
<r8tor9
>();
33 private static final Set
<String
> filer8only
= new HashSet
<String
>();
35 private static final String
[] specialhoblibfunc
= {
37 "BuildResourceDescriptorHob",
43 "BuildMemoryAllocationHob"
45 private static final String
[] peiserviceslibfunc
= {
62 //---------------------------------------inner classes---------------------------------------//
63 private static class r8tor9
{
64 r8tor9(String r8
, String r9
) {
68 public String r8thing
;
69 public String r9thing
;
72 private class IdleLaplace
extends Common
.Laplace
{
73 public String
operation(String wholeline
) {
74 return replaceLibrary (wholeline
, mi
.hashmacro
);
77 public boolean recognize(String filename
) {
78 return filename
.contains(".h") || filename
.contains(".H") || filename
.contains(".uni");
81 public String
namechange(String oldname
) {
82 if (oldname
.contains(".H")) {
83 return oldname
.replaceFirst(".H", ".h");
89 private class DxsLaplace
extends Common
.Laplace
{
90 public String
operation(String wholeline
) {
91 wholeline
= replaceMacro(wholeline
, mi
.hashnonlocalmacro
);
92 if (mi
.getModuleType().equals("PEIM")) {
93 return addincludefile(wholeline
, "\\<PeimDepex.h\\>");
95 return addincludefile(wholeline
, "\\<DxeDepex.h\\>");
99 public boolean recognize(String filename
) {
100 return filename
.contains(".dxs");
103 public String
namechange(String oldname
) {
108 private class CLaplace
extends Common
.Laplace
{
109 public String
operation(String wholeline
) {
110 // remove EFI_DRIVER_ENTRY_POINT
111 wholeline
= wholeline
.replaceAll("(EFI_[A-Z]+_ENTRY_POINT\\s*\\(\\s*\\w(\\w|\\d)*\\s*\\))", MigrationTool
.MIGRATIONCOMMENT
+ " $1");
112 // redefine module entry point for some self-relocated modules
113 wholeline
= wholeline
.replaceAll (mi
.entrypoint
+ "([^{]*?})", "_ModuleEntryPoint" + "$1");
114 // remove R8 library contractor
115 wholeline
= wholeline
.replaceAll ("(\\b(?:Efi|Dxe)InitializeDriverLib\\b)", MigrationTool
.MIGRATIONCOMMENT
+ " $1");
116 // Add Library Class for potential reference of gBS, gRT & gDS.
117 if (Common
.find (wholeline
, "\\bg?BS\\b")) {
118 mi
.hashrequiredr9libs
.add("UefiBootServicesTableLib");
120 if (Common
.find (wholeline
, "\\bg?RT\\b")) {
121 mi
.hashrequiredr9libs
.add ("UefiRuntimeServicesTableLib");
123 if (Common
.find (wholeline
, "\\bgDS\\b")) {
124 mi
.hashrequiredr9libs
.add ("DxeServicesTableLib");
127 wholeline
= replaceLibrary (wholeline
, mi
.hashnonlocalfunc
);
128 wholeline
= replaceLibrary (wholeline
, mi
.hashmacro
);
130 wholeline
= replaceMacro (wholeline
, mi
.hashnonlocalmacro
);
133 replaceGuid(wholeline
, mi
.guids
, "guid", fileguid
);
134 replaceGuid(wholeline
, mi
.ppis
, "ppi", fileppi
);
135 replaceGuid(wholeline
, mi
.protocols
, "protocol", fileprotocol
);
138 if (mi
.getModuleType().matches("PEIM")) {
140 // Try to remove PeiServicesTablePointer;
142 wholeline
= dropPeiServicesPointer (wholeline
);
144 // Drop the possible return Status of Hob building function.
146 wholeline
= drophobLibReturnStatus (wholeline
);
149 // Expand obsolete R8 macro.
151 wholeline
= replaceObsoleteMacro (wholeline
);
153 show(filefunc
, "function");
154 show(filemacro
, "macro");
155 show(fileguid
, "guid");
156 show(fileppi
, "ppi");
157 show(fileprotocol
, "protocol");
158 if (!filer8only
.isEmpty()) {
159 MigrationTool
.ui
.println("Converting r8only : " + filer8only
);
166 fileprotocol
.clear();
172 public boolean recognize(String filename
) {
173 return filename
.contains(".c") || filename
.contains(".C");
176 public String
namechange(String oldname
) {
177 if (oldname
.contains(".C")) {
178 return oldname
.replaceFirst(".C", ".c");
184 //---------------------------------------inner classes---------------------------------------//
186 //-------------------------------------process functions-------------------------------------//
187 private static final String
addincludefile(String wholeline
, String hfile
) {
188 return wholeline
.replaceFirst("(\\*/\\s)", "$1\n#include " + hfile
+ "\n");
191 private static final void show(Set
<r8tor9
> hash
, String sh
) {
192 Iterator
<r8tor9
> it
= hash
.iterator();
194 if (!hash
.isEmpty()) {
195 MigrationTool
.ui
.print("Converting " + sh
+ " : ");
196 while (it
.hasNext()) {
198 MigrationTool
.ui
.print("[" + temp
.r8thing
+ "->" + temp
.r9thing
+ "] ");
200 MigrationTool
.ui
.println("");
204 private static final void replaceGuid(String line
, Set
<String
> hash
, String kind
, Set
<r8tor9
> filehash
) {
208 it
= hash
.iterator();
209 while (it
.hasNext()) {
211 if ((r9thing
= MigrationTool
.db
.getR9Guidname(r8thing
)) != null) {
212 if (!r8thing
.equals(r9thing
)) {
213 if (line
.contains(r8thing
)) {
214 line
= line
.replaceAll(r8thing
, r9thing
);
215 filehash
.add(new r8tor9(r8thing
, r9thing
));
222 private final String
dropPeiServicesPointer (String wholeline
) {
223 String peiServicesTablePointer
;
224 String peiServicesTableCaller
;
225 String regPeiServices
;
229 peiServicesTablePointer
= "\\w(?:\\w|[0-9]|->)*";
230 peiServicesTableCaller
= "\\(\\*\\*?\\s*(" + peiServicesTablePointer
+ ")\\s*\\)[.-]>?\\s*";
231 for (int i
= 0; i
< peiserviceslibfunc
.length
; i
++) {
232 regPeiServices
= peiServicesTableCaller
+ peiserviceslibfunc
[i
] + "\\s*\\(\\s*\\1\\s*,(\\t| )*";
233 ptnPei
= Pattern
.compile (regPeiServices
);
234 mtrPei
= ptnPei
.matcher (wholeline
);
236 wholeline
= mtrPei
.replaceAll("PeiServices" + peiserviceslibfunc
[i
] + " (");
237 mi
.hashrequiredr9libs
.add("PeiServicesLib");
240 regPeiServices
= peiServicesTableCaller
+ "(CopyMem|SetMem)" + "\\s*\\((\\t| )*";
241 ptnPei
= Pattern
.compile (regPeiServices
);
242 mtrPei
= ptnPei
.matcher (wholeline
);
244 wholeline
= mtrPei
.replaceAll("$2 (");
245 mi
.hashrequiredr9libs
.add("BaseMemoryLib");
248 ptnPei
= Pattern
.compile("#%+(\\s*\\(+\\s*)" + peiServicesTablePointer
+ "\\s*,\\s*", Pattern
.MULTILINE
);
249 mtrPei
= ptnPei
.matcher(wholeline
);
250 while (mtrPei
.find()) {
251 wholeline
= mtrPei
.replaceAll("$1");
257 private final String
drophobLibReturnStatus (String wholeline
) { // or use regex to find pattern "Status = ..."
258 Pattern ptnhobstatus
;
259 Matcher mtrhobstatus
;
260 String templine
= wholeline
;
261 for (int i
= 0; i
< specialhoblibfunc
.length
; i
++) {
262 ptnhobstatus
= Pattern
.compile("(Status\\s*=\\s*)?" + specialhoblibfunc
[i
] + "(.*?\\)\\s*;)", Pattern
.DOTALL
);
263 mtrhobstatus
= ptnhobstatus
.matcher(templine
);
264 if (mtrhobstatus
.find()) {
265 templine
= mtrhobstatus
.replaceAll(specialhoblibfunc
[i
] + mtrhobstatus
.group(2) + "\n " +
266 MigrationTool
.MIGRATIONCOMMENT
+ "R9 Hob-building library functions will assert if build failure.\n Status = EFI_SUCCESS;");
272 private final String
replaceMacro (String wholeline
, Set
<String
> symbolSet
) {
277 it
= symbolSet
.iterator();
278 while (it
.hasNext()) { //macros are all assumed MdePkg currently
280 //mi.hashrequiredr9libs.add(MigrationTool.db.getR9Lib(r8thing));
281 if ((r9thing
= MigrationTool
.db
.getR9Macro(r8thing
)) != null) {
282 if (wholeline
.contains(r8thing
)) {
283 String findString
= "(?<!(?:\\d|\\w))" + r8thing
+ "(?!(?:\\d|\\w))";
284 wholeline
= wholeline
.replaceAll(findString
, r9thing
);
285 filemacro
.add(new r8tor9(r8thing
, r9thing
));
292 private final String
replaceLibrary (String wholeline
, Set
<String
> symbolSet
) {
293 boolean addr8
= false;
294 // start replacing names
298 // Converting non-locla function
299 it
= symbolSet
.iterator();
300 while (it
.hasNext()) {
302 mi
.addLibraryClass(MigrationTool
.db
.getR9Lib(r8thing
), UsageTypes
.ALWAYS_CONSUMED
);
303 //mi.hashrequiredr9libs.add(MigrationTool.db.getR9Lib(r8thing)); // add a library here
306 if ((r9thing
= MigrationTool
.db
.getR9Func(r8thing
)) != null) {
307 if (!r8thing
.equals(r9thing
)) {
308 if (wholeline
.contains(r8thing
)) {
309 wholeline
= wholeline
.replaceAll(r8thing
, r9thing
);
310 filefunc
.add(new r8tor9(r8thing
, r9thing
));
311 Iterator
<r8tor9
> rt
= filefunc
.iterator();
312 while (rt
.hasNext()) {
314 if (MigrationTool
.db
.r8only
.contains(temp
.r8thing
)) {
315 filer8only
.add(r8thing
);
316 mi
.hashr8only
.add(r8thing
);
323 } //is any of the guids changed?
325 wholeline
= addincludefile(wholeline
, "\"R8Lib.h\"");
330 private final String
replaceObsoleteMacro (String wholeline
) {
332 mtrmac
= Pattern
.compile("EFI_IDIV_ROUND\\((.*), (.*)\\)").matcher(wholeline
);
334 wholeline
= mtrmac
.replaceAll("\\($1 \\/ $2 \\+ \\(\\(\\(2 \\* \\($1 \\% $2\\)\\) \\< $2\\) \\? 0 \\: 1\\)\\)");
336 mtrmac
= Pattern
.compile("EFI_MIN\\((.*), (.*)\\)").matcher(wholeline
);
338 wholeline
= mtrmac
.replaceAll("\\(\\($1 \\< $2\\) \\? $1 \\: $2\\)");
340 mtrmac
= Pattern
.compile("EFI_MAX\\((.*), (.*)\\)").matcher(wholeline
);
342 wholeline
= mtrmac
.replaceAll("\\(\\($1 \\> $2\\) \\? $1 \\: $2\\)");
344 mtrmac
= Pattern
.compile("EFI_UINTN_ALIGNED\\((.*)\\)").matcher(wholeline
);
346 wholeline
= mtrmac
.replaceAll("\\(\\(\\(UINTN\\) $1\\) \\& \\(sizeof \\(UINTN\\) \\- 1\\)\\)");
348 if (wholeline
.contains("EFI_UINTN_ALIGN_MASK")) {
349 wholeline
= wholeline
.replaceAll("EFI_UINTN_ALIGN_MASK", "(sizeof (UINTN) - 1)");
354 private final void addr8only() throws Exception
{
355 String paragraph
= null;
356 String line
= Common
.file2string(MigrationTool
.db
.DatabasePath
+ File
.separator
+ "R8Lib.c");
357 PrintWriter outfile1
= new PrintWriter(new BufferedWriter(new FileWriter(MigrationTool
.ModuleInfoMap
.get(mi
) + File
.separator
+ "Migration_" + mi
.modulename
+ File
.separator
+ "R8Lib.c")));
358 PrintWriter outfile2
= new PrintWriter(new BufferedWriter(new FileWriter(MigrationTool
.ModuleInfoMap
.get(mi
) + File
.separator
+ "Migration_" + mi
.modulename
+ File
.separator
+ "R8Lib.h")));
359 Pattern ptnr8only
= Pattern
.compile("////#?(\\w*)?(.*?R8_(\\w*).*?)////~", Pattern
.DOTALL
);
360 Matcher mtrr8only
= ptnr8only
.matcher(line
);
361 Matcher mtrr8onlyhead
;
364 Matcher mtrr8onlyheadcomment
= Critic
.PTN_NEW_HEAD_COMMENT
.matcher(line
);
365 if (mtrr8onlyheadcomment
.find()) {
366 outfile1
.append(mtrr8onlyheadcomment
.group() + "\n\n");
367 outfile2
.append(mtrr8onlyheadcomment
.group() + "\n\n");
371 while (mtrr8only
.find()) {
372 if (mi
.hashr8only
.contains(mtrr8only
.group(3))) {
373 paragraph
= mtrr8only
.group(2);
374 outfile1
.append(paragraph
+ "\n\n");
375 if (mtrr8only
.group(1).length() != 0) {
376 mi
.hashrequiredr9libs
.add(mtrr8only
.group(1));
379 while ((mtrr8onlyhead
= Func
.ptnbrace
.matcher(paragraph
)).find()) {
380 paragraph
= mtrr8onlyhead
.replaceAll(";");
382 outfile2
.append(paragraph
+ "\n\n");
390 mi
.localmodulesources
.add("R8Lib.h");
391 mi
.localmodulesources
.add("R8Lib.c");
393 //-------------------------------------process functions-------------------------------------//
395 //-----------------------------------ForDoAll-----------------------------------//
396 public void run(String filepath
) throws Exception
{
397 String inname
= filepath
.replace(mi
.temppath
+ File
.separator
, "");
398 String tempinpath
= mi
.temppath
+ File
.separator
;
399 String tempoutpath
= MigrationTool
.ModuleInfoMap
.get(mi
) + File
.separator
+ "Migration_" + mi
.modulename
+ File
.separator
;
401 Iterator
<Common
.Laplace
> itLaplace
= Laplaces
.iterator();
402 while (itLaplace
.hasNext()) {
403 Common
.Laplace lap
= itLaplace
.next();
404 if (lap
.recognize(inname
)) {
405 MigrationTool
.ui
.println("\nHandling file: " + inname
);
406 lap
.transform(tempinpath
+ inname
, tempoutpath
+ lap
.namechange(inname
));
411 public boolean filter(File dir
) {
414 //-----------------------------------ForDoAll-----------------------------------//
416 private final void setModuleInfo(ModuleInfo moduleinfo
) {
420 private final void start() throws Exception
{
421 Laplaces
.add(new DxsLaplace());
422 Laplaces
.add(new CLaplace());
423 Laplaces
.add(new IdleLaplace());
425 Common
.toDoAll(mi
.temppath
, this, Common
.FILE
);
427 if (!mi
.hashr8only
.isEmpty()) {
434 public static final void fireAt(ModuleInfo moduleinfo
) throws Exception
{
435 SFReplacer
.setModuleInfo(moduleinfo
);