groovy unpack/pack tasks can run, but still failed unit test
cfig
9 years ago
parent
460c5b05eb
commit
896d6065f9
@ -0,0 +1,347 @@
|
||||
package cfig.bootimg
|
||||
|
||||
import java.nio.ByteBuffer;
|
||||
import joptsimple.OptionParser;
|
||||
import joptsimple.OptionSet;
|
||||
import java.nio.channels.FileChannel;
|
||||
import java.nio.ByteOrder;
|
||||
import java.util.regex.Matcher;
|
||||
import java.util.regex.Pattern;
|
||||
import java.security.MessageDigest;
|
||||
|
||||
/**
|
||||
* Created by yu at 10:52 on 2016-06-18
|
||||
*/
|
||||
class Packer {
|
||||
CArgs parse_cmdline(String[] inArgs) {
|
||||
OptionParser parser = new OptionParser();
|
||||
parser.accepts("kernel", "path to the kernel").withRequiredArg();
|
||||
parser.accepts("ramdisk", "path to the ramdisk").withRequiredArg();
|
||||
parser.accepts("second", "path to the 2nd bootloader").withRequiredArg();
|
||||
parser.accepts("cmdline", "extra arguments to be passed on the kernel command line").withRequiredArg();
|
||||
parser.accepts("base", "base address").withRequiredArg();
|
||||
parser.accepts("kernel_offset", "kernel offset").withRequiredArg();
|
||||
parser.accepts("ramdisk_offset", "ramdisk offset").withRequiredArg();
|
||||
parser.accepts("second_offset", "2nd bootloader offset").withRequiredArg();
|
||||
parser.accepts("os_version", "operating system version").withRequiredArg();
|
||||
parser.accepts("os_patch_level", "operating system patch level").withRequiredArg();
|
||||
parser.accepts("tags_offset", "tags offset").withRequiredArg();
|
||||
parser.accepts("board", "board name").withRequiredArg();
|
||||
parser.accepts("pagesize", "page size").withRequiredArg();
|
||||
parser.accepts("id", "print the image ID on standard output");
|
||||
parser.accepts("output", "output file name").withRequiredArg();
|
||||
|
||||
OptionSet options = parser.parse(inArgs)
|
||||
CArgs ret = new CArgs();
|
||||
|
||||
ret.kernel = options.valueOf("kernel")
|
||||
|
||||
ret.output = options.valueOf("output")
|
||||
|
||||
ret.ramdisk = options.valueOf("ramdisk")
|
||||
|
||||
ret.second = options.valueOf("second")
|
||||
|
||||
if (options.has("board")) {
|
||||
ret.board = options.valueOf("board")
|
||||
} else {
|
||||
ret.board = ""
|
||||
}
|
||||
|
||||
ret.id = options.has("id")
|
||||
|
||||
if (options.has("base")) {
|
||||
ret.base = Integer.decode(options.valueOf("base"))
|
||||
} else {
|
||||
ret.base = 0x10000000;
|
||||
}
|
||||
|
||||
if (options.has("kernel_offset")) {
|
||||
ret.kernel_offset = Integer.decode(options.valueOf("kernel_offset"))
|
||||
} else {
|
||||
ret.kernel_offset = 0x00008000;
|
||||
}
|
||||
|
||||
if (options.has("ramdisk_offset")) {
|
||||
ret.ramdisk_offset = Integer.decode(options.valueOf("ramdisk_offset"))
|
||||
} else {
|
||||
ret.ramdisk_offset = 0x01000000
|
||||
}
|
||||
|
||||
ret.os_version = options.valueOf("os_version")
|
||||
|
||||
ret.os_patch_level = options.valueOf("os_patch_level")
|
||||
|
||||
if (options.has("second_offset")) {
|
||||
ret.second_offset = Integer.decode(options.valueOf("second_offset"))
|
||||
} else {
|
||||
ret.second_offset = 0x00f00000
|
||||
}
|
||||
|
||||
if (options.has("tags_offset")) {
|
||||
ret.tags_offset = Integer.decode(options.valueOf("tags_offset"))
|
||||
} else {
|
||||
ret.tags_offset = 0x00000100
|
||||
}
|
||||
|
||||
if (options.has("pagesize")) {
|
||||
ret.pagesize = Integer.decode(options.valueOf("pagesize"))
|
||||
} else {
|
||||
ret.pagesize = 2048
|
||||
}
|
||||
|
||||
if (options.has("cmdline")) {
|
||||
ret.cmdline = options.valueOf("cmdline")
|
||||
} else {
|
||||
ret.cmdline = ""
|
||||
}
|
||||
|
||||
if (ret.cmdline.length() > 1536) {
|
||||
println("cmdline length must <= 1536, current is " + ret.cmdline.length());
|
||||
printUsage(parser);
|
||||
}
|
||||
if (null == ret.kernel) {
|
||||
println("kernel must not be empty");
|
||||
printUsage(parser);
|
||||
}
|
||||
if (null == ret.output) {
|
||||
println("output file must not be empty");
|
||||
printUsage(parser);
|
||||
}
|
||||
if (ret.board.length() > 16) {
|
||||
println("board name length must <= 16")
|
||||
printUsage(parser);
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
byte[] write_header(CArgs inArgs) {
|
||||
ByteBuffer bf = ByteBuffer.allocate(1024 * 32);
|
||||
bf.order(ByteOrder.LITTLE_ENDIAN);
|
||||
|
||||
//header start
|
||||
bf.put("ANDROID!".getBytes())
|
||||
bf.putInt((int) new File(inArgs.kernel).length());
|
||||
bf.putInt(inArgs.base + inArgs.kernel_offset)
|
||||
|
||||
if (null == inArgs.ramdisk) {
|
||||
bf.putInt(0)
|
||||
} else {
|
||||
bf.putInt((int) new File(inArgs.ramdisk).length());
|
||||
}
|
||||
|
||||
bf.putInt(inArgs.base + inArgs.ramdisk_offset)
|
||||
|
||||
if (null == inArgs.second) {
|
||||
bf.putInt(0)
|
||||
} else {
|
||||
bf.putInt((int) new File(inArgs.second).length());
|
||||
}
|
||||
|
||||
bf.putInt(inArgs.base + inArgs.second_offset)
|
||||
bf.putInt(inArgs.base + inArgs.tags_offset)
|
||||
bf.putInt(inArgs.pagesize)
|
||||
bf.putInt(0);
|
||||
bf.putInt((parse_os_version(inArgs.os_version) << 11) | parse_os_patch_level(inArgs.os_patch_level))
|
||||
|
||||
bf.put(inArgs.board.getBytes())
|
||||
bf.put(new byte[16 - inArgs.board.length()])
|
||||
bf.put(inArgs.cmdline.substring(0, Math.min(512, inArgs.cmdline.length())).getBytes())
|
||||
bf.put(new byte[512 - Math.min(512, inArgs.cmdline.length())])
|
||||
byte[] img_id = hashFile(inArgs.kernel, inArgs.ramdisk, inArgs.second)
|
||||
bf.put(img_id)
|
||||
bf.put(new byte[32 - img_id.length])
|
||||
|
||||
if (inArgs.cmdline.length() > 512) {
|
||||
bf.put(inArgs.cmdline.substring(512).getBytes())
|
||||
bf.put(new byte[1024 + 512 - inArgs.cmdline.length()])
|
||||
} else {
|
||||
bf.put(new byte[1024])
|
||||
}
|
||||
|
||||
//padding
|
||||
pad_file(bf, inArgs.pagesize)
|
||||
|
||||
//write
|
||||
FileOutputStream fos = new FileOutputStream(inArgs.output, false);
|
||||
fos.write(bf.array(), 0, bf.position())
|
||||
fos.close();
|
||||
|
||||
return img_id;
|
||||
}
|
||||
|
||||
void printUsage(OptionParser p) {
|
||||
ByteArrayOutputStream out = new ByteArrayOutputStream();
|
||||
out.write("Usage: mkbootimg <option>\n".getBytes());
|
||||
p.printHelpOn(out);
|
||||
System.out.println(out.toString());
|
||||
out.close();
|
||||
System.exit(1);
|
||||
}
|
||||
|
||||
void write_padded_file(ByteBuffer inBF, String srcFile, int padding) {
|
||||
if (null == srcFile) {
|
||||
return;
|
||||
}
|
||||
InputStream is = new FileInputStream(new File(srcFile))
|
||||
int byteRead;
|
||||
byte[] dataRead = new byte[128]
|
||||
while (true) {
|
||||
byteRead = is.read(dataRead)
|
||||
if (-1 == byteRead) {
|
||||
break;
|
||||
}
|
||||
inBF.put(dataRead, 0, byteRead);
|
||||
}
|
||||
is.close();
|
||||
pad_file(inBF, padding)
|
||||
}
|
||||
|
||||
void pad_file(ByteBuffer inBF, int padding) {
|
||||
int pad = (padding - (inBF.position() & (padding - 1))) & (padding - 1);
|
||||
inBF.put(new byte[pad]);
|
||||
}
|
||||
|
||||
void write_data(CArgs inArgs) {
|
||||
ByteBuffer bf = ByteBuffer.allocate(1024 * 1024 * 64);
|
||||
bf.order(ByteOrder.LITTLE_ENDIAN);
|
||||
|
||||
write_padded_file(bf, inArgs.kernel, inArgs.pagesize)
|
||||
write_padded_file(bf, inArgs.ramdisk, inArgs.pagesize)
|
||||
write_padded_file(bf, inArgs.second, inArgs.pagesize)
|
||||
|
||||
//write
|
||||
FileOutputStream fos = new FileOutputStream(inArgs.output, true);
|
||||
fos.write(bf.array(), 0, bf.position())
|
||||
fos.close();
|
||||
}
|
||||
|
||||
int parse_os_patch_level(String x) {
|
||||
if (null == x) {
|
||||
return 0;
|
||||
}
|
||||
int ret = 0
|
||||
Pattern pattern = Pattern.compile("^(\\d{4})-(\\d{2})-(\\d{2})")
|
||||
Matcher matcher = pattern.matcher(x)
|
||||
if (matcher.find()) {
|
||||
int y = Integer.decode(matcher.group(1)) - 2000
|
||||
int m = Integer.decode(matcher.group(2))
|
||||
// 7 bits allocated for the year, 4 bits for the month
|
||||
assert y >= 0 && y < 128
|
||||
assert m > 0 && m <= 12
|
||||
ret = (y << 4) | m
|
||||
} else {
|
||||
throw new IllegalArgumentException("invalid os_patch_level")
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
int parse_os_version(String x) {
|
||||
int ret = 0;
|
||||
if (null != x) {
|
||||
Pattern pattern = Pattern.compile("^(\\d{1,3})(?:\\.(\\d{1,3})(?:\\.(\\d{1,3}))?)?");
|
||||
Matcher m = pattern.matcher(x)
|
||||
if (m.find()) {
|
||||
int a = Integer.decode(m.group(1))
|
||||
int b = 0;
|
||||
int c = 0;
|
||||
if (m.groupCount() >= 2) {
|
||||
b = Integer.decode(m.group(2))
|
||||
}
|
||||
if (m.groupCount() == 3) {
|
||||
c = Integer.decode(m.group(3))
|
||||
}
|
||||
assert a < 128
|
||||
assert b < 128
|
||||
assert c < 128
|
||||
ret = ((a << 14) | (b << 7) | c)
|
||||
} else {
|
||||
throw new IllegalArgumentException("invalid os_version")
|
||||
}
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
void test() {
|
||||
ByteBuffer b2 = ByteBuffer.allocate(1024);
|
||||
b2.order(ByteOrder.LITTLE_ENDIAN);
|
||||
b2.putInt(Integer.MAX_VALUE); //4 bytes
|
||||
println("max: " + Integer.MAX_VALUE)
|
||||
println("min: " + Integer.MIN_VALUE)
|
||||
b2.putInt(0x11111111)
|
||||
b2.putInt(Integer.MIN_VALUE);
|
||||
b2.putInt(0x11111111)
|
||||
b2.put("welcome".getBytes())
|
||||
b2.put(new byte[5]);
|
||||
b2.putInt(0x11111111)
|
||||
b2.putInt(0);
|
||||
b2.putInt(0x11111111)
|
||||
//b2.put((byte)0);
|
||||
b2.flip();
|
||||
FileChannel fc2 = new FileOutputStream(new File("ftest"), false).getChannel();
|
||||
fc2.write(b2);
|
||||
fc2.close();
|
||||
|
||||
//ByteBuffer bf = ByteBuffer.allocate(1024 * 1024 * 50);
|
||||
//bf.order(ByteOrder.LITTLE_ENDIAN);
|
||||
//bf.flip()
|
||||
//boolean append = false;
|
||||
//FileChannel fc = new FileOutputStream(new File("f1"), append).getChannel();
|
||||
//fc.write(bf);
|
||||
//fc.close();
|
||||
//
|
||||
//FileOutputStream stream = new FileOutputStream("f2");
|
||||
//stream.write(bf.array(), 0, bf.position())
|
||||
//stream.close();
|
||||
}
|
||||
|
||||
byte[] hashFile(String... inFiles) {
|
||||
MessageDigest md = MessageDigest.getInstance("SHA1")
|
||||
|
||||
for (String item : inFiles) {
|
||||
ByteBuffer itemBF = ByteBuffer.allocate(4).order(ByteOrder.LITTLE_ENDIAN);
|
||||
if (null == item) {
|
||||
md.update(itemBF.putInt(0).array())
|
||||
} else {
|
||||
InputStream is = new FileInputStream(new File(item))
|
||||
int byteRead;
|
||||
byte[] dataRead = new byte[128]
|
||||
while (true) {
|
||||
byteRead = is.read(dataRead)
|
||||
if (-1 == byteRead) {
|
||||
break;
|
||||
}
|
||||
md.update(dataRead, 0, byteRead)
|
||||
}
|
||||
is.close();
|
||||
md.update(itemBF.putInt((int) new File(item).length()).array())
|
||||
}
|
||||
}
|
||||
|
||||
return md.digest();
|
||||
}
|
||||
|
||||
void dumpBytes(byte[] inData) {
|
||||
StringBuffer sb = new StringBuffer("");
|
||||
for (int i = 0; i < inData.length; i++) {
|
||||
sb.append(Integer.toString((inData[i] & 0xff) + 0x100, 16).substring(1));
|
||||
}
|
||||
println("0x" + sb.toString());
|
||||
}
|
||||
|
||||
void mkbootimg(CArgs theArgs) {
|
||||
byte[] img_id = write_header(theArgs)
|
||||
write_data(theArgs)
|
||||
if (theArgs.id) {
|
||||
ByteBuffer bf = ByteBuffer.allocate(32);
|
||||
bf.order(ByteOrder.LITTLE_ENDIAN);
|
||||
bf.put(img_id);
|
||||
bf.put(new byte[32 - img_id.length])
|
||||
dumpBytes(bf.array());
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,182 @@
|
||||
package cfig.bootimg
|
||||
|
||||
import groovy.json.JsonBuilder
|
||||
|
||||
import java.nio.ByteBuffer
|
||||
import java.nio.ByteOrder
|
||||
|
||||
/**
|
||||
* Created by yu at 10:58 on 2016-06-18
|
||||
*/
|
||||
class Parser {
|
||||
|
||||
int readInt(InputStream is) {
|
||||
ByteBuffer bf = ByteBuffer.allocate(128);
|
||||
bf.order(ByteOrder.LITTLE_ENDIAN);
|
||||
byte[] data4 = new byte[4];
|
||||
assert 4 == is.read(data4)
|
||||
bf.clear();
|
||||
bf.put(data4);
|
||||
bf.flip();
|
||||
return bf.getInt();
|
||||
}
|
||||
|
||||
byte[] readBytes(InputStream is, int len) {
|
||||
byte[] data4 = new byte[len];
|
||||
assert len == is.read(data4)
|
||||
return data4;
|
||||
}
|
||||
|
||||
String unparse_os_version(int x) {
|
||||
int a = x >> 14;
|
||||
int b = (x - (a << 14)) >> 7;
|
||||
int c = x & 0x7f;
|
||||
|
||||
return String.format("%d.%d.%d", a, b, c);
|
||||
}
|
||||
|
||||
String unparse_os_patch_level(int x) {
|
||||
int y = x >> 4;
|
||||
int m = x & 0xf;
|
||||
y += 2000;
|
||||
|
||||
return String.format("%d-%02d-%02d", y, m, 0);
|
||||
}
|
||||
|
||||
int get_header_len(int pagesize) {
|
||||
int pad = (pagesize - (1632 & (pagesize - 1))) & (pagesize - 1);
|
||||
return pad + 1632;
|
||||
}
|
||||
|
||||
int get_pad_len(int position, int pagesize) {
|
||||
return (pagesize - (position & (pagesize - 1))) & (pagesize - 1);
|
||||
}
|
||||
|
||||
String bytes2String(byte[] inData) {
|
||||
StringBuilder sb = new StringBuilder("");
|
||||
for (int i = 0; i < inData.length; i++) {
|
||||
sb.append(Integer.toString((inData[i] & 0xff) + 0x100, 16).substring(1));
|
||||
}
|
||||
return sb.toString();
|
||||
}
|
||||
|
||||
void parse_header(String fileName, CImgInfo inImgInfo) {
|
||||
InputStream is = new FileInputStream(new File(fileName))
|
||||
assert Arrays.equals(readBytes(is, 8), "ANDROID!".getBytes())
|
||||
inImgInfo.kernel_len = readInt(is);
|
||||
inImgInfo.kernel_offset = readInt(is);
|
||||
inImgInfo.ramdisk_len = readInt(is);
|
||||
inImgInfo.ramdisk_offset = readInt(is);
|
||||
inImgInfo.second_len = readInt(is);
|
||||
inImgInfo.second_offset = readInt(is);
|
||||
inImgInfo.tags_offset = readInt(is);
|
||||
inImgInfo.pagesize = readInt(is);
|
||||
assert 0 == readInt(is) //reserved
|
||||
int os_and_patch = readInt(is)
|
||||
if (0 != os_and_patch) { //treated as 'reserved' in this boot image
|
||||
inImgInfo.os_version = unparse_os_version(os_and_patch >> 11)
|
||||
inImgInfo.os_patch_level = unparse_os_patch_level(os_and_patch & 0x7ff)
|
||||
}
|
||||
inImgInfo.board = new String(readBytes(is, 16), "UTF-8").trim();
|
||||
inImgInfo.cmdline = new String(readBytes(is, 512), "UTF-8")
|
||||
inImgInfo.hash = readBytes(is, 32); //hash
|
||||
inImgInfo.cmdline += new String(readBytes(is, 1024), "UTF-8")
|
||||
inImgInfo.cmdline = inImgInfo.cmdline.trim();
|
||||
is.close();
|
||||
|
||||
//calc subimg positions
|
||||
inImgInfo.kernel_pos = get_header_len(inImgInfo.pagesize)
|
||||
inImgInfo.ramdisk_pos = inImgInfo.kernel_pos + inImgInfo.kernel_len + get_pad_len(inImgInfo.kernel_len, inImgInfo.pagesize)
|
||||
inImgInfo.second_pos = inImgInfo.ramdisk_pos + inImgInfo.ramdisk_len + get_pad_len(inImgInfo.ramdisk_len, inImgInfo.pagesize)
|
||||
|
||||
//adjust args
|
||||
if (inImgInfo.kernel_offset > 0x10000000) {
|
||||
inImgInfo.base = 0x10000000;
|
||||
inImgInfo.kernel_offset -= inImgInfo.base;
|
||||
inImgInfo.ramdisk_offset -= inImgInfo.base;
|
||||
inImgInfo.second_offset -= inImgInfo.base;
|
||||
inImgInfo.tags_offset -= inImgInfo.base;
|
||||
}
|
||||
}
|
||||
|
||||
void extract_img_header(CImgInfo inImgInfo) {
|
||||
JsonBuilder jb = new JsonBuilder();
|
||||
String hashString = bytes2String(inImgInfo.hash);
|
||||
jb.bootimg {
|
||||
args {
|
||||
// kernel inImgInfo.kernel;
|
||||
// ramdisk inImgInfo.ramdisk;
|
||||
// second inImgInfo.second;
|
||||
// output inImgInfo.output;
|
||||
base "0x" + Integer.toHexString(inImgInfo.base);
|
||||
kernel_offset "0x" + Integer.toHexString(inImgInfo.kernel_offset);
|
||||
ramdisk_offset "0x" + Integer.toHexString(inImgInfo.ramdisk_offset);
|
||||
second_offset "0x" + Integer.toHexString(inImgInfo.second_offset);
|
||||
tags_offset "0x" + Integer.toHexString(inImgInfo.tags_offset);
|
||||
pagesize inImgInfo.pagesize;
|
||||
board inImgInfo.board;
|
||||
cmdline inImgInfo.cmdline;
|
||||
os_version inImgInfo.os_version;
|
||||
os_patch_level inImgInfo.os_patch_level;
|
||||
id inImgInfo.id;
|
||||
}
|
||||
img {
|
||||
kernel_pos inImgInfo.kernel_pos;
|
||||
kernel_len inImgInfo.kernel_len;
|
||||
ramdisk_pos inImgInfo.ramdisk_pos;
|
||||
ramdisk_len inImgInfo.ramdisk_len;
|
||||
second_pos inImgInfo.second_pos;
|
||||
second_len inImgInfo.second_len;
|
||||
hash hashString;
|
||||
}
|
||||
}
|
||||
FileWriter fw = new FileWriter(inImgInfo.cfg);
|
||||
fw.write(jb.toPrettyString());
|
||||
fw.flush();
|
||||
fw.close();
|
||||
}
|
||||
|
||||
void extract_img_data(String inBootImg, String outImgName, int offset, int length) {
|
||||
if (0 == length) {
|
||||
return;
|
||||
}
|
||||
RandomAccessFile inRaf = new RandomAccessFile(inBootImg, "r");
|
||||
RandomAccessFile outRaf = new RandomAccessFile(outImgName, "rw");
|
||||
inRaf.seek(offset);
|
||||
byte[] data = new byte[length];
|
||||
assert length == inRaf.read(data)
|
||||
outRaf.write(data);
|
||||
outRaf.close();
|
||||
inRaf.close();
|
||||
}
|
||||
|
||||
void printUsage() {
|
||||
println("Usage: abootimg <path_to_boot_image> [work_dir]");
|
||||
System.exit(1);
|
||||
}
|
||||
|
||||
void abootimg(String[] arg) {
|
||||
CImgInfo imgInfo = new CImgInfo();
|
||||
String fileName;
|
||||
String workDir = "unzip_boot";
|
||||
if (1 == arg.length) {
|
||||
fileName = arg[0];
|
||||
} else if (2 == arg.length) {
|
||||
fileName = arg[0];
|
||||
workDir = arg[1];
|
||||
} else {
|
||||
printUsage();
|
||||
}
|
||||
imgInfo.kernel = workDir + File.separator + imgInfo.kernel;
|
||||
imgInfo.ramdisk = workDir + File.separator + imgInfo.ramdisk;
|
||||
imgInfo.second = workDir + File.separator + imgInfo.second;
|
||||
imgInfo.cfg = workDir + File.separator + imgInfo.cfg;
|
||||
|
||||
parse_header(fileName, imgInfo);
|
||||
new File(workDir).mkdirs();
|
||||
extract_img_data(fileName, imgInfo.kernel, imgInfo.kernel_pos, imgInfo.kernel_len)
|
||||
extract_img_data(fileName, imgInfo.ramdisk, imgInfo.ramdisk_pos, imgInfo.ramdisk_len)
|
||||
extract_img_data(fileName, imgInfo.second, imgInfo.second_pos, imgInfo.second_len)
|
||||
extract_img_header(imgInfo);
|
||||
}
|
||||
}
|
||||
@ -1,182 +1,8 @@
|
||||
package cfig.bootimg
|
||||
|
||||
import groovy.json.JsonBuilder
|
||||
import java.nio.ByteBuffer
|
||||
|
||||
// http://mvnrepository.com/artifact/net.sf.jopt-simple/jopt-simple
|
||||
//@Grapes(
|
||||
// @Grab(group='net.sf.jopt-simple', module='jopt-simple', version='5.0.1')
|
||||
//)
|
||||
import java.nio.ByteOrder
|
||||
|
||||
int readInt(InputStream is) {
|
||||
ByteBuffer bf = ByteBuffer.allocate(128);
|
||||
bf.order(ByteOrder.LITTLE_ENDIAN);
|
||||
byte[] data4 = new byte[4];
|
||||
assert 4 == is.read(data4)
|
||||
bf.clear();
|
||||
bf.put(data4);
|
||||
bf.flip();
|
||||
return bf.getInt();
|
||||
}
|
||||
|
||||
byte[] readBytes(InputStream is, int len) {
|
||||
byte[] data4 = new byte[len];
|
||||
assert len == is.read(data4)
|
||||
return data4;
|
||||
}
|
||||
|
||||
String unparse_os_version(int x) {
|
||||
int a = x >> 14;
|
||||
int b = (x - (a << 14)) >> 7;
|
||||
int c = x & 0x7f;
|
||||
|
||||
return String.format("%d.%d.%d", a, b, c);
|
||||
}
|
||||
|
||||
String unparse_os_patch_level(int x) {
|
||||
int y = x >> 4;
|
||||
int m = x & 0xf;
|
||||
y += 2000;
|
||||
|
||||
return String.format("%d-%d-%d", y, m, 0);
|
||||
}
|
||||
|
||||
int get_header_len(int pagesize) {
|
||||
int pad = (pagesize - (1632 & (pagesize - 1))) & (pagesize - 1);
|
||||
return pad + 1632;
|
||||
}
|
||||
|
||||
int get_pad_len(int position, int pagesize) {
|
||||
return (pagesize - (position & (pagesize - 1))) & (pagesize - 1);
|
||||
}
|
||||
|
||||
String bytes2String(byte[] inData) {
|
||||
StringBuilder sb = new StringBuilder("");
|
||||
for (int i = 0; i < inData.length; i++) {
|
||||
sb.append(Integer.toString((inData[i] & 0xff) + 0x100, 16).substring(1));
|
||||
}
|
||||
return sb.toString();
|
||||
}
|
||||
|
||||
void parse_header(String fileName, CImgInfo inImgInfo) {
|
||||
InputStream is = new FileInputStream(new File(fileName))
|
||||
assert Arrays.equals(readBytes(is, 8), "ANDROID!".getBytes())
|
||||
inImgInfo.kernel_len = readInt(is);
|
||||
inImgInfo.kernel_offset = readInt(is);
|
||||
inImgInfo.ramdisk_len = readInt(is);
|
||||
inImgInfo.ramdisk_offset = readInt(is);
|
||||
inImgInfo.second_len = readInt(is);
|
||||
inImgInfo.second_offset = readInt(is);
|
||||
inImgInfo.tags_offset = readInt(is);
|
||||
inImgInfo.pagesize = readInt(is);
|
||||
assert 0 == readInt(is) //reserved
|
||||
int os_and_patch = readInt(is)
|
||||
if (0 != os_and_patch) { //treated as 'reserved' in this boot image
|
||||
inImgInfo.os_version = unparse_os_version(os_and_patch >> 11)
|
||||
inImgInfo.os_patch_level = unparse_os_patch_level(os_and_patch & 0x7ff)
|
||||
}
|
||||
inImgInfo.board = new String(readBytes(is, 16), "UTF-8").trim();
|
||||
inImgInfo.cmdline = new String(readBytes(is, 512), "UTF-8")
|
||||
inImgInfo.hash = readBytes(is, 32); //hash
|
||||
inImgInfo.cmdline += new String(readBytes(is, 1024), "UTF-8")
|
||||
inImgInfo.cmdline = inImgInfo.cmdline.trim();
|
||||
is.close();
|
||||
|
||||
//calc subimg positions
|
||||
inImgInfo.kernel_pos = get_header_len(inImgInfo.pagesize)
|
||||
inImgInfo.ramdisk_pos = inImgInfo.kernel_pos + inImgInfo.kernel_len + get_pad_len(inImgInfo.kernel_len, inImgInfo.pagesize)
|
||||
inImgInfo.second_pos = inImgInfo.ramdisk_pos + inImgInfo.ramdisk_len + get_pad_len(inImgInfo.ramdisk_len, inImgInfo.pagesize)
|
||||
|
||||
//adjust args
|
||||
if (inImgInfo.kernel_offset > 0x10000000) {
|
||||
inImgInfo.base = 0x10000000;
|
||||
inImgInfo.kernel_offset -= inImgInfo.base;
|
||||
inImgInfo.ramdisk_offset -= inImgInfo.base;
|
||||
inImgInfo.second_offset -= inImgInfo.base;
|
||||
inImgInfo.tags_offset -= inImgInfo.base;
|
||||
}
|
||||
}
|
||||
|
||||
void extract_img_header(CImgInfo inImgInfo) {
|
||||
JsonBuilder jb = new JsonBuilder();
|
||||
hashString = bytes2String(inImgInfo.hash);
|
||||
jb.bootimg {
|
||||
args {
|
||||
// kernel inImgInfo.kernel;
|
||||
// ramdisk inImgInfo.ramdisk;
|
||||
// second inImgInfo.second;
|
||||
// output inImgInfo.output;
|
||||
base "0x" + Integer.toHexString(inImgInfo.base);
|
||||
kernel_offset "0x" + Integer.toHexString(inImgInfo.kernel_offset);
|
||||
ramdisk_offset "0x" + Integer.toHexString(inImgInfo.ramdisk_offset);
|
||||
second_offset "0x" + Integer.toHexString(inImgInfo.second_offset);
|
||||
tags_offset "0x" + Integer.toHexString(inImgInfo.tags_offset);
|
||||
pagesize inImgInfo.pagesize;
|
||||
board inImgInfo.board;
|
||||
cmdline inImgInfo.cmdline;
|
||||
os_version inImgInfo.os_version;
|
||||
os_patch_level inImgInfo.os_patch_level;
|
||||
id inImgInfo.id;
|
||||
}
|
||||
img {
|
||||
kernel_pos inImgInfo.kernel_pos;
|
||||
kernel_len inImgInfo.kernel_len;
|
||||
ramdisk_pos inImgInfo.ramdisk_pos;
|
||||
ramdisk_len inImgInfo.ramdisk_len;
|
||||
second_pos inImgInfo.second_pos;
|
||||
second_len inImgInfo.second_len;
|
||||
hash hashString;
|
||||
}
|
||||
}
|
||||
FileWriter fw = new FileWriter(inImgInfo.cfg);
|
||||
fw.write(jb.toPrettyString());
|
||||
fw.flush();
|
||||
fw.close();
|
||||
}
|
||||
|
||||
void extract_img_data(String inBootImg, String outImgName, int offset, int length) {
|
||||
if (0 == length) {
|
||||
return;
|
||||
}
|
||||
RandomAccessFile inRaf = new RandomAccessFile(inBootImg, "r");
|
||||
RandomAccessFile outRaf = new RandomAccessFile(outImgName, "rw");
|
||||
inRaf.seek(offset);
|
||||
byte[] data = new byte[length];
|
||||
assert length == inRaf.read(data)
|
||||
outRaf.write(data);
|
||||
outRaf.close();
|
||||
inRaf.close();
|
||||
}
|
||||
|
||||
void printUsage() {
|
||||
println("Usage: abootimg <path_to_boot_image> [work_dir]");
|
||||
System.exit(1);
|
||||
}
|
||||
|
||||
void abootimg(String[] arg) {
|
||||
CImgInfo imgInfo = new CImgInfo();
|
||||
String fileName;
|
||||
String workDir = "unzip_boot";
|
||||
if (1 == arg.length) {
|
||||
fileName = arg[0];
|
||||
} else if (2 == arg.length) {
|
||||
fileName = arg[0];
|
||||
workDir = arg[1];
|
||||
} else {
|
||||
printUsage();
|
||||
}
|
||||
imgInfo.kernel = workDir + File.separator + imgInfo.kernel;
|
||||
imgInfo.ramdisk = workDir + File.separator + imgInfo.ramdisk;
|
||||
imgInfo.second = workDir + File.separator + imgInfo.second;
|
||||
imgInfo.cfg = workDir + File.separator + imgInfo.cfg;
|
||||
|
||||
parse_header(fileName, imgInfo);
|
||||
new File(workDir).mkdir();
|
||||
extract_img_data(fileName, imgInfo.kernel, imgInfo.kernel_pos, imgInfo.kernel_len)
|
||||
extract_img_data(fileName, imgInfo.ramdisk, imgInfo.ramdisk_pos, imgInfo.ramdisk_len)
|
||||
extract_img_data(fileName, imgInfo.second, imgInfo.second_pos, imgInfo.second_len)
|
||||
extract_img_header(imgInfo);
|
||||
}
|
||||
|
||||
abootimg(args);
|
||||
new Parser().abootimg(args);
|
||||
|
||||
@ -0,0 +1,11 @@
|
||||
package cfig.bootimg
|
||||
|
||||
if (2 != args.length) {
|
||||
println("Usage:\n\trepack <out_file> <work_dir>");
|
||||
println("It will create <out_file> from <work_dir>/kernel, <work_dir>/ramdisk.img.gz and <work_dir>/second");
|
||||
println("Example:\n\trepack boot.img.clear build/unzip_boot");
|
||||
System.exit(1);
|
||||
}
|
||||
|
||||
CImgInfo aInfo = CImgInfo.fromJson(args[0], args[1]);
|
||||
new Packer().mkbootimg(aInfo);
|
||||
@ -0,0 +1,28 @@
|
||||
package cfig.bootimg;
|
||||
|
||||
void Run(List<String> inCmd, String inWorkdir = null) {
|
||||
println("CMD:" + inCmd)
|
||||
if (inWorkdir == null) {
|
||||
inWorkdir = ".";
|
||||
}
|
||||
ProcessBuilder pb = new ProcessBuilder(inCmd)
|
||||
.directory(new File(inWorkdir))
|
||||
.redirectErrorStream(true);
|
||||
Process p = pb.start()
|
||||
p.inputStream.eachLine {println it}
|
||||
p.waitFor();
|
||||
assert 0 == p.exitValue()
|
||||
}
|
||||
|
||||
if (3 != args.length) {
|
||||
println("Usage:\n\trepack_with_cmd <out_file> <work_dir> <mkbootimg_bin>");
|
||||
println("It will create <out_file> from <work_dir>/kernel, <work_dir>/ramdisk.img.gz and <work_dir>/second");
|
||||
println("with the program <mkbootimg_bin>");
|
||||
println("Example:\n\trepack boot.img.clear build/unzip_boot mkbootimg");
|
||||
System.exit(1);
|
||||
}
|
||||
|
||||
CImgInfo aInfo = CImgInfo.fromJson(args[0], args[1]);
|
||||
List<String> cmdArgs = aInfo.toCommandList();
|
||||
cmdArgs.add(0, args[2]);
|
||||
Run(cmdArgs);
|
||||
Loading…
Reference in New Issue