From 73f7b9c29ce70f890a9d674abfbb9be21cb7de5c Mon Sep 17 00:00:00 2001
From: jwang36 <jwang36@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Mon, 5 Feb 2007 06:25:58 +0000
Subject: [PATCH 1/1] Add genDigest() method to class FfsProcess to generate
 MD5 digest value for the FFS layout. This is intended to solve re-generation
 issue of FFS when its layout has been changed but no related section files
 have been  changed.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@2353 6f19259b-4bc3-4df7-8a09-765794883524
---
 .../org/tianocore/build/FfsProcess.java       | 84 +++++++++++++++++++
 1 file changed, 84 insertions(+)

diff --git a/Tools/Java/Source/GenBuild/org/tianocore/build/FfsProcess.java b/Tools/Java/Source/GenBuild/org/tianocore/build/FfsProcess.java
index dd86346b7f..f63b7d95f0 100644
--- a/Tools/Java/Source/GenBuild/org/tianocore/build/FfsProcess.java
+++ b/Tools/Java/Source/GenBuild/org/tianocore/build/FfsProcess.java
@@ -14,6 +14,11 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 package org.tianocore.build;
 
 import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileOutputStream;
+import java.io.FileReader;
+import java.io.FileWriter;
+import java.security.MessageDigest;
 import java.util.Vector;
 
 import javax.xml.namespace.QName;
@@ -55,6 +60,7 @@ public class FfsProcess {
 
     private BuildOptionsDocument.BuildOptions.Ffs ffsXmlObject;
 
+    private Project project = null;
     ///
     /// ANT script to call GenFfs
     ///
@@ -113,6 +119,7 @@ public class FfsProcess {
               If can't find FFS Layout in FPD.
     **/
     public boolean initSections(String buildType, Project project, FpdModuleIdentification fpdModuleId) throws BuildException {
+        this.project = project;
         //
         // Try to find Ffs layout from FPD file
         //
@@ -121,6 +128,7 @@ public class FfsProcess {
         for (int i = 0; i < ffsArray.length; i++) {
             if (isMatch(ffsArray[i].getFfsKey(), buildType)) {
                 ffsXmlObject = ffsArray[i];
+                genDigest();
                 return true;
             }
         }
@@ -196,6 +204,13 @@ public class FfsProcess {
             pathEle.setAttribute("name", getSectionFile(basename, section));
             sourceEle.appendChild(pathEle);
         }
+        //
+        // add FFS layout digest into the source file list
+        // 
+        Element pathEle = document.createElement("file");
+        pathEle.setAttribute("name", "${DEST_DIR_OUTPUT}" + File.separator + "ffs.md5");
+        sourceEle.appendChild(pathEle);
+
         String[] result = sections.toArray(new String[sections.size()]);
 
         outofdateEle.appendChild(sourceEle);
@@ -439,4 +454,73 @@ public class FfsProcess {
     public Element getFfsNode() {
         return ffsNode;
     }
+
+    private void genDigest() {
+        String digestFilePath = project.getProperty("DEST_DIR_OUTPUT");
+        if (digestFilePath == null) {
+            EdkLog.log(EdkLog.EDK_WARNING, "Warning: cannot get DEST_DIR_OUTPUT!");
+            return;
+        }
+
+        //
+        // use MD5 algorithm
+        // 
+        MessageDigest md5 = null;
+        try {
+            md5 = MessageDigest.getInstance("MD5");
+            //
+            // convert the FFS layout XML DOM tree into string and use it to
+            // calculate its MD5 digest value
+            // 
+            md5.update(ffsXmlObject.xmlText().getBytes());
+        } catch (Exception e) {
+            EdkLog.log(EdkLog.EDK_WARNING, "Warning: " + e.getMessage());
+            return;
+        }
+
+        //
+        // get the MD5 digest value
+        // 
+        byte[] digest = md5.digest();
+
+        //
+        // put the digest in a file named "ffs.md5" if it doesn't exist, otherwise
+        // we will compare the digest in the file with the one just calculated
+        // 
+        digestFilePath += File.separator + "ffs.md5";
+        File digestFile = new File(digestFilePath);
+        if (digestFile.exists()) {
+            byte[] oldDigest = new byte[digest.length];
+            try {
+                FileInputStream fIn = new FileInputStream(digestFile);
+                fIn.read(oldDigest);
+                fIn.close();
+            } catch (Exception e) {
+                throw new BuildException(e.getMessage());
+            }
+
+            boolean noChange = true;
+            for (int i = 0; i < oldDigest.length; ++i) {
+                if (digest[i] != oldDigest[i]) {
+                    noChange = false;
+                    break;
+                }
+            }
+
+            if (noChange) {
+                return;
+            }
+        }
+
+        //
+        // update the "ffs.md5" file content with new digest value
+        // 
+        try {
+            FileOutputStream fOut = new FileOutputStream(digestFile);
+            fOut.write(digest);
+            fOut.close();
+        } catch (Exception e) {
+            throw new BuildException(e.getMessage());
+        }
+    }
 }
-- 
2.39.2