From 397d69c72a6d6c9703b8e50f6e03dfcc67216af2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ryszard=20Wi=C5=9Bniewski?= <brut.alll@gmail.com>
Date: Thu, 3 Jun 2010 09:25:42 +0200
Subject: [PATCH] +Androlib.installFramework()

---
 src/brut/androlib/Androlib.java              |  5 ++
 src/brut/androlib/res/AndrolibResources.java | 65 +++++++++++++++++++-
 2 files changed, 67 insertions(+), 3 deletions(-)

diff --git a/src/brut/androlib/Androlib.java b/src/brut/androlib/Androlib.java
index 37c3547a..9ec6067d 100644
--- a/src/brut/androlib/Androlib.java
+++ b/src/brut/androlib/Androlib.java
@@ -345,6 +345,11 @@ public class Androlib {
         mAndRes.publicizeResources(arscFile);
     }
 
+    public void installFramework(File frameFile, String tag)
+            throws AndrolibException {
+        mAndRes.installFramework(frameFile, tag);
+    }
+
     public boolean isFrameworkApk(ResTable resTable) {
         for (ResPackage pkg : resTable.listMainPackages()) {
             if (pkg.getId() < 64) {
diff --git a/src/brut/androlib/res/AndrolibResources.java b/src/brut/androlib/res/AndrolibResources.java
index 26cbbd98..fbbc2ef1 100644
--- a/src/brut/androlib/res/AndrolibResources.java
+++ b/src/brut/androlib/res/AndrolibResources.java
@@ -22,6 +22,7 @@ import brut.androlib.err.CantFindFrameworkResException;
 import brut.androlib.res.data.*;
 import brut.androlib.res.data.value.ResXmlSerializable;
 import brut.androlib.res.decoder.*;
+import brut.androlib.res.decoder.ARSCDecoder.ARSCData;
 import brut.androlib.res.decoder.ARSCDecoder.FlagsOffset;
 import brut.androlib.res.util.ExtFile;
 import brut.androlib.res.util.ExtMXSerializer;
@@ -31,6 +32,7 @@ import brut.util.*;
 import java.io.*;
 import java.util.*;
 import java.util.logging.Logger;
+import java.util.zip.*;
 import org.apache.commons.io.IOUtils;
 import org.xmlpull.v1.XmlSerializer;
 
@@ -347,6 +349,58 @@ final public class AndrolibResources {
         throw new CantFindFrameworkResException(id);
     }
 
+    public void installFramework(File frameFile, String tag)
+            throws AndrolibException {
+        InputStream in = null;
+        ZipOutputStream out = null;
+        try {
+            ZipFile zip = new ZipFile(frameFile);
+            ZipEntry entry = zip.getEntry("resources.arsc");
+
+            if (entry == null) {
+                throw new AndrolibException("Can't find resources.arsc file");
+            }
+
+            in = zip.getInputStream(entry);
+            byte[] data = IOUtils.toByteArray(in);
+
+            ARSCData arsc = ARSCDecoder.decode(
+                new ByteArrayInputStream(data), true);
+            publicizeResources(data, arsc.getFlagsOffsets());
+
+            File outFile = new File(getFrameworkDir(),
+                String.valueOf(arsc.getOnePackage().getId()) +
+                (tag == null ? "" : '-' + tag) + ".apk");
+
+            out = new ZipOutputStream(new FileOutputStream(outFile));
+            out.setMethod(ZipOutputStream.STORED);
+            CRC32 crc = new CRC32();
+            crc.update(data);
+            entry = new ZipEntry("resources.arsc");
+            entry.setSize(data.length);
+            entry.setCrc(crc.getValue());
+            out.putNextEntry(entry);
+            out.write(data);
+
+            LOGGER.info("Framework installed to: " + outFile);
+        } catch (ZipException ex) {
+            throw new AndrolibException(ex);
+        } catch (IOException ex) {
+            throw new AndrolibException(ex);
+        } finally {
+            if (in != null) {
+                try {
+                    in.close();
+                } catch (IOException ex) {}
+            }
+            if (out != null) {
+                try {
+                    out.close();
+                } catch (IOException ex) {}
+            }
+        }
+    }
+
     public void publicizeResources(File arscFile) throws AndrolibException {
         byte[] data = new byte[(int) arscFile.length()];
 
@@ -377,9 +431,14 @@ final public class AndrolibResources {
     }
 
     public void publicizeResources(byte[] arsc) throws AndrolibException {
-        for (FlagsOffset flags :
-                ARSCDecoder.decode(new ByteArrayInputStream(arsc), true)
-                .getFlagsOffsets()) {
+        publicizeResources(arsc,
+            ARSCDecoder.decode(new ByteArrayInputStream(arsc), true)
+            .getFlagsOffsets());
+    }
+
+    public void publicizeResources(byte[] arsc, FlagsOffset[] flagsOffsets)
+            throws AndrolibException {
+        for (FlagsOffset flags : flagsOffsets) {
             int offset = flags.offset + 3;
             int end = offset + 4 * flags.count;
             while(offset < end) {