mirror of
https://github.com/moparisthebest/davmail
synced 2024-12-13 11:12:22 -05:00
Update Jsmooth patch with 64bits exe support
git-svn-id: http://svn.code.sf.net/p/davmail/code/trunk@1681 3d1905a2-6b24-0410-a738-b14d5a86fcbd
This commit is contained in:
mguessan
parent
3a058507c9
commit
b35b3db630
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,408 @@
|
|||||||
|
/*
|
||||||
|
JSmooth: a VM wrapper toolkit for Windows
|
||||||
|
Copyright (C) 2003 Rodrigo Reyes <reyes@charabia.net>
|
||||||
|
|
||||||
|
This program is free software; you can redistribute it and/or modify
|
||||||
|
it under the terms of the GNU General Public License as published by
|
||||||
|
the Free Software Foundation; either version 2 of the License, or
|
||||||
|
(at your option) any later version.
|
||||||
|
|
||||||
|
This program is distributed in the hope that it will be useful,
|
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
GNU General Public License for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU General Public License
|
||||||
|
along with this program; if not, write to the Free Software
|
||||||
|
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
||||||
|
|
||||||
|
*/
|
||||||
|
|
||||||
|
/*
|
||||||
|
* JSmoothModelBean.java
|
||||||
|
*
|
||||||
|
* Created on 7 aout 2003, 18:32
|
||||||
|
*/
|
||||||
|
|
||||||
|
package net.charabia.jsmoothgen.application;
|
||||||
|
|
||||||
|
import java.io.File;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.util.Iterator;
|
||||||
|
import java.util.Vector;
|
||||||
|
|
||||||
|
public class JSmoothModelBean {
|
||||||
|
private String m_skeletonName;
|
||||||
|
private String m_executableName;
|
||||||
|
private String m_currentDirectory;
|
||||||
|
|
||||||
|
private String m_iconLocation;
|
||||||
|
private boolean m_embedJar = false;
|
||||||
|
private String m_jarLocation;
|
||||||
|
private String m_mainClassName;
|
||||||
|
private String m_arguments;
|
||||||
|
private String[] m_classPath;
|
||||||
|
private String m_minimumVersion = "";
|
||||||
|
private String m_maximumVersion = "";
|
||||||
|
private String[] m_jvmSearch = null;
|
||||||
|
|
||||||
|
private int m_maxHeap = -1;
|
||||||
|
private int m_initialHeap = -1;
|
||||||
|
|
||||||
|
private String m_vmParamter;
|
||||||
|
|
||||||
|
private int m_uacRequireAdmin;
|
||||||
|
|
||||||
|
private String m_noJvmMessage;
|
||||||
|
private String m_noJvmURL;
|
||||||
|
|
||||||
|
private String m_bundledJVM = null;
|
||||||
|
|
||||||
|
private JavaPropertyPair[] m_javaprops = new JavaPropertyPair[0];
|
||||||
|
private JSmoothModelBean.Property[] m_skelproperties;
|
||||||
|
|
||||||
|
static public class Property {
|
||||||
|
public String Key;
|
||||||
|
public String Value;
|
||||||
|
|
||||||
|
public void setKey(String key) {
|
||||||
|
this.Key = key;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getKey() {
|
||||||
|
return this.Key;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setValue(String val) {
|
||||||
|
this.Value = val;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getValue() {
|
||||||
|
return this.Value;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String toString() {
|
||||||
|
return getKey() + "==" + getValue();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
transient Vector m_listeners = new Vector();
|
||||||
|
|
||||||
|
public static interface Listener {
|
||||||
|
public void dataChanged();
|
||||||
|
}
|
||||||
|
|
||||||
|
transient Vector m_skeletonChangedListener = new Vector();
|
||||||
|
|
||||||
|
public static interface SkeletonChangedListener {
|
||||||
|
public void skeletonChanged();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Creates a new instance of JSmoothModelBean
|
||||||
|
*/
|
||||||
|
public JSmoothModelBean() {
|
||||||
|
}
|
||||||
|
|
||||||
|
public void addListener(JSmoothModelBean.Listener l) {
|
||||||
|
m_listeners.add(l);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void removeListener(JSmoothModelBean.Listener l) {
|
||||||
|
m_listeners.remove(l);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void addSkeletonChangedListener(JSmoothModelBean.SkeletonChangedListener l) {
|
||||||
|
m_skeletonChangedListener.add(l);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void removeSkeletonChangedListener(JSmoothModelBean.SkeletonChangedListener l) {
|
||||||
|
m_skeletonChangedListener.remove(l);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
private void fireChanged() {
|
||||||
|
for (Iterator i = m_listeners.iterator(); i.hasNext();) {
|
||||||
|
JSmoothModelBean.Listener l = (JSmoothModelBean.Listener)i.next();
|
||||||
|
l.dataChanged();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void fireSkeletonChanged() {
|
||||||
|
for (Iterator i = m_skeletonChangedListener.iterator(); i.hasNext();) {
|
||||||
|
JSmoothModelBean.SkeletonChangedListener l = (JSmoothModelBean.SkeletonChangedListener)i.next();
|
||||||
|
l.skeletonChanged();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setSkeletonName(String name) {
|
||||||
|
if (name != m_skeletonName) {
|
||||||
|
m_skeletonName = name;
|
||||||
|
fireSkeletonChanged();
|
||||||
|
fireChanged();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getSkeletonName() {
|
||||||
|
return m_skeletonName;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setExecutableName(String name) {
|
||||||
|
if (name != m_executableName) {
|
||||||
|
m_executableName = name;
|
||||||
|
fireChanged();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
public void setCurrentDirectory(String curdir) {
|
||||||
|
if (curdir != m_currentDirectory) {
|
||||||
|
m_currentDirectory = curdir;
|
||||||
|
fireChanged();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getCurrentDirectory() {
|
||||||
|
return m_currentDirectory;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getExecutableName() {
|
||||||
|
return m_executableName;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setIconLocation(String name) {
|
||||||
|
if (name != m_iconLocation) {
|
||||||
|
m_iconLocation = name;
|
||||||
|
fireChanged();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getIconLocation() {
|
||||||
|
return m_iconLocation;
|
||||||
|
}
|
||||||
|
|
||||||
|
public boolean getEmbeddedJar() {
|
||||||
|
return m_embedJar;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setEmbeddedJar(boolean b) {
|
||||||
|
m_embedJar = b;
|
||||||
|
fireChanged();
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setJarLocation(String name) {
|
||||||
|
if (name != m_jarLocation) {
|
||||||
|
m_jarLocation = name;
|
||||||
|
fireChanged();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getJarLocation() {
|
||||||
|
return m_jarLocation;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
public void setMainClassName(String name) {
|
||||||
|
if (name != m_mainClassName) {
|
||||||
|
m_mainClassName = name;
|
||||||
|
fireChanged();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getMainClassName() {
|
||||||
|
return m_mainClassName;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setArguments(String args) {
|
||||||
|
m_arguments = args;
|
||||||
|
fireChanged();
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getArguments() {
|
||||||
|
return m_arguments;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setClassPath(String[] cp) {
|
||||||
|
m_classPath = cp;
|
||||||
|
fireChanged();
|
||||||
|
}
|
||||||
|
|
||||||
|
public String[] getClassPath() {
|
||||||
|
return m_classPath;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setMaximumVersion(String version) {
|
||||||
|
m_maximumVersion = version;
|
||||||
|
fireChanged();
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getMaximumVersion() {
|
||||||
|
return m_maximumVersion;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setMinimumVersion(String version) {
|
||||||
|
m_minimumVersion = version;
|
||||||
|
fireChanged();
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getMinimumVersion() {
|
||||||
|
return m_minimumVersion;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setJVMSearchPath(String[] path) {
|
||||||
|
m_jvmSearch = path;
|
||||||
|
fireChanged();
|
||||||
|
}
|
||||||
|
|
||||||
|
public String[] getJVMSearchPath() {
|
||||||
|
return m_jvmSearch;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setSkeletonProperties(JSmoothModelBean.Property[] props) {
|
||||||
|
// for (int i=0; i<props.length; i++)
|
||||||
|
// System.out.println("SET PROPERTY: " + props[i].getIdName() + "=" + props[i].getValue());
|
||||||
|
|
||||||
|
m_skelproperties = props;
|
||||||
|
fireChanged();
|
||||||
|
}
|
||||||
|
|
||||||
|
public JSmoothModelBean.Property[] getSkeletonProperties() {
|
||||||
|
return m_skelproperties;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setNoJvmMessage(String msg) {
|
||||||
|
m_noJvmMessage = msg;
|
||||||
|
fireChanged();
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getNoJvmMessage() {
|
||||||
|
return m_noJvmMessage;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setNoJvmURL(String url) {
|
||||||
|
m_noJvmURL = url;
|
||||||
|
fireChanged();
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getNoJvmURL() {
|
||||||
|
return m_noJvmURL;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getBundledJVMPath() {
|
||||||
|
return m_bundledJVM;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setBundledJVMPath(String path) {
|
||||||
|
m_bundledJVM = path;
|
||||||
|
fireChanged();
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setJavaProperties(JavaPropertyPair[] pairs) {
|
||||||
|
m_javaprops = pairs;
|
||||||
|
}
|
||||||
|
|
||||||
|
public JavaPropertyPair[] getJavaProperties() {
|
||||||
|
return m_javaprops;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setMaximumMemoryHeap(int size) {
|
||||||
|
m_maxHeap = size;
|
||||||
|
}
|
||||||
|
|
||||||
|
public int getMaximumMemoryHeap() {
|
||||||
|
return m_maxHeap;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setInitialMemoryHeap(int size) {
|
||||||
|
m_initialHeap = size;
|
||||||
|
}
|
||||||
|
|
||||||
|
public int getInitialMemoryHeap() {
|
||||||
|
return m_initialHeap;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setVmParameter(String parameter) {
|
||||||
|
m_vmParamter = parameter;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getVmParameter() {
|
||||||
|
return m_vmParamter;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setUacRequireAdministrator(int parameter) {
|
||||||
|
m_uacRequireAdmin = parameter;
|
||||||
|
}
|
||||||
|
|
||||||
|
public int getUacRequireAdministrator() {
|
||||||
|
return m_uacRequireAdmin;
|
||||||
|
}
|
||||||
|
|
||||||
|
public boolean isUacRequireAdmin() {
|
||||||
|
return m_uacRequireAdmin > 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String[] normalizePaths(java.io.File filebase) {
|
||||||
|
return normalizePaths(filebase, true);
|
||||||
|
}
|
||||||
|
|
||||||
|
public String[] normalizePaths(java.io.File filebase, boolean toRelativePath) {
|
||||||
|
// System.out.println("Normalize Path " + filebase + " / " + toRelativePath);
|
||||||
|
Vector result = new Vector();
|
||||||
|
|
||||||
|
m_iconLocation = checkRelativePath(filebase, m_iconLocation, result, "Icon location", toRelativePath);
|
||||||
|
m_jarLocation = checkRelativePath(filebase, m_jarLocation, result, "Jar location", toRelativePath);
|
||||||
|
m_bundledJVM = checkRelativePath(filebase, m_bundledJVM, result, "Bundle JVM location", toRelativePath);
|
||||||
|
m_executableName = checkRelativePath(filebase, m_executableName, result, "Executable location", toRelativePath);
|
||||||
|
|
||||||
|
if (m_executableName != null) {
|
||||||
|
File exebase = new File(m_executableName);
|
||||||
|
if (exebase.isAbsolute() == false)
|
||||||
|
exebase = new File(filebase, exebase.toString()).getParentFile();
|
||||||
|
|
||||||
|
// System.out.println("EXE FILEBASE: " + exebase.toString());
|
||||||
|
if ((m_currentDirectory != null) && (m_currentDirectory.indexOf("${") >= 0))
|
||||||
|
m_currentDirectory = checkRelativePath(exebase, m_currentDirectory, result, "Current directory", toRelativePath);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (m_classPath != null) {
|
||||||
|
for (int i = 0; i < m_classPath.length; i++) {
|
||||||
|
m_classPath[i] = checkRelativePath(filebase, m_classPath[i], result, "Classpath entry (" + i + ")", toRelativePath);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (result.size() == 0)
|
||||||
|
return null;
|
||||||
|
|
||||||
|
String[] res = new String[result.size()];
|
||||||
|
result.toArray(res);
|
||||||
|
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
|
||||||
|
private String checkRelativePath(java.io.File root, String value, java.util.Vector errors, String name, boolean toRelativePath) {
|
||||||
|
if (value == null)
|
||||||
|
return value;
|
||||||
|
|
||||||
|
if (toRelativePath) {
|
||||||
|
File nf = JSmoothModelPersistency.makePathRelativeIfPossible(root, new File(value));
|
||||||
|
if (nf.isAbsolute()) {
|
||||||
|
errors.add(name);
|
||||||
|
}
|
||||||
|
return nf.toString();
|
||||||
|
} else {
|
||||||
|
File nf = new File(value);
|
||||||
|
if (nf.isAbsolute() == false) {
|
||||||
|
nf = new File(root, value);
|
||||||
|
nf = nf.getAbsoluteFile();
|
||||||
|
|
||||||
|
try {
|
||||||
|
nf = nf.getCanonicalFile();
|
||||||
|
nf = nf.getAbsoluteFile();
|
||||||
|
} catch (IOException iox) {
|
||||||
|
// do nothing
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nf.toString();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,250 @@
|
|||||||
|
/*
|
||||||
|
JSmooth: a VM wrapper toolkit for Windows
|
||||||
|
Copyright (C) 2003 Rodrigo Reyes <reyes@charabia.net>
|
||||||
|
|
||||||
|
This program is free software; you can redistribute it and/or modify
|
||||||
|
it under the terms of the GNU General Public License as published by
|
||||||
|
the Free Software Foundation; either version 2 of the License, or
|
||||||
|
(at your option) any later version.
|
||||||
|
|
||||||
|
This program is distributed in the hope that it will be useful,
|
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
GNU General Public License for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU General Public License
|
||||||
|
along with this program; if not, write to the Free Software
|
||||||
|
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
||||||
|
|
||||||
|
*/
|
||||||
|
|
||||||
|
package net.charabia.jsmoothgen.application;
|
||||||
|
|
||||||
|
import java.io.File;
|
||||||
|
|
||||||
|
import net.charabia.jsmoothgen.skeleton.SkeletonBean;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author Rodrigo
|
||||||
|
*/
|
||||||
|
public class PropertiesBuilder {
|
||||||
|
/**
|
||||||
|
* Creates a text containing all the relevant properties of a
|
||||||
|
* JSmoothModelBean object. The properties are output in the form
|
||||||
|
* "key=value".
|
||||||
|
* <p/>
|
||||||
|
* <p/>
|
||||||
|
* Note that all the paths are converted to be made relative to
|
||||||
|
* the basedir parameter provided. All the paths converted are
|
||||||
|
* expected to be relative to the targetted executable binary
|
||||||
|
* (before the conversion is applied, that is).
|
||||||
|
*/
|
||||||
|
static public String makeProperties(File basedir, JSmoothModelBean obj) {
|
||||||
|
StringBuffer out = new StringBuffer();
|
||||||
|
|
||||||
|
addPair("arguments", obj.getArguments(), out);
|
||||||
|
addPair("mainclassname", obj.getMainClassName(), out);
|
||||||
|
addPair("jvmsearch", makePathConc(obj.getJVMSearchPath()), out);
|
||||||
|
addPair("minversion", obj.getMinimumVersion(), out);
|
||||||
|
addPair("maxversion", obj.getMaximumVersion(), out);
|
||||||
|
|
||||||
|
addPair("currentdir", obj.getCurrentDirectory(), out);
|
||||||
|
|
||||||
|
if (obj.getEmbeddedJar() && (obj.getJarLocation().trim().length() > 0)) {
|
||||||
|
addPair("embedjar", "true", out);
|
||||||
|
} else {
|
||||||
|
addPair("embedjar", "false", out);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (obj.getMaximumMemoryHeap() > 1) {
|
||||||
|
addPair("maxheap", Integer.toString(obj.getMaximumMemoryHeap()), out);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (obj.getInitialMemoryHeap() > 1) {
|
||||||
|
addPair("initialheap", Integer.toString(obj.getInitialMemoryHeap()), out);
|
||||||
|
}
|
||||||
|
|
||||||
|
if ((obj.getVmParameter() != null) && (!obj.getVmParameter().isEmpty())) {
|
||||||
|
addPair("vmparameter", obj.getVmParameter(), out);
|
||||||
|
}
|
||||||
|
// BundledVM & classpaths are changed to be accessible
|
||||||
|
// from the current directory
|
||||||
|
File curdir = new File(obj.getExecutableName()).getParentFile();
|
||||||
|
|
||||||
|
if (curdir == null)
|
||||||
|
curdir = basedir.getAbsoluteFile();
|
||||||
|
|
||||||
|
if (curdir.isAbsolute() == false) {
|
||||||
|
curdir = new File(basedir, curdir.toString());
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
// System.out.println("... curdir1 : " + curdir.toString());
|
||||||
|
|
||||||
|
if (obj.getCurrentDirectory() != null) {
|
||||||
|
File newcurdir = new File(obj.getCurrentDirectory());
|
||||||
|
// System.out.println("... curdir1.5 : " + obj.getCurrentDirectory());
|
||||||
|
|
||||||
|
if (!"${EXECUTABLEPATH}".equalsIgnoreCase(obj.getCurrentDirectory())) {
|
||||||
|
if (newcurdir.isAbsolute() == false) {
|
||||||
|
curdir = new File(curdir, newcurdir.toString());
|
||||||
|
} else
|
||||||
|
curdir = newcurdir;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// System.out.println("... curdir2 : " + curdir.toString());
|
||||||
|
|
||||||
|
if (obj.getBundledJVMPath() != null)
|
||||||
|
addPair("bundledvm", getRenormalizedPathIfNeeded(obj.getBundledJVMPath(), basedir, curdir), out);
|
||||||
|
|
||||||
|
if (obj.getClassPath() != null) {
|
||||||
|
String[] relcp = new String[obj.getClassPath().length];
|
||||||
|
for (int i = 0; i < relcp.length; i++) {
|
||||||
|
relcp[i] = getRenormalizedPathIfNeeded(obj.getClassPath()[i], basedir, curdir);
|
||||||
|
}
|
||||||
|
addPair("classpath", makePathConc(relcp), out);
|
||||||
|
}
|
||||||
|
|
||||||
|
//
|
||||||
|
// Adds all the skeleton-specific properties
|
||||||
|
//
|
||||||
|
if (obj.getSkeletonProperties() != null) {
|
||||||
|
for (int i = 0; i < obj.getSkeletonProperties().length; i++) {
|
||||||
|
JSmoothModelBean.Property prop = obj.getSkeletonProperties()[i];
|
||||||
|
if (prop.getKey() != null) {
|
||||||
|
String val = prop.getValue();
|
||||||
|
if (val == null)
|
||||||
|
val = "";
|
||||||
|
addPair("skel_" + prop.getKey(), val, out);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
//
|
||||||
|
// Adds all the java properties. Those properties are
|
||||||
|
// typically passed as -Dname=value arguments for the sun's
|
||||||
|
// JVM.
|
||||||
|
//
|
||||||
|
|
||||||
|
JavaPropertyPair[] javapairs = obj.getJavaProperties();
|
||||||
|
if (javapairs != null) {
|
||||||
|
addPair("javapropertiescount", new Integer(javapairs.length).toString(), out);
|
||||||
|
for (int i = 0; i < javapairs.length; i++) {
|
||||||
|
addPair("javaproperty_name_" + i, javapairs[i].getName(), out);
|
||||||
|
addPair("javaproperty_value_" + i, javapairs[i].getValue(), out);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return out.toString();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Create a manifest entry for windows right elevation
|
||||||
|
*
|
||||||
|
* @param data
|
||||||
|
* @param skel
|
||||||
|
* @return the manifest xml
|
||||||
|
*/
|
||||||
|
public static String makeManifest(JSmoothModelBean data, SkeletonBean skel) {
|
||||||
|
StringBuilder retVal = new StringBuilder();
|
||||||
|
String platform = "x86";
|
||||||
|
String shortName = skel.getShortName();
|
||||||
|
if (shortName.contains("64")) {
|
||||||
|
platform = "ia64";
|
||||||
|
}
|
||||||
|
retVal.append("<?xml version=\"1.0\" encoding=\"utf-8\" standalone=\"yes\"?>").append("\n");
|
||||||
|
retVal.append("<assembly xmlns=\"urn:schemas-microsoft-com:asm.v1\" manifestVersion=\"1.0\">").append("\n");
|
||||||
|
retVal.append(" <assemblyIdentity type=\"win32\"").append("\n");
|
||||||
|
retVal.append(" name=\"").append(shortName).append("\"").append("\n");
|
||||||
|
retVal.append(" version=\"1.0.0.0\"").append("\n");
|
||||||
|
retVal.append(" processorArchitecture=\"").append(platform).append("\"").append("\n");
|
||||||
|
retVal.append(" />").append("\n");
|
||||||
|
retVal.append(" <dependency>").append("\n");
|
||||||
|
retVal.append(" <dependentAssembly>").append("\n");
|
||||||
|
retVal
|
||||||
|
.append(
|
||||||
|
" <assemblyIdentity type=\"win32\" name=\"Microsoft.Windows.Common-Controls\" version=\"6.0.0.0\" processorArchitecture=\"*\" publicKeyToken=\"6595b64144ccf1df\" language=\"*\"/>")
|
||||||
|
.append("\n");
|
||||||
|
retVal.append(" </dependentAssembly>").append("\n");
|
||||||
|
retVal.append(" </dependency>").append("\n");
|
||||||
|
if (data.isUacRequireAdmin()) {
|
||||||
|
retVal.append(" <trustInfo xmlns=\"urn:schemas-microsoft-com:asm.v2\">").append("\n");
|
||||||
|
retVal.append(" <security>").append("\n");
|
||||||
|
retVal.append(" <requestedPrivileges>").append("\n");
|
||||||
|
retVal.append(" <requestedExecutionLevel level=\"requireAdministrator\"/>").append("\n");
|
||||||
|
retVal.append(" </requestedPrivileges>").append("\n");
|
||||||
|
retVal.append(" </security>").append("\n");
|
||||||
|
retVal.append(" </trustInfo>").append("\n");
|
||||||
|
}
|
||||||
|
retVal.append("</assembly>");
|
||||||
|
return retVal.toString();
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Converts a path relative to previousbasedir into a path
|
||||||
|
* relative to newbasedir.
|
||||||
|
*/
|
||||||
|
static public String getRenormalizedPathIfNeeded(String value, File previousbasedir, File newbasedir) {
|
||||||
|
// File f = new File(value);
|
||||||
|
// if (f.isAbsolute())
|
||||||
|
// return value;
|
||||||
|
|
||||||
|
if (newbasedir == null)
|
||||||
|
return value;
|
||||||
|
|
||||||
|
if (value == null)
|
||||||
|
return "";
|
||||||
|
|
||||||
|
File abs = new File(previousbasedir, value).getAbsoluteFile();
|
||||||
|
File n = JSmoothModelPersistency.makePathRelativeIfPossible(newbasedir, abs);
|
||||||
|
|
||||||
|
return n.toString();
|
||||||
|
}
|
||||||
|
|
||||||
|
static public String escapeString(String str) {
|
||||||
|
if (str == null)
|
||||||
|
return "";
|
||||||
|
|
||||||
|
StringBuffer out = new StringBuffer();
|
||||||
|
for (int i = 0; i < str.length(); i++) {
|
||||||
|
char c = str.charAt(i);
|
||||||
|
switch (c) {
|
||||||
|
case '\n':
|
||||||
|
out.append("\\n");
|
||||||
|
break;
|
||||||
|
case '\t':
|
||||||
|
out.append("\\t");
|
||||||
|
break;
|
||||||
|
case '\r':
|
||||||
|
out.append("\\r");
|
||||||
|
break;
|
||||||
|
case '\\':
|
||||||
|
out.append("\\\\");
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
out.append(c);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return out.toString();
|
||||||
|
}
|
||||||
|
|
||||||
|
static private void addPair(String name, String value, StringBuffer out) {
|
||||||
|
out.append(escapeString(name));
|
||||||
|
out.append("=");
|
||||||
|
out.append(escapeString(value));
|
||||||
|
out.append("\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
static public String makePathConc(String[] elements) {
|
||||||
|
StringBuffer buf = new StringBuffer();
|
||||||
|
if (elements != null)
|
||||||
|
for (int i = 0; i < elements.length; i++) {
|
||||||
|
buf.append(elements[i]);
|
||||||
|
if ((i + 1) < elements.length)
|
||||||
|
buf.append(";");
|
||||||
|
}
|
||||||
|
return buf.toString();
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
@ -0,0 +1,126 @@
|
|||||||
|
/*
|
||||||
|
JSmooth: a VM wrapper toolkit for Windows
|
||||||
|
Copyright (C) 2003 Rodrigo Reyes <reyes@charabia.net>
|
||||||
|
|
||||||
|
This program is free software; you can redistribute it and/or modify
|
||||||
|
it under the terms of the GNU General Public License as published by
|
||||||
|
the Free Software Foundation; either version 2 of the License, or
|
||||||
|
(at your option) any later version.
|
||||||
|
|
||||||
|
This program is distributed in the hope that it will be useful,
|
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
GNU General Public License for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU General Public License
|
||||||
|
along with this program; if not, write to the Free Software
|
||||||
|
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
||||||
|
|
||||||
|
*/
|
||||||
|
|
||||||
|
package net.charabia.jsmoothgen.application.gui.util;
|
||||||
|
|
||||||
|
import java.awt.*;
|
||||||
|
import java.awt.event.*;
|
||||||
|
import javax.swing.*;
|
||||||
|
import javax.swing.border.*;
|
||||||
|
import java.io.*;
|
||||||
|
import java.util.*;
|
||||||
|
import javax.swing.text.html.*;
|
||||||
|
import javax.swing.event.*;
|
||||||
|
import java.net.*;
|
||||||
|
|
||||||
|
/**
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
|
||||||
|
public class HTMLPane extends JPanel
|
||||||
|
{
|
||||||
|
private JScrollPane m_scroller;
|
||||||
|
private JEditorPane m_html;
|
||||||
|
private URL m_baseurl;
|
||||||
|
|
||||||
|
edu.stanford.ejalbert.BrowserLauncher m_launcher;
|
||||||
|
|
||||||
|
class Hyperactive implements HyperlinkListener {
|
||||||
|
|
||||||
|
public void hyperlinkUpdate(HyperlinkEvent e) {
|
||||||
|
if (e.getEventType() == HyperlinkEvent.EventType.ACTIVATED) {
|
||||||
|
|
||||||
|
JEditorPane pane = (JEditorPane)e.getSource();
|
||||||
|
if (e instanceof HTMLFrameHyperlinkEvent) {
|
||||||
|
HTMLFrameHyperlinkEvent evt = (HTMLFrameHyperlinkEvent)e;
|
||||||
|
HTMLDocument doc = (HTMLDocument)pane.getDocument();
|
||||||
|
doc.processHTMLFrameHyperlinkEvent(evt);
|
||||||
|
} else {
|
||||||
|
try {
|
||||||
|
URL nurl = e.getURL();
|
||||||
|
if (nurl == null)
|
||||||
|
nurl = new URL(m_baseurl, e.getDescription());
|
||||||
|
if (jsmooth.Native.isAvailable())
|
||||||
|
{
|
||||||
|
jsmooth.Native.shellExecute(jsmooth.Native.SHELLEXECUTE_OPEN, nurl.toString(), null, null, jsmooth.Native.SW_NORMAL);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
m_launcher.openURLinBrowser(nurl.toExternalForm());
|
||||||
|
} catch (Throwable t) {
|
||||||
|
t.printStackTrace();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
public HTMLPane()
|
||||||
|
{
|
||||||
|
try {
|
||||||
|
m_baseurl = new File(".").toURI().toURL();
|
||||||
|
} catch (Exception ex) { ex.printStackTrace(); }
|
||||||
|
m_html = new JEditorPane("text/html","<html></html>") {
|
||||||
|
public boolean getScrollableTracksViewportWidth()
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
HTMLEditorKit hek = new HTMLEditorKit();
|
||||||
|
m_html.setEditorKit(hek);
|
||||||
|
|
||||||
|
m_scroller = new JScrollPane(m_html);
|
||||||
|
setLayout(new BorderLayout());
|
||||||
|
m_html.setEditable(false);
|
||||||
|
add(m_scroller, BorderLayout.CENTER);
|
||||||
|
// add(m_html, BorderLayout.CENTER);
|
||||||
|
m_html.addHyperlinkListener(new Hyperactive());
|
||||||
|
|
||||||
|
try {
|
||||||
|
m_launcher = new edu.stanford.ejalbert.BrowserLauncher();
|
||||||
|
}catch (Exception ex)
|
||||||
|
{
|
||||||
|
ex.printStackTrace();
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
public java.awt.Dimension getPreferredSize()
|
||||||
|
{
|
||||||
|
return new java.awt.Dimension(200,200);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setPage(URL url)
|
||||||
|
{
|
||||||
|
try {
|
||||||
|
URL u = new URL(m_baseurl, url.toExternalForm());
|
||||||
|
m_html.setPage(u);
|
||||||
|
} catch (Exception ex) { ex.printStackTrace(); }
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setText(String s)
|
||||||
|
{
|
||||||
|
m_html.setContentType("text/html");
|
||||||
|
m_html.setText(s);
|
||||||
|
m_html.setCaretPosition(0);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
}
|
||||||
478
jsmooth-0.9.9-7-patch/src/net/charabia/jsmoothgen/pe/PEFile.java
Normal file
478
jsmooth-0.9.9-7-patch/src/net/charabia/jsmoothgen/pe/PEFile.java
Normal file
@ -0,0 +1,478 @@
|
|||||||
|
/*
|
||||||
|
JSmooth: a VM wrapper toolkit for Windows
|
||||||
|
Copyright (C) 2003 Rodrigo Reyes <reyes@charabia.net>
|
||||||
|
|
||||||
|
This program is free software; you can redistribute it and/or modify
|
||||||
|
it under the terms of the GNU General Public License as published by
|
||||||
|
the Free Software Foundation; either version 2 of the License, or
|
||||||
|
(at your option) any later version.
|
||||||
|
|
||||||
|
This program is distributed in the hope that it will be useful,
|
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
GNU General Public License for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU General Public License
|
||||||
|
along with this program; if not, write to the Free Software
|
||||||
|
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
||||||
|
|
||||||
|
*/
|
||||||
|
|
||||||
|
/*
|
||||||
|
* PEFile.java
|
||||||
|
*
|
||||||
|
* Created on 28 juillet 2003, 21:28
|
||||||
|
*/
|
||||||
|
|
||||||
|
package net.charabia.jsmoothgen.pe;
|
||||||
|
|
||||||
|
import java.io.File;
|
||||||
|
import java.io.FileInputStream;
|
||||||
|
import java.io.FileNotFoundException;
|
||||||
|
import java.io.FileOutputStream;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.nio.ByteBuffer;
|
||||||
|
import java.nio.channels.FileChannel;
|
||||||
|
import java.util.Vector;
|
||||||
|
|
||||||
|
import net.charabia.jsmoothgen.pe.res.ResIcon;
|
||||||
|
import net.charabia.jsmoothgen.pe.res.ResIconDir;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author Rodrigo
|
||||||
|
*/
|
||||||
|
public class PEFile {
|
||||||
|
private File m_file;
|
||||||
|
private FileInputStream m_in = null;
|
||||||
|
private FileChannel m_channel = null;
|
||||||
|
|
||||||
|
private PEOldMSHeader m_oldmsheader;
|
||||||
|
private PEHeader m_header;
|
||||||
|
private Vector m_sections = new Vector();
|
||||||
|
|
||||||
|
private PEResourceDirectory m_resourceDir;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Creates a new instance of PEFile
|
||||||
|
*/
|
||||||
|
public PEFile(File f) {
|
||||||
|
m_file = f;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void close() throws IOException {
|
||||||
|
m_in.close();
|
||||||
|
}
|
||||||
|
|
||||||
|
public void open() throws FileNotFoundException, IOException {
|
||||||
|
m_in = new FileInputStream(m_file);
|
||||||
|
m_channel = m_in.getChannel();
|
||||||
|
|
||||||
|
m_oldmsheader = new PEOldMSHeader(this);
|
||||||
|
|
||||||
|
m_oldmsheader.read();
|
||||||
|
// m_oldmsheader.dump(System.out);
|
||||||
|
long headoffset = m_oldmsheader.e_lfanew;
|
||||||
|
|
||||||
|
m_header = new PEHeader(this, headoffset);
|
||||||
|
m_header.read();
|
||||||
|
// m_header.dump(System.out);
|
||||||
|
|
||||||
|
int seccount = m_header.NumberOfSections;
|
||||||
|
// System.out.println("LOADING " + seccount + " sections...");
|
||||||
|
|
||||||
|
long offset = headoffset + (m_header.NumberOfRvaAndSizes * 8) + 24 + getPeHeaderOffset();
|
||||||
|
|
||||||
|
for (int i = 0; i < seccount; i++) {
|
||||||
|
// System.out.println("Offset: " + offset + " (" + this.m_channel.position());
|
||||||
|
|
||||||
|
PESection sect = new PESection(this, offset);
|
||||||
|
sect.read();
|
||||||
|
// sect.dump(System.out);
|
||||||
|
m_sections.add(sect);
|
||||||
|
offset += 40;
|
||||||
|
}
|
||||||
|
// System.out.println("After sections: " + this.m_channel.position() + " (" + offset + ")");
|
||||||
|
|
||||||
|
ByteBuffer resbuf = null;
|
||||||
|
long resourceoffset = m_header.ResourceDirectory_VA;
|
||||||
|
for (int i = 0; i < seccount; i++) {
|
||||||
|
PESection sect = (PESection)m_sections.get(i);
|
||||||
|
if (sect.VirtualAddress == resourceoffset) {
|
||||||
|
// System.out.println(" Resource section found: " + resourceoffset);
|
||||||
|
PEResourceDirectory prd = new PEResourceDirectory(this, sect);
|
||||||
|
resbuf = prd.buildResource(sect.VirtualAddress);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private int getPeHeaderOffset() {
|
||||||
|
int pe32Offset = 96;
|
||||||
|
if (m_header.isPe32Plus()) {
|
||||||
|
// It is a pe32+ header (x64)
|
||||||
|
pe32Offset = 112;
|
||||||
|
}
|
||||||
|
return pe32Offset;
|
||||||
|
}
|
||||||
|
|
||||||
|
public FileChannel getChannel() {
|
||||||
|
return m_channel;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void main(String args[]) throws IOException, CloneNotSupportedException, Exception {
|
||||||
|
// (no)PEFile pe = new PEFile(new File("F:/Program Files/LAN Search PRO/lansearch.exe"));
|
||||||
|
|
||||||
|
PEFile pe = new PEFile(new File("c:/scratch/rc3.exe"));
|
||||||
|
// PEFile pe = new PEFile(new File("c:/projects/jwrap/Copie.exe"));
|
||||||
|
// PEFile pe = new PEFile(new File("c:/projects/jwrap/test.exe"));
|
||||||
|
// PEFile pe = new PEFile(new File("F:/Program Files/bQuery/bQuery.exe"));
|
||||||
|
// PEFile pe = new PEFile(new File("F:/Program Files/Server Query/query.exe"));
|
||||||
|
// PEFile pe = new PEFile(new File("F:/Program Files/AvRack/rtlrack.exe"));
|
||||||
|
pe.open();
|
||||||
|
System.out.println("OldMSHeader");
|
||||||
|
pe.m_oldmsheader.dump(System.out);
|
||||||
|
System.out.println("COFFHeader");
|
||||||
|
pe.m_header.dump(System.out);
|
||||||
|
|
||||||
|
// System.out.println("===============\nADDING A RES");
|
||||||
|
// File fout = new File("F:/Documents and Settings/Rodrigo/Mes documents/projects/jsmooth/skeletons/simplewrap/gen-application.jar");
|
||||||
|
// FileInputStream fis = new FileInputStream(fout);
|
||||||
|
//
|
||||||
|
// ByteBuffer data = ByteBuffer.allocate((int)fout.length());
|
||||||
|
// data.order(ByteOrder.LITTLE_ENDIAN);
|
||||||
|
// FileChannel fischan = fis.getChannel();
|
||||||
|
// fischan.read(data);
|
||||||
|
// data.position(0);
|
||||||
|
// fis.close();
|
||||||
|
|
||||||
|
PEResourceDirectory resdir = pe.getResourceDirectory();
|
||||||
|
System.out.println("ResourceDirectory");
|
||||||
|
resdir.dump(System.out);
|
||||||
|
|
||||||
|
// DataEntry inputResData = resdir.getData("JAVA", "#" + String.valueOf(103), "#" + String.valueOf(1033));
|
||||||
|
// ByteBuffer inputResDataBuffer = ByteBuffer.allocate(inputResData.diskSize() + 1024);
|
||||||
|
// inputResDataBuffer.order(ByteOrder.LITTLE_ENDIAN);
|
||||||
|
// inputResData.buildBuffer(inputResDataBuffer, 0, 0);
|
||||||
|
// int inputResDataBufferSize = inputResDataBuffer.position();
|
||||||
|
// inputResDataBuffer.flip();
|
||||||
|
// int offset = inputResDataBuffer.getInt();
|
||||||
|
// inputResDataBuffer.position(offset);
|
||||||
|
// StringBuilder inputResDataString = new StringBuilder(inputResDataBufferSize);
|
||||||
|
// while (inputResDataBuffer.position() <= inputResDataBufferSize - 2) {
|
||||||
|
// byte dummyByte = inputResDataBuffer.get();
|
||||||
|
// inputResDataString.append((char)dummyByte);
|
||||||
|
// }
|
||||||
|
// // Modify the data...
|
||||||
|
// String newInputResDataString = inputResDataString.toString();
|
||||||
|
// newInputResDataString = newInputResDataString.replace("samplejar", "ThisIsMyJarAndOnlyMine");
|
||||||
|
//
|
||||||
|
// inputResDataBuffer = ByteBuffer.allocate(newInputResDataString.length() + 2);
|
||||||
|
// for (int index = 0; index < newInputResDataString.length(); index++) { // C- do not change because buffer can be modified during loop
|
||||||
|
// inputResDataBuffer.put((byte)newInputResDataString.charAt(index));
|
||||||
|
// }
|
||||||
|
// inputResDataBuffer.put((byte)0);
|
||||||
|
// inputResDataBuffer.put((byte)0);
|
||||||
|
// inputResDataBuffer.position(0);
|
||||||
|
//
|
||||||
|
// boolean resb = resdir.replaceResource("JAVA", 102, 1033, inputResDataBuffer);
|
||||||
|
|
||||||
|
// PEResourceDirectory.DataEntry entry = resdir.getData("#14", "A", "#1033");
|
||||||
|
// entry.Data.position(0);
|
||||||
|
// System.out.println("DataEntry found : " + entry + " (size=" + entry.Data.remaining() + ")");
|
||||||
|
// entry.Data.position(0);
|
||||||
|
//
|
||||||
|
// ResIconDir rid = new ResIconDir(entry.Data);
|
||||||
|
// System.out.println("ResIconDir :");
|
||||||
|
// System.out.println(rid.toString());
|
||||||
|
// int iconid = rid.getEntries()[0].dwImageOffset;
|
||||||
|
// System.out.println("Icon Index: " + iconid);
|
||||||
|
//
|
||||||
|
// PEResourceDirectory.DataEntry iconentry = resdir.getData("#3", "#"+iconid, "#1033");
|
||||||
|
// iconentry.Data.position(0);
|
||||||
|
// ResIcon icon = new ResIcon(iconentry.Data);
|
||||||
|
// System.out.println("Icon :");
|
||||||
|
// System.out.println(icon.toString());
|
||||||
|
|
||||||
|
// java.awt.Image img = java.awt.Toolkit.getDefaultToolkit().getImage ("c:\\test.gif");
|
||||||
|
// java.awt.Image img = java.awt.Toolkit.getDefaultToolkit().getImage ("c:\\gnome-day2.png");
|
||||||
|
// java.awt.Image img = java.awt.Toolkit.getDefaultToolkit().getImage("c:\\gnome-color-browser2.png");
|
||||||
|
//
|
||||||
|
// java.awt.MediaTracker mt = new java.awt.MediaTracker(new javax.swing.JLabel("toto"));
|
||||||
|
// mt.addImage(img, 1);
|
||||||
|
// try {
|
||||||
|
// mt.waitForAll();
|
||||||
|
// } catch (Exception exc) {
|
||||||
|
// exc.printStackTrace();
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// ResIcon newicon = new ResIcon(img);
|
||||||
|
//
|
||||||
|
// pe.replaceDefaultIcon(newicon);
|
||||||
|
|
||||||
|
// System.out.println("-----------------\nNEW ICON:");
|
||||||
|
// System.out.println(newicon.toString());
|
||||||
|
//
|
||||||
|
// rid.getEntries()[0].bWidth = (short)newicon.Width;
|
||||||
|
// rid.getEntries()[0].bHeight = (short)(newicon.Height/2);
|
||||||
|
// rid.getEntries()[0].bColorCount = (short)(1 <<newicon.BitsPerPixel);
|
||||||
|
// rid.getEntries()[0].wBitCount = newicon.BitsPerPixel;
|
||||||
|
// rid.getEntries()[0].dwBytesInRes = newicon.getData().remaining();
|
||||||
|
//
|
||||||
|
// iconentry.Data = newicon.getData();
|
||||||
|
// iconentry.Size = iconentry.Data.remaining();
|
||||||
|
//
|
||||||
|
// entry.setData(rid.getData());
|
||||||
|
// System.out.println("POST CHANGE ResIconDir :");
|
||||||
|
// System.out.println(rid.toString());
|
||||||
|
|
||||||
|
// ResIcon test = new ResIcon(icon.getData());
|
||||||
|
// System.out.println("PROOF-TEST:\n" + test.toString());
|
||||||
|
|
||||||
|
// / BACK
|
||||||
|
//
|
||||||
|
// rid.getEntries()[0].bWidth = (short)icon.Width;
|
||||||
|
// rid.getEntries()[0].bHeight = (short)(icon.Height/2);
|
||||||
|
// rid.getEntries()[0].bColorCount = (short)(1 <<icon.BitsPerPixel);
|
||||||
|
// rid.getEntries()[0].wBitCount = icon.BitsPerPixel;
|
||||||
|
// iconentry.Data = icon.getData();
|
||||||
|
// iconentry.Size = iconentry.Data.remaining();
|
||||||
|
|
||||||
|
// resdir.addNewResource("POUET", "A666", "#1033", data);
|
||||||
|
|
||||||
|
// resdir.dump(System.out);
|
||||||
|
|
||||||
|
// System.out.println("New size = " + resdir.size());
|
||||||
|
File out = new File("c:/scratch/COPIE.exe");
|
||||||
|
pe.dumpTo(out);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
public PEResourceDirectory getResourceDirectory() throws IOException {
|
||||||
|
if (m_resourceDir != null)
|
||||||
|
return m_resourceDir;
|
||||||
|
|
||||||
|
long resourceoffset = m_header.ResourceDirectory_VA;
|
||||||
|
for (int i = 0; i < m_sections.size(); i++) {
|
||||||
|
PESection sect = (PESection)m_sections.get(i);
|
||||||
|
if (sect.VirtualAddress == resourceoffset) {
|
||||||
|
m_resourceDir = new PEResourceDirectory(this, sect);
|
||||||
|
return m_resourceDir;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void dumpTo(File destination) throws IOException, CloneNotSupportedException {
|
||||||
|
int outputcount = 0;
|
||||||
|
FileOutputStream fos = new FileOutputStream(destination);
|
||||||
|
FileChannel out = fos.getChannel();
|
||||||
|
|
||||||
|
//
|
||||||
|
// Make a copy of the Header, for safe modifications
|
||||||
|
//
|
||||||
|
PEOldMSHeader oldmsheader = (PEOldMSHeader)this.m_oldmsheader.clone();
|
||||||
|
PEHeader peheader = (PEHeader)m_header.clone();
|
||||||
|
Vector sections = new Vector();
|
||||||
|
for (int i = 0; i < m_sections.size(); i++) {
|
||||||
|
PESection sect = (PESection)m_sections.get(i);
|
||||||
|
PESection cs = (PESection)sect.clone();
|
||||||
|
sections.add(cs);
|
||||||
|
}
|
||||||
|
|
||||||
|
//
|
||||||
|
// First, write the old MS Header, the one starting
|
||||||
|
// with "MZ"...
|
||||||
|
//
|
||||||
|
long newexeoffset = oldmsheader.e_lfanew;
|
||||||
|
ByteBuffer msheadbuffer = oldmsheader.get();
|
||||||
|
outputcount = out.write(msheadbuffer);
|
||||||
|
this.m_channel.position(64);
|
||||||
|
out.transferFrom(this.m_channel, 64, newexeoffset - 64);
|
||||||
|
|
||||||
|
|
||||||
|
//
|
||||||
|
// Then Write the new Header...
|
||||||
|
//
|
||||||
|
ByteBuffer headbuffer = peheader.get();
|
||||||
|
out.position(newexeoffset);
|
||||||
|
outputcount = out.write(headbuffer);
|
||||||
|
|
||||||
|
//
|
||||||
|
// After the header, there are all the section
|
||||||
|
// headers...
|
||||||
|
//
|
||||||
|
long offset = oldmsheader.e_lfanew + (m_header.NumberOfRvaAndSizes * 8) + 24 + getPeHeaderOffset();
|
||||||
|
out.position(offset);
|
||||||
|
for (int i = 0; i < sections.size(); i++) {
|
||||||
|
// System.out.println(" offset: " + out.position());
|
||||||
|
PESection sect = (PESection)sections.get(i);
|
||||||
|
|
||||||
|
ByteBuffer buf = sect.get();
|
||||||
|
outputcount = out.write(buf);
|
||||||
|
}
|
||||||
|
|
||||||
|
//
|
||||||
|
// Now, we write the real data: each of the section
|
||||||
|
// and their data...
|
||||||
|
//
|
||||||
|
|
||||||
|
// Not sure why it's always at 1024... ?
|
||||||
|
offset = 1024;
|
||||||
|
|
||||||
|
//
|
||||||
|
// Dump each section data
|
||||||
|
//
|
||||||
|
|
||||||
|
long virtualAddress = offset;
|
||||||
|
if ((virtualAddress % peheader.SectionAlignment) > 0)
|
||||||
|
virtualAddress += peheader.SectionAlignment - (virtualAddress % peheader.SectionAlignment);
|
||||||
|
|
||||||
|
long resourceoffset = m_header.ResourceDirectory_VA;
|
||||||
|
for (int i = 0; i < sections.size(); i++) {
|
||||||
|
PESection sect = (PESection)sections.get(i);
|
||||||
|
if (resourceoffset == sect.VirtualAddress) {
|
||||||
|
// System.out.println("Dumping RES section " + i + " at " + offset + " from " + sect.PointerToRawData + " (VA=" + virtualAddress + ")");
|
||||||
|
out.position(offset);
|
||||||
|
long sectoffset = offset;
|
||||||
|
PEResourceDirectory prd = this.getResourceDirectory();
|
||||||
|
ByteBuffer resbuf = prd.buildResource(sect.VirtualAddress);
|
||||||
|
resbuf.position(0);
|
||||||
|
|
||||||
|
out.write(resbuf);
|
||||||
|
offset += resbuf.capacity();
|
||||||
|
long rem = offset % this.m_header.FileAlignment;
|
||||||
|
if (rem != 0)
|
||||||
|
offset += this.m_header.FileAlignment - rem;
|
||||||
|
|
||||||
|
if (out.size() + 1 < offset) {
|
||||||
|
ByteBuffer padder = ByteBuffer.allocate(1);
|
||||||
|
out.write(padder, offset - 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
long virtualSize = resbuf.capacity();
|
||||||
|
if ((virtualSize % peheader.SectionAlignment) > 0)
|
||||||
|
virtualSize += peheader.SectionAlignment - (virtualSize % peheader.SectionAlignment);
|
||||||
|
|
||||||
|
sect.PointerToRawData = sectoffset;
|
||||||
|
sect.SizeOfRawData = resbuf.capacity();
|
||||||
|
if ((sect.SizeOfRawData % this.m_header.FileAlignment) > 0)
|
||||||
|
sect.SizeOfRawData += (this.m_header.FileAlignment - (sect.SizeOfRawData % this.m_header.FileAlignment));
|
||||||
|
sect.VirtualAddress = virtualAddress;
|
||||||
|
sect.VirtualSize = virtualSize;
|
||||||
|
// System.out.println(" VS=" + virtualSize + " at VA=" + virtualAddress);
|
||||||
|
virtualAddress += virtualSize;
|
||||||
|
|
||||||
|
} else if (sect.PointerToRawData > 0) {
|
||||||
|
// System.out.println("Dumping section " + i + "/" + sect.getName() + " at " + offset + " from " + sect.PointerToRawData + " (VA=" + virtualAddress + ")");
|
||||||
|
out.position(offset);
|
||||||
|
this.m_channel.position(sect.PointerToRawData);
|
||||||
|
long sectoffset = offset;
|
||||||
|
|
||||||
|
out.position(offset + sect.SizeOfRawData);
|
||||||
|
ByteBuffer padder = ByteBuffer.allocate(1);
|
||||||
|
out.write(padder, offset + sect.SizeOfRawData - 1);
|
||||||
|
|
||||||
|
long outted = out.transferFrom(this.m_channel, offset, sect.SizeOfRawData);
|
||||||
|
offset += sect.SizeOfRawData;
|
||||||
|
// System.out.println("offset before alignment, " + offset);
|
||||||
|
|
||||||
|
long rem = offset % this.m_header.FileAlignment;
|
||||||
|
if (rem != 0) {
|
||||||
|
offset += this.m_header.FileAlignment - rem;
|
||||||
|
}
|
||||||
|
// System.out.println("offset after alignment, " + offset);
|
||||||
|
|
||||||
|
// long virtualSize = sect.SizeOfRawData;
|
||||||
|
// if ((virtualSize % peheader.SectionAlignment)>0)
|
||||||
|
// virtualSize += peheader.SectionAlignment - (virtualSize%peheader.SectionAlignment);
|
||||||
|
|
||||||
|
sect.PointerToRawData = sectoffset;
|
||||||
|
// sect.SizeOfRawData =
|
||||||
|
sect.VirtualAddress = virtualAddress;
|
||||||
|
// sect.VirtualSize = virtualSize;
|
||||||
|
|
||||||
|
virtualAddress += sect.VirtualSize;
|
||||||
|
if ((virtualAddress % peheader.SectionAlignment) > 0)
|
||||||
|
virtualAddress += peheader.SectionAlignment - (virtualAddress % peheader.SectionAlignment);
|
||||||
|
|
||||||
|
} else {
|
||||||
|
// generally a BSS, with a virtual size but no
|
||||||
|
// data in the file...
|
||||||
|
// System.out.println("Dumping section " + i + " at " + offset + " from " + sect.PointerToRawData + " (VA=" + virtualAddress + ")");
|
||||||
|
long virtualSize = sect.VirtualSize;
|
||||||
|
if ((virtualSize % peheader.SectionAlignment) > 0)
|
||||||
|
virtualSize += peheader.SectionAlignment - (virtualSize % peheader.SectionAlignment);
|
||||||
|
|
||||||
|
sect.VirtualAddress = virtualAddress;
|
||||||
|
// sect.VirtualSize = virtualSize;
|
||||||
|
virtualAddress += virtualSize;
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
//
|
||||||
|
// Now that all the sections have been written, we have the
|
||||||
|
// correct VirtualAddress and Sizes, so we can update the new
|
||||||
|
// header and all the section headers...
|
||||||
|
|
||||||
|
peheader.updateVAAndSize(m_sections, sections);
|
||||||
|
headbuffer = peheader.get();
|
||||||
|
out.position(newexeoffset);
|
||||||
|
outputcount = out.write(headbuffer);
|
||||||
|
|
||||||
|
// peheader.dump(System.out);
|
||||||
|
// System.out.println("Dumping the section again...");
|
||||||
|
offset = oldmsheader.e_lfanew + (m_header.NumberOfRvaAndSizes * 8) + 24 + getPeHeaderOffset();
|
||||||
|
out.position(offset);
|
||||||
|
for (int i = 0; i < sections.size(); i++) {
|
||||||
|
// System.out.println(" offset: " + out.position());
|
||||||
|
PESection sect = (PESection)sections.get(i);
|
||||||
|
// sect.dump(System.out);
|
||||||
|
ByteBuffer buf = sect.get();
|
||||||
|
outputcount = out.write(buf);
|
||||||
|
}
|
||||||
|
|
||||||
|
fos.flush();
|
||||||
|
fos.close();
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
*/
|
||||||
|
|
||||||
|
public void replaceDefaultIcon(ResIcon icon) throws Exception {
|
||||||
|
PEResourceDirectory resdir = getResourceDirectory();
|
||||||
|
|
||||||
|
PEResourceDirectory.DataEntry entry = resdir.getData("#14", null, null);
|
||||||
|
if (entry == null) {
|
||||||
|
throw new Exception("Can't find any icon group in the file!");
|
||||||
|
}
|
||||||
|
|
||||||
|
entry.Data.position(0);
|
||||||
|
// System.out.println("DataEntry found : " + entry + " (size=" + entry.Data.remaining() + ")");
|
||||||
|
entry.Data.position(0);
|
||||||
|
|
||||||
|
ResIconDir rid = new ResIconDir(entry.Data);
|
||||||
|
// System.out.println("ResIconDir :");
|
||||||
|
// System.out.println(rid.toString());
|
||||||
|
int iconid = rid.getEntries()[0].dwImageOffset;
|
||||||
|
// System.out.println("Icon Index: " + iconid);
|
||||||
|
|
||||||
|
PEResourceDirectory.DataEntry iconentry = resdir.getData("#3", "#" + iconid, null);
|
||||||
|
iconentry.Data.position(0);
|
||||||
|
// System.out.println("Icon :");
|
||||||
|
// System.out.println(icon.toString());
|
||||||
|
|
||||||
|
rid.getEntries()[0].bWidth = (short)icon.Width;
|
||||||
|
rid.getEntries()[0].bHeight = (short)(icon.Height / 2);
|
||||||
|
rid.getEntries()[0].bColorCount = (short)(1 << icon.BitsPerPixel);
|
||||||
|
rid.getEntries()[0].wBitCount = icon.BitsPerPixel;
|
||||||
|
rid.getEntries()[0].dwBytesInRes = icon.getData().remaining();
|
||||||
|
|
||||||
|
iconentry.Data = icon.getData();
|
||||||
|
iconentry.Size = iconentry.Data.remaining();
|
||||||
|
|
||||||
|
entry.setData(rid.getData());
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
@ -0,0 +1,482 @@
|
|||||||
|
/*
|
||||||
|
JSmooth: a VM wrapper toolkit for Windows
|
||||||
|
Copyright (C) 2003 Rodrigo Reyes <reyes@charabia.net>
|
||||||
|
|
||||||
|
This program is free software; you can redistribute it and/or modify
|
||||||
|
it under the terms of the GNU General Public License as published by
|
||||||
|
the Free Software Foundation; either version 2 of the License, or
|
||||||
|
(at your option) any later version.
|
||||||
|
|
||||||
|
This program is distributed in the hope that it will be useful,
|
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
GNU General Public License for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU General Public License
|
||||||
|
along with this program; if not, write to the Free Software
|
||||||
|
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
||||||
|
|
||||||
|
*/
|
||||||
|
|
||||||
|
/*
|
||||||
|
* PEHeader.java
|
||||||
|
*
|
||||||
|
* Created on 28 juillet 2003, 21:38
|
||||||
|
*/
|
||||||
|
|
||||||
|
package net.charabia.jsmoothgen.pe;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.PrintStream;
|
||||||
|
import java.nio.ByteBuffer;
|
||||||
|
import java.nio.ByteOrder;
|
||||||
|
import java.nio.channels.FileChannel;
|
||||||
|
import java.util.Vector;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author Rodrigo Reyes
|
||||||
|
*/
|
||||||
|
public class PEHeader implements Cloneable {
|
||||||
|
private int PeMagic; // 0
|
||||||
|
public int Machine; // 4
|
||||||
|
public int NumberOfSections; // 6
|
||||||
|
public long TimeDateStamp; // 8
|
||||||
|
public long PointerToSymbolTable; // C
|
||||||
|
public long NumberOfSymbols; // 10
|
||||||
|
public int SizeOfOptionalHeader; // 14
|
||||||
|
public int Characteristics; // 16
|
||||||
|
|
||||||
|
// Optional Header
|
||||||
|
public int Magic; // 18
|
||||||
|
public short MajorLinkerVersion; // 1a
|
||||||
|
public short MinorLinkerVersion; // 1b
|
||||||
|
public long SizeOfCode; // 1c
|
||||||
|
public long SizeOfInitializedData; // 20
|
||||||
|
public long SizeOfUninitializedData; // 24
|
||||||
|
public long AddressOfEntryPoint; // 28
|
||||||
|
public long BaseOfCode; // 2c
|
||||||
|
public long BaseOfData; // NT additional fields. 30
|
||||||
|
public long ImageBase; // 34
|
||||||
|
public long SectionAlignment; // 38
|
||||||
|
public long FileAlignment; // 3c
|
||||||
|
public int MajorOperatingSystemVersion; // 40
|
||||||
|
public int MinorOperatingSystemVersion; // 42
|
||||||
|
public int MajorImageVersion; // 44
|
||||||
|
public int MinorImageVersion; // 46
|
||||||
|
public int MajorSubsystemVersion; // 48
|
||||||
|
public int MinorSubsystemVersion; // 4a
|
||||||
|
public long Reserved1; // 4c
|
||||||
|
public long SizeOfImage; // 50
|
||||||
|
public long SizeOfHeaders; // 54
|
||||||
|
public long CheckSum; // 58
|
||||||
|
public int Subsystem; // 5c
|
||||||
|
public int DllCharacteristics; // 5e
|
||||||
|
public long SizeOfStackReserve; // 60
|
||||||
|
public long SizeOfStackCommit; // 64
|
||||||
|
public long SizeOfHeapReserve; // 68
|
||||||
|
public long SizeOfHeapCommit; // 6c
|
||||||
|
public long LoaderFlags; // 70
|
||||||
|
public long NumberOfRvaAndSizes; // 74
|
||||||
|
|
||||||
|
public long ExportDirectory_VA; // 78
|
||||||
|
public long ExportDirectory_Size; // 7c
|
||||||
|
public long ImportDirectory_VA; // 80
|
||||||
|
public long ImportDirectory_Size; // 84
|
||||||
|
public long ResourceDirectory_VA; // 88
|
||||||
|
public long ResourceDirectory_Size; // 8c
|
||||||
|
public long ExceptionDirectory_VA; // 90
|
||||||
|
public long ExceptionDirectory_Size; // 94
|
||||||
|
public long SecurityDirectory_VA; // 98
|
||||||
|
public long SecurityDirectory_Size; // 9c
|
||||||
|
public long BaseRelocationTable_VA; // a0
|
||||||
|
public long BaseRelocationTable_Size; // a4
|
||||||
|
public long DebugDirectory_VA; // a8
|
||||||
|
public long DebugDirectory_Size; // ac
|
||||||
|
public long ArchitectureSpecificData_VA; // b0
|
||||||
|
public long ArchitectureSpecificData_Size; // b4
|
||||||
|
public long RVAofGP_VA; // b8
|
||||||
|
public long RVAofGP_Size; // bc
|
||||||
|
public long TLSDirectory_VA; // c0
|
||||||
|
public long TLSDirectory_Size; // c4
|
||||||
|
public long LoadConfigurationDirectory_VA; // c8
|
||||||
|
public long LoadConfigurationDirectory_Size; // cc
|
||||||
|
public long BoundImportDirectoryinheaders_VA; // d0
|
||||||
|
public long BoundImportDirectoryinheaders_Size; // d4
|
||||||
|
public long ImportAddressTable_VA; // d8
|
||||||
|
public long ImportAddressTable_Size; // dc
|
||||||
|
public long DelayLoadImportDescriptors_VA; // e0
|
||||||
|
public long DelayLoadImportDescriptors_Size; // e4
|
||||||
|
public long COMRuntimedescriptor_VA; // e8
|
||||||
|
public long COMRuntimedescriptor_Size; // ec
|
||||||
|
|
||||||
|
private long m_baseoffset;
|
||||||
|
private PEFile m_pe;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Creates a new instance of PEHeader
|
||||||
|
*/
|
||||||
|
public PEHeader(PEFile pef, long baseoffset) {
|
||||||
|
m_pe = pef;
|
||||||
|
m_baseoffset = baseoffset;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Object clone() throws CloneNotSupportedException {
|
||||||
|
return super.clone();
|
||||||
|
}
|
||||||
|
|
||||||
|
public void read() throws IOException {
|
||||||
|
FileChannel ch = m_pe.getChannel();
|
||||||
|
ByteBuffer head = ByteBuffer.allocate(350);
|
||||||
|
head.order(ByteOrder.LITTLE_ENDIAN);
|
||||||
|
ch.position(m_baseoffset);
|
||||||
|
ch.read(head);
|
||||||
|
head.position(0);
|
||||||
|
|
||||||
|
PeMagic = head.getInt();
|
||||||
|
// System.out.println("MAGIC::: " + pemagic);
|
||||||
|
Machine = head.getShort(); // 4
|
||||||
|
NumberOfSections = head.getShort(); // 6
|
||||||
|
TimeDateStamp = head.getInt(); // 8
|
||||||
|
PointerToSymbolTable = head.getInt(); // C
|
||||||
|
NumberOfSymbols = head.getInt(); // 10
|
||||||
|
SizeOfOptionalHeader = head.getShort(); // 14
|
||||||
|
Characteristics = head.getShort(); // 16
|
||||||
|
// Optional Header
|
||||||
|
|
||||||
|
Magic = head.getShort(); // 18
|
||||||
|
MajorLinkerVersion = head.get(); // 1a
|
||||||
|
MinorLinkerVersion = head.get(); // 1b
|
||||||
|
SizeOfCode = head.getInt(); // 1c
|
||||||
|
SizeOfInitializedData = head.getInt(); // 20
|
||||||
|
SizeOfUninitializedData = head.getInt(); // 24
|
||||||
|
AddressOfEntryPoint = head.getInt(); // 28
|
||||||
|
BaseOfCode = head.getInt(); // 2c
|
||||||
|
if (isPe32Plus()) {
|
||||||
|
ImageBase = head.getLong(); // 34
|
||||||
|
} else {
|
||||||
|
BaseOfData = head.getInt(); // // NT additional fields. // 30
|
||||||
|
ImageBase = head.getInt(); // 34
|
||||||
|
}
|
||||||
|
SectionAlignment = head.getInt(); // 38
|
||||||
|
FileAlignment = head.getInt(); // 3c
|
||||||
|
MajorOperatingSystemVersion = head.getShort(); // 40
|
||||||
|
MinorOperatingSystemVersion = head.getShort(); // 42
|
||||||
|
MajorImageVersion = head.getShort(); // 44
|
||||||
|
MinorImageVersion = head.getShort(); // 46
|
||||||
|
MajorSubsystemVersion = head.getShort(); // 48
|
||||||
|
MinorSubsystemVersion = head.getShort(); // 4a
|
||||||
|
Reserved1 = head.getInt(); // 4c
|
||||||
|
SizeOfImage = head.getInt(); // 50
|
||||||
|
SizeOfHeaders = head.getInt(); // 54
|
||||||
|
CheckSum = head.getInt(); // 58
|
||||||
|
Subsystem = head.getShort(); // 5c
|
||||||
|
DllCharacteristics = head.getShort(); // 5e
|
||||||
|
if (isPe32Plus()) {
|
||||||
|
SizeOfStackReserve = head.getLong(); // 60
|
||||||
|
SizeOfStackCommit = head.getLong(); // 64
|
||||||
|
SizeOfHeapReserve = head.getLong(); // 68
|
||||||
|
SizeOfHeapCommit = head.getLong(); // 6c
|
||||||
|
} else {
|
||||||
|
SizeOfStackReserve = head.getInt(); // 60
|
||||||
|
SizeOfStackCommit = head.getInt(); // 64
|
||||||
|
SizeOfHeapReserve = head.getInt(); // 68
|
||||||
|
SizeOfHeapCommit = head.getInt(); // 6c
|
||||||
|
}
|
||||||
|
LoaderFlags = head.getInt(); // 70
|
||||||
|
NumberOfRvaAndSizes = head.getInt(); // 74
|
||||||
|
|
||||||
|
ExportDirectory_VA = head.getInt(); // 78
|
||||||
|
ExportDirectory_Size = head.getInt(); // 7c
|
||||||
|
ImportDirectory_VA = head.getInt(); // 80
|
||||||
|
ImportDirectory_Size = head.getInt(); // 84
|
||||||
|
ResourceDirectory_VA = head.getInt(); // 88
|
||||||
|
ResourceDirectory_Size = head.getInt(); // 8c
|
||||||
|
ExceptionDirectory_VA = head.getInt(); // 90
|
||||||
|
ExceptionDirectory_Size = head.getInt(); // 94
|
||||||
|
SecurityDirectory_VA = head.getInt(); // 98
|
||||||
|
SecurityDirectory_Size = head.getInt(); // 9c
|
||||||
|
BaseRelocationTable_VA = head.getInt(); // a0
|
||||||
|
BaseRelocationTable_Size = head.getInt(); // a4
|
||||||
|
DebugDirectory_VA = head.getInt(); // a8
|
||||||
|
DebugDirectory_Size = head.getInt(); // ac
|
||||||
|
ArchitectureSpecificData_VA = head.getInt(); // b0
|
||||||
|
ArchitectureSpecificData_Size = head.getInt(); // b4
|
||||||
|
RVAofGP_VA = head.getInt(); // b8
|
||||||
|
RVAofGP_Size = head.getInt(); // bc
|
||||||
|
TLSDirectory_VA = head.getInt(); // c0
|
||||||
|
TLSDirectory_Size = head.getInt(); // c4
|
||||||
|
LoadConfigurationDirectory_VA = head.getInt(); // c8
|
||||||
|
LoadConfigurationDirectory_Size = head.getInt(); // cc
|
||||||
|
BoundImportDirectoryinheaders_VA = head.getInt(); // d0
|
||||||
|
BoundImportDirectoryinheaders_Size = head.getInt(); // d4
|
||||||
|
ImportAddressTable_VA = head.getInt(); // d8
|
||||||
|
ImportAddressTable_Size = head.getInt(); // dc
|
||||||
|
DelayLoadImportDescriptors_VA = head.getInt(); // e0
|
||||||
|
DelayLoadImportDescriptors_Size = head.getInt(); // e4
|
||||||
|
COMRuntimedescriptor_VA = head.getInt(); // e8
|
||||||
|
COMRuntimedescriptor_Size = head.getInt(); // ec
|
||||||
|
}
|
||||||
|
|
||||||
|
public void dump(PrintStream out) {
|
||||||
|
out.println("HEADER:");
|
||||||
|
out.println("int Machine=" + Machine + " // 4");
|
||||||
|
out.println("int NumberOfSections=" + NumberOfSections + " // 6");
|
||||||
|
out.println("long TimeDateStamp=" + TimeDateStamp + " // 8");
|
||||||
|
out.println("long PointerToSymbolTable=" + PointerToSymbolTable + " // C");
|
||||||
|
out.println("long NumberOfSymbols=" + NumberOfSymbols + " // 10");
|
||||||
|
out.println("int SizeOfOptionalHeader=" + SizeOfOptionalHeader + " // 14");
|
||||||
|
out.println("int Characteristics=" + Characteristics + " // 16");
|
||||||
|
// Optional Header
|
||||||
|
|
||||||
|
out.println("int Magic=" + Magic + " // 18");
|
||||||
|
out.println("short MajorLinkerVersion=" + MajorLinkerVersion + " // 1a");
|
||||||
|
out.println("short MinorLinkerVersion=" + MinorLinkerVersion + " // 1b");
|
||||||
|
out.println("long SizeOfCode=" + SizeOfCode + " // 1c");
|
||||||
|
out.println("long SizeOfInitializedData=" + SizeOfInitializedData + " // 20");
|
||||||
|
out.println("long SizeOfUninitializedData=" + SizeOfUninitializedData + " // 24");
|
||||||
|
out.println("long AddressOfEntryPoint=" + AddressOfEntryPoint + " // 28");
|
||||||
|
out.println("long BaseOfCode=" + BaseOfCode + " // 2c");
|
||||||
|
out.println("long BaseOfData=" + BaseOfData + " // // NT additional fields. // 30");
|
||||||
|
//
|
||||||
|
out.println("long ImageBase=" + ImageBase + " // 34");
|
||||||
|
out.println("long SectionAlignment=" + SectionAlignment + " // 38");
|
||||||
|
out.println("long FileAlignment=" + FileAlignment + " // 3c");
|
||||||
|
out.println("int MajorOperatingSystemVersion=" + MajorOperatingSystemVersion + " // 40");
|
||||||
|
out.println("int MinorOperatingSystemVersion=" + MinorOperatingSystemVersion + " // 42");
|
||||||
|
out.println("int MajorImageVersion=" + MajorImageVersion + " // 44");
|
||||||
|
out.println("int MinorImageVersion=" + MinorImageVersion + " // 46");
|
||||||
|
out.println("int MajorSubsystemVersion=" + MajorSubsystemVersion + " // 48");
|
||||||
|
out.println("int MinorSubsystemVersion=" + MinorSubsystemVersion + " // 4a");
|
||||||
|
out.println("long Reserved1=" + Reserved1 + " // 4c");
|
||||||
|
out.println("long SizeOfImage=" + SizeOfImage + " // 50");
|
||||||
|
out.println("long SizeOfHeaders=" + SizeOfHeaders + " // 54");
|
||||||
|
out.println("long CheckSum=" + CheckSum + " // 58");
|
||||||
|
out.println("int Subsystem=" + Subsystem + " // 5c");
|
||||||
|
out.println("int DllCharacteristics=" + DllCharacteristics + " // 5e");
|
||||||
|
out.println("long SizeOfStackReserve=" + SizeOfStackReserve + " // 60");
|
||||||
|
out.println("long SizeOfStackCommit=" + SizeOfStackCommit + " // 64");
|
||||||
|
out.println("long SizeOfHeapReserve=" + SizeOfHeapReserve + " // 68");
|
||||||
|
out.println("long SizeOfHeapCommit=" + SizeOfHeapCommit + " // 6c");
|
||||||
|
out.println("long LoaderFlags=" + LoaderFlags + " // 70");
|
||||||
|
out.println("long NumberOfRvaAndSizes=" + NumberOfRvaAndSizes + " // 74");
|
||||||
|
|
||||||
|
out.println("long ExportDirectory_VA=" + ExportDirectory_VA + " // 78");
|
||||||
|
out.println("long ExportDirectory_Size=" + ExportDirectory_Size + " // 7c");
|
||||||
|
out.println("long ImportDirectory_VA=" + ImportDirectory_VA + " // 80");
|
||||||
|
out.println("long ImportDirectory_Size=" + ImportDirectory_Size + " // 84");
|
||||||
|
out.println("long ResourceDirectory_VA=" + ResourceDirectory_VA + " // 88");
|
||||||
|
out.println("long ResourceDirectory_Size=" + ResourceDirectory_Size + " // 8c");
|
||||||
|
out.println("long ExceptionDirectory_VA=" + ExceptionDirectory_VA + " // 90");
|
||||||
|
out.println("long ExceptionDirectory_Size=" + ExceptionDirectory_Size + " // 94");
|
||||||
|
out.println("long SecurityDirectory_VA=" + SecurityDirectory_VA + " // 98");
|
||||||
|
out.println("long SecurityDirectory_Size=" + SecurityDirectory_Size + " // 9c");
|
||||||
|
out.println("long BaseRelocationTable_VA=" + BaseRelocationTable_VA + " // a0");
|
||||||
|
out.println("long BaseRelocationTable_Size=" + BaseRelocationTable_Size + " // a4");
|
||||||
|
out.println("long DebugDirectory_VA=" + DebugDirectory_VA + " // a8");
|
||||||
|
out.println("long DebugDirectory_Size=" + DebugDirectory_Size + " // ac");
|
||||||
|
out.println("long ArchitectureSpecificData_VA=" + ArchitectureSpecificData_VA + " // b0");
|
||||||
|
out.println("long ArchitectureSpecificData_Size=" + ArchitectureSpecificData_Size + " // b4");
|
||||||
|
out.println("long RVAofGP_VA=" + RVAofGP_VA + " // b8");
|
||||||
|
out.println("long RVAofGP_Size=" + RVAofGP_Size + " // bc");
|
||||||
|
out.println("long TLSDirectory_VA=" + TLSDirectory_VA + " // c0");
|
||||||
|
out.println("long TLSDirectory_Size=" + TLSDirectory_Size + " // c4");
|
||||||
|
out.println("long LoadConfigurationDirectory_VA=" + LoadConfigurationDirectory_VA + " // c8");
|
||||||
|
out.println("long LoadConfigurationDirectory_Size=" + LoadConfigurationDirectory_Size + " // cc");
|
||||||
|
out.println("long BoundImportDirectoryinheaders_VA=" + BoundImportDirectoryinheaders_VA + " // d0");
|
||||||
|
out.println("long BoundImportDirectoryinheaders_Size=" + BoundImportDirectoryinheaders_Size + " // d4");
|
||||||
|
out.println("long ImportAddressTable_VA=" + ImportAddressTable_VA + " // d8");
|
||||||
|
out.println("long ImportAddressTable_Size=" + ImportAddressTable_Size + " // dc");
|
||||||
|
out.println("long DelayLoadImportDescriptors_VA=" + DelayLoadImportDescriptors_VA + " // e0");
|
||||||
|
out.println("long DelayLoadImportDescriptors_Size=" + DelayLoadImportDescriptors_Size + " // e4");
|
||||||
|
out.println("long COMRuntimedescriptor_VA=" + COMRuntimedescriptor_VA + " // e8");
|
||||||
|
out.println("long COMRuntimedescriptor_Size=" + COMRuntimedescriptor_Size + " // ec");
|
||||||
|
}
|
||||||
|
|
||||||
|
public ByteBuffer get() {
|
||||||
|
ByteBuffer head = ByteBuffer.allocate(16 + this.SizeOfOptionalHeader);
|
||||||
|
head.order(ByteOrder.LITTLE_ENDIAN);
|
||||||
|
head.position(0);
|
||||||
|
|
||||||
|
head.putInt(PeMagic);
|
||||||
|
|
||||||
|
head.putShort((short)Machine); // 4
|
||||||
|
head.putShort((short)NumberOfSections); // 6
|
||||||
|
head.putInt((int)TimeDateStamp); // 8
|
||||||
|
head.putInt((int)PointerToSymbolTable); // C
|
||||||
|
head.putInt((int)NumberOfSymbols); // 10
|
||||||
|
head.putShort((short)SizeOfOptionalHeader); // 14
|
||||||
|
head.putShort((short)Characteristics); // 16
|
||||||
|
// Optional Header
|
||||||
|
|
||||||
|
head.putShort((short)Magic); // 18
|
||||||
|
head.put((byte)MajorLinkerVersion); // 1a
|
||||||
|
head.put((byte)MinorLinkerVersion); // 1b
|
||||||
|
head.putInt((int)SizeOfCode); // 1c
|
||||||
|
head.putInt((int)SizeOfInitializedData); // 20
|
||||||
|
head.putInt((int)SizeOfUninitializedData); // 24
|
||||||
|
head.putInt((int)AddressOfEntryPoint); // 28
|
||||||
|
head.putInt((int)BaseOfCode); // 2c
|
||||||
|
if (isPe32Plus()) {
|
||||||
|
head.putLong(ImageBase); // 34
|
||||||
|
} else {
|
||||||
|
head.putInt((int)BaseOfData); // // NT additional fields. // 30
|
||||||
|
head.putInt((int)ImageBase); // 34
|
||||||
|
}
|
||||||
|
head.putInt((int)SectionAlignment); // 38
|
||||||
|
head.putInt((int)FileAlignment); // 3c
|
||||||
|
head.putShort((short)MajorOperatingSystemVersion); // 40
|
||||||
|
head.putShort((short)MinorOperatingSystemVersion); // 42
|
||||||
|
head.putShort((short)MajorImageVersion); // 44
|
||||||
|
head.putShort((short)MinorImageVersion); // 46
|
||||||
|
head.putShort((short)MajorSubsystemVersion); // 48
|
||||||
|
head.putShort((short)MinorSubsystemVersion); // 4a
|
||||||
|
head.putInt((int)Reserved1); // 4c
|
||||||
|
head.putInt((int)SizeOfImage); // 50
|
||||||
|
head.putInt((int)SizeOfHeaders); // 54
|
||||||
|
head.putInt((int)CheckSum); // 58
|
||||||
|
head.putShort((short)Subsystem); // 5c
|
||||||
|
head.putShort((short)DllCharacteristics); // 5e
|
||||||
|
if (isPe32Plus()) {
|
||||||
|
head.putLong(SizeOfStackReserve); // 60
|
||||||
|
head.putLong(SizeOfStackCommit); // 64
|
||||||
|
head.putLong(SizeOfHeapReserve); // 68
|
||||||
|
head.putLong(SizeOfHeapCommit); // 6c
|
||||||
|
} else {
|
||||||
|
head.putInt((int)SizeOfStackReserve); // 60
|
||||||
|
head.putInt((int)SizeOfStackCommit); // 64
|
||||||
|
head.putInt((int)SizeOfHeapReserve); // 68
|
||||||
|
head.putInt((int)SizeOfHeapCommit); // 6c
|
||||||
|
}
|
||||||
|
head.putInt((int)LoaderFlags); // 70
|
||||||
|
head.putInt((int)NumberOfRvaAndSizes); // 74
|
||||||
|
|
||||||
|
head.putInt((int)ExportDirectory_VA); // 78
|
||||||
|
head.putInt((int)ExportDirectory_Size); // 7c
|
||||||
|
head.putInt((int)ImportDirectory_VA); // 80
|
||||||
|
head.putInt((int)ImportDirectory_Size); // 84
|
||||||
|
head.putInt((int)ResourceDirectory_VA); // 88
|
||||||
|
head.putInt((int)ResourceDirectory_Size); // 8c
|
||||||
|
head.putInt((int)ExceptionDirectory_VA); // 90
|
||||||
|
head.putInt((int)ExceptionDirectory_Size); // 94
|
||||||
|
head.putInt((int)SecurityDirectory_VA); // 98
|
||||||
|
head.putInt((int)SecurityDirectory_Size); // 9c
|
||||||
|
head.putInt((int)BaseRelocationTable_VA); // a0
|
||||||
|
head.putInt((int)BaseRelocationTable_Size); // a4
|
||||||
|
head.putInt((int)DebugDirectory_VA); // a8
|
||||||
|
head.putInt((int)DebugDirectory_Size); // ac
|
||||||
|
head.putInt((int)ArchitectureSpecificData_VA); // b0
|
||||||
|
head.putInt((int)ArchitectureSpecificData_Size); // b4
|
||||||
|
head.putInt((int)RVAofGP_VA); // b8
|
||||||
|
head.putInt((int)RVAofGP_Size); // bc
|
||||||
|
head.putInt((int)TLSDirectory_VA); // c0
|
||||||
|
head.putInt((int)TLSDirectory_Size); // c4
|
||||||
|
head.putInt((int)LoadConfigurationDirectory_VA); // c8
|
||||||
|
head.putInt((int)LoadConfigurationDirectory_Size); // cc
|
||||||
|
head.putInt((int)BoundImportDirectoryinheaders_VA); // d0
|
||||||
|
head.putInt((int)BoundImportDirectoryinheaders_Size); // d4
|
||||||
|
head.putInt((int)ImportAddressTable_VA); // d8
|
||||||
|
head.putInt((int)ImportAddressTable_Size); // dc
|
||||||
|
head.putInt((int)DelayLoadImportDescriptors_VA); // e0
|
||||||
|
head.putInt((int)DelayLoadImportDescriptors_Size); // e4
|
||||||
|
head.putInt((int)COMRuntimedescriptor_VA); // e8
|
||||||
|
head.putInt((int)COMRuntimedescriptor_Size); // ec
|
||||||
|
|
||||||
|
head.position(0);
|
||||||
|
return head;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void updateVAAndSize(Vector oldsections, Vector newsections) {
|
||||||
|
long codebase = findNewVA(this.BaseOfCode, oldsections, newsections);
|
||||||
|
long codesize = findNewSize(this.BaseOfCode, oldsections, newsections);
|
||||||
|
// System.out.println("New BaseOfCode=" + codebase + " (size=" + codesize + ")");
|
||||||
|
this.BaseOfCode = codebase;
|
||||||
|
this.SizeOfCode = codesize;
|
||||||
|
|
||||||
|
this.AddressOfEntryPoint = findNewVA(this.AddressOfEntryPoint, oldsections, newsections);
|
||||||
|
|
||||||
|
long database = findNewVA(this.BaseOfData, oldsections, newsections);
|
||||||
|
long datasize = findNewSize(this.BaseOfData, oldsections, newsections);
|
||||||
|
// System.out.println("New BaseOfData=" + database + " (size=" + datasize + ")");
|
||||||
|
this.BaseOfData = database;
|
||||||
|
|
||||||
|
long imagesize = 0;
|
||||||
|
for (int i = 0; i < newsections.size(); i++) {
|
||||||
|
PESection sect = (PESection)newsections.get(i);
|
||||||
|
long curmax = sect.VirtualAddress + sect.VirtualSize;
|
||||||
|
if (curmax > imagesize)
|
||||||
|
imagesize = curmax;
|
||||||
|
}
|
||||||
|
this.SizeOfImage = imagesize;
|
||||||
|
|
||||||
|
// this.SizeOfInitializedData = datasize;
|
||||||
|
|
||||||
|
ExportDirectory_Size = findNewSize(ExportDirectory_VA, oldsections, newsections);
|
||||||
|
ExportDirectory_VA = findNewVA(ExportDirectory_VA, oldsections, newsections);
|
||||||
|
ImportDirectory_Size = findNewSize(ImportDirectory_VA, oldsections, newsections);
|
||||||
|
ImportDirectory_VA = findNewVA(ImportDirectory_VA, oldsections, newsections);
|
||||||
|
ResourceDirectory_Size = findNewSize(ResourceDirectory_VA, oldsections, newsections);
|
||||||
|
ResourceDirectory_VA = findNewVA(ResourceDirectory_VA, oldsections, newsections);
|
||||||
|
ExceptionDirectory_Size = findNewSize(ExceptionDirectory_VA, oldsections, newsections);
|
||||||
|
ExceptionDirectory_VA = findNewVA(ExceptionDirectory_VA, oldsections, newsections);
|
||||||
|
SecurityDirectory_Size = findNewSize(SecurityDirectory_VA, oldsections, newsections);
|
||||||
|
SecurityDirectory_VA = findNewVA(SecurityDirectory_VA, oldsections, newsections);
|
||||||
|
BaseRelocationTable_Size = findNewSize(BaseRelocationTable_VA, oldsections, newsections);
|
||||||
|
BaseRelocationTable_VA = findNewVA(BaseRelocationTable_VA, oldsections, newsections);
|
||||||
|
DebugDirectory_Size = findNewSize(DebugDirectory_VA, oldsections, newsections);
|
||||||
|
DebugDirectory_VA = findNewVA(DebugDirectory_VA, oldsections, newsections);
|
||||||
|
ArchitectureSpecificData_Size = findNewSize(ArchitectureSpecificData_VA, oldsections, newsections);
|
||||||
|
ArchitectureSpecificData_VA = findNewVA(ArchitectureSpecificData_VA, oldsections, newsections);
|
||||||
|
RVAofGP_Size = findNewSize(RVAofGP_VA, oldsections, newsections);
|
||||||
|
RVAofGP_VA = findNewVA(RVAofGP_VA, oldsections, newsections);
|
||||||
|
TLSDirectory_Size = findNewSize(TLSDirectory_VA, oldsections, newsections);
|
||||||
|
TLSDirectory_VA = findNewVA(TLSDirectory_VA, oldsections, newsections);
|
||||||
|
LoadConfigurationDirectory_Size = findNewSize(LoadConfigurationDirectory_VA, oldsections, newsections);
|
||||||
|
LoadConfigurationDirectory_VA = findNewVA(LoadConfigurationDirectory_VA, oldsections, newsections);
|
||||||
|
BoundImportDirectoryinheaders_Size = findNewSize(BoundImportDirectoryinheaders_VA, oldsections, newsections);
|
||||||
|
BoundImportDirectoryinheaders_VA = findNewVA(BoundImportDirectoryinheaders_VA, oldsections, newsections);
|
||||||
|
ImportAddressTable_Size = findNewSize(ImportAddressTable_VA, oldsections, newsections);
|
||||||
|
ImportAddressTable_VA = findNewVA(ImportAddressTable_VA, oldsections, newsections);
|
||||||
|
DelayLoadImportDescriptors_Size = findNewSize(DelayLoadImportDescriptors_VA, oldsections, newsections);
|
||||||
|
DelayLoadImportDescriptors_VA = findNewVA(DelayLoadImportDescriptors_VA, oldsections, newsections);
|
||||||
|
COMRuntimedescriptor_Size = findNewSize(COMRuntimedescriptor_VA, oldsections, newsections);
|
||||||
|
COMRuntimedescriptor_VA = findNewVA(COMRuntimedescriptor_VA, oldsections, newsections);
|
||||||
|
}
|
||||||
|
|
||||||
|
public boolean isPe32Plus() {
|
||||||
|
return Magic == 523;
|
||||||
|
}
|
||||||
|
|
||||||
|
private long findNewVA(long current, Vector oldsections, Vector newsections) {
|
||||||
|
for (int i = 0; i < oldsections.size(); i++) {
|
||||||
|
PESection sect = (PESection)oldsections.get(i);
|
||||||
|
if (sect.VirtualAddress == current) {
|
||||||
|
PESection newsect = (PESection)newsections.get(i);
|
||||||
|
|
||||||
|
// System.out.println("Translation VA found for " + current + " = " + i + " (" +newsect.VirtualAddress + ")=" + newsect.getName());
|
||||||
|
return newsect.VirtualAddress;
|
||||||
|
} else if ((current > sect.VirtualAddress) && (current < (sect.VirtualAddress + sect.VirtualSize))) {
|
||||||
|
long diff = current - sect.VirtualAddress;
|
||||||
|
PESection newsect = (PESection)newsections.get(i);
|
||||||
|
// System.out.println("Translation VA found INSIDE " + current + " = " + i + " (" +newsect.VirtualAddress + ")=" + newsect.getName());
|
||||||
|
return newsect.VirtualAddress + diff;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
private long findNewSize(long current, Vector oldsections, Vector newsections) {
|
||||||
|
for (int i = 0; i < oldsections.size(); i++) {
|
||||||
|
PESection sect = (PESection)oldsections.get(i);
|
||||||
|
if (sect.VirtualAddress == current) {
|
||||||
|
PESection newsect = (PESection)newsections.get(i);
|
||||||
|
// System.out.println("Translation Size found for " + current + " = " + i + " (" +newsect.VirtualAddress + ")=" + newsect.getName());
|
||||||
|
// System.out.println(" Old size " + sect.VirtualSize + " vs new size " + newsect.VirtualSize);
|
||||||
|
return newsect.VirtualSize;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
@ -0,0 +1,625 @@
|
|||||||
|
/*
|
||||||
|
JSmooth: a VM wrapper toolkit for Windows
|
||||||
|
Copyright (C) 2003 Rodrigo Reyes <reyes@charabia.net>
|
||||||
|
|
||||||
|
This program is free software; you can redistribute it and/or modify
|
||||||
|
it under the terms of the GNU General Public License as published by
|
||||||
|
the Free Software Foundation; either version 2 of the License, or
|
||||||
|
(at your option) any later version.
|
||||||
|
|
||||||
|
This program is distributed in the hope that it will be useful,
|
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
GNU General Public License for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU General Public License
|
||||||
|
along with this program; if not, write to the Free Software
|
||||||
|
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
||||||
|
|
||||||
|
*/
|
||||||
|
|
||||||
|
/*
|
||||||
|
* PEResourceDirectory.java
|
||||||
|
*
|
||||||
|
* Created on 2 aout 2003, 01:28
|
||||||
|
*/
|
||||||
|
|
||||||
|
package net.charabia.jsmoothgen.pe;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.PrintStream;
|
||||||
|
import java.nio.ByteBuffer;
|
||||||
|
import java.nio.ByteOrder;
|
||||||
|
import java.nio.channels.FileChannel;
|
||||||
|
import java.util.Iterator;
|
||||||
|
import java.util.Vector;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author Rodrigo
|
||||||
|
*/
|
||||||
|
public class PEResourceDirectory {
|
||||||
|
|
||||||
|
/*
|
||||||
|
typedef struct IMAGE_RESOURCE_DIRECTORY {
|
||||||
|
uint32_t Characteristics;
|
||||||
|
uint32_t TimeDateStamp;
|
||||||
|
uint16_t MajorVersion;
|
||||||
|
uint16_t MinorVersion;
|
||||||
|
uint16_t NumberOfNamedEntries;
|
||||||
|
uint16_t NumberOfIdEntries;
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
|
public class DataEntry {
|
||||||
|
long OffsetToData; // To update at each change
|
||||||
|
long Size;
|
||||||
|
long CodePage; // never changed
|
||||||
|
long Reserved; // never changed
|
||||||
|
ByteBuffer Data;
|
||||||
|
|
||||||
|
public DataEntry(ByteBuffer data) {
|
||||||
|
this.Data = data;
|
||||||
|
this.Size = data.capacity();
|
||||||
|
}
|
||||||
|
|
||||||
|
public DataEntry(FileChannel chan, long offset) throws IOException {
|
||||||
|
long orgpos = chan.position();
|
||||||
|
chan.position(PEResourceDirectory.this.m_offsetBase + offset);
|
||||||
|
ByteBuffer buf = ByteBuffer.allocate(16);
|
||||||
|
buf.order(ByteOrder.LITTLE_ENDIAN);
|
||||||
|
chan.read(buf);
|
||||||
|
buf.position(0);
|
||||||
|
|
||||||
|
OffsetToData = buf.getInt();
|
||||||
|
Size = buf.getInt();
|
||||||
|
CodePage = buf.getInt();
|
||||||
|
Reserved = buf.getInt();
|
||||||
|
|
||||||
|
long datapos = PEResourceDirectory.this.m_master.PointerToRawData + (OffsetToData - PEResourceDirectory.this.m_master.VirtualAddress);
|
||||||
|
Data = ByteBuffer.allocate((int)Size);
|
||||||
|
Data.order(ByteOrder.LITTLE_ENDIAN);
|
||||||
|
chan.position(datapos);
|
||||||
|
chan.read(Data);
|
||||||
|
Data.position(0);
|
||||||
|
|
||||||
|
chan.position(orgpos);
|
||||||
|
}
|
||||||
|
|
||||||
|
public int diskSize() {
|
||||||
|
int size = 16 + (int)this.Size;
|
||||||
|
if ((size % 4) > 0)
|
||||||
|
size += 4 - (size % 4);
|
||||||
|
return size;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void dump(PrintStream out, int level) {
|
||||||
|
indent(level, out);
|
||||||
|
out.println("OffsetToData=" + OffsetToData);
|
||||||
|
indent(level, out);
|
||||||
|
out.println("Size=" + Size);
|
||||||
|
indent(level, out);
|
||||||
|
out.println("CodePage=" + CodePage);
|
||||||
|
indent(level, out);
|
||||||
|
out.println("Reserved=" + Reserved);
|
||||||
|
indent(level, out);
|
||||||
|
out.print("Data={ ");
|
||||||
|
for (int i = 0; i < this.Data.capacity(); i++) {
|
||||||
|
out.print("" + Integer.toHexString((byte)Data.get()) + ",");
|
||||||
|
}
|
||||||
|
out.println(" }");
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
private void indent(int level, PrintStream out) {
|
||||||
|
for (int i = 0; i < level; i++)
|
||||||
|
out.print(" ");
|
||||||
|
}
|
||||||
|
|
||||||
|
public int buildBuffer(ByteBuffer buffer, long virtualBaseOffset, int dataOffset) {
|
||||||
|
// System.out.println("Building Data Entry buffer @ " + buffer.position() + " (" + dataOffset + ")");
|
||||||
|
|
||||||
|
dataOffset = buffer.position() + 16;
|
||||||
|
|
||||||
|
buffer.putInt((int)(dataOffset + virtualBaseOffset));
|
||||||
|
buffer.putInt((int)Size);
|
||||||
|
buffer.putInt((int)CodePage);
|
||||||
|
buffer.putInt((int)Reserved);
|
||||||
|
|
||||||
|
Data.position(0);
|
||||||
|
buffer.put(Data);
|
||||||
|
|
||||||
|
dataOffset += Size;
|
||||||
|
if ((dataOffset % 4) > 0)
|
||||||
|
dataOffset += (4 - (dataOffset % 4));
|
||||||
|
|
||||||
|
return dataOffset;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void setData(ByteBuffer data) {
|
||||||
|
Data = data;
|
||||||
|
Size = data.capacity();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public class ResourceEntry {
|
||||||
|
int Id;
|
||||||
|
String Name;
|
||||||
|
|
||||||
|
ImageResourceDirectory Directory;
|
||||||
|
DataEntry Data;
|
||||||
|
|
||||||
|
public ResourceEntry(int id, DataEntry data) {
|
||||||
|
this.Id = id;
|
||||||
|
this.Data = data;
|
||||||
|
}
|
||||||
|
|
||||||
|
public ResourceEntry(String name, DataEntry data) {
|
||||||
|
this.Name = name;
|
||||||
|
this.Data = data;
|
||||||
|
}
|
||||||
|
|
||||||
|
public ResourceEntry(int id, ImageResourceDirectory dir) {
|
||||||
|
this.Id = id;
|
||||||
|
this.Directory = dir;
|
||||||
|
}
|
||||||
|
|
||||||
|
public ResourceEntry(String name, ImageResourceDirectory dir) {
|
||||||
|
this.Name = name;
|
||||||
|
this.Directory = dir;
|
||||||
|
}
|
||||||
|
|
||||||
|
public ResourceEntry(FileChannel chan) throws IOException {
|
||||||
|
// System.out.println("Resource Entry Offset: " + chan.position());
|
||||||
|
ByteBuffer buf = ByteBuffer.allocate(8);
|
||||||
|
buf.order(ByteOrder.LITTLE_ENDIAN);
|
||||||
|
chan.read(buf);
|
||||||
|
buf.position(0);
|
||||||
|
long orgchanpos = chan.position();
|
||||||
|
int val = buf.getInt();
|
||||||
|
long offsetToData = buf.getInt();
|
||||||
|
// System.out.println("Entry: Val=" + val);
|
||||||
|
// System.out.println(" Off=" + offsetToData);
|
||||||
|
|
||||||
|
if (val < 0) {
|
||||||
|
val &= 0x7FFFFFFF;
|
||||||
|
Name = extractStringAt(chan, val);
|
||||||
|
Id = -1;
|
||||||
|
// System.out.println(" String at " + val + " = " + Name);
|
||||||
|
} else {
|
||||||
|
Id = val;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (offsetToData < 0) {
|
||||||
|
offsetToData &= 0x7FFFFFFF;
|
||||||
|
long orgpos = chan.position();
|
||||||
|
chan.position(PEResourceDirectory.this.m_offsetBase + offsetToData);
|
||||||
|
Directory = new PEResourceDirectory.ImageResourceDirectory(chan);
|
||||||
|
chan.position(orgpos);
|
||||||
|
} else {
|
||||||
|
Data = new DataEntry(chan, offsetToData);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public String extractStringAt(FileChannel chan, int offset) throws IOException {
|
||||||
|
long orgchanpos = chan.position();
|
||||||
|
chan.position(PEResourceDirectory.this.m_offsetBase + offset);
|
||||||
|
|
||||||
|
ByteBuffer sizebuf = ByteBuffer.allocate(2);
|
||||||
|
sizebuf.order(ByteOrder.LITTLE_ENDIAN);
|
||||||
|
chan.read(sizebuf);
|
||||||
|
sizebuf.position(0);
|
||||||
|
|
||||||
|
int size = sizebuf.getShort();
|
||||||
|
ByteBuffer buffer = ByteBuffer.allocate(size * 2);
|
||||||
|
buffer.order(ByteOrder.LITTLE_ENDIAN);
|
||||||
|
chan.read(buffer);
|
||||||
|
buffer.position(0);
|
||||||
|
|
||||||
|
StringBuffer buf = new StringBuffer(size);
|
||||||
|
for (int i = 0; i < size; i++) {
|
||||||
|
int c = buffer.getShort();
|
||||||
|
buf.append((char)c);
|
||||||
|
}
|
||||||
|
|
||||||
|
chan.position(orgchanpos);
|
||||||
|
return buf.toString();
|
||||||
|
}
|
||||||
|
|
||||||
|
public int diskSize() {
|
||||||
|
int size = 8;
|
||||||
|
if (Name != null)
|
||||||
|
size += (Name.length() * 2) + 2;
|
||||||
|
|
||||||
|
if (Directory != null)
|
||||||
|
size += Directory.diskSize();
|
||||||
|
else if (Data != null)
|
||||||
|
size += Data.diskSize();
|
||||||
|
|
||||||
|
if ((size % 4) > 0)
|
||||||
|
size += 4 - (size % 4);
|
||||||
|
return size;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void dump(PrintStream out, int level) {
|
||||||
|
indent(level, out);
|
||||||
|
if (this.Name != null)
|
||||||
|
out.println("Name=" + Name);
|
||||||
|
else
|
||||||
|
out.println("Id=#" + Id);
|
||||||
|
|
||||||
|
indent(level, out);
|
||||||
|
if (this.Directory != null) {
|
||||||
|
out.println("ENTRY: DIRECTORY POINTER");
|
||||||
|
this.Directory.dump(out, level + 1);
|
||||||
|
} else {
|
||||||
|
out.println("ENTRY: DATA ENTRY");
|
||||||
|
Data.dump(out, level + 1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void indent(int level, PrintStream out) {
|
||||||
|
for (int i = 0; i < level; i++)
|
||||||
|
out.print(" ");
|
||||||
|
}
|
||||||
|
|
||||||
|
public int buildBuffer(ByteBuffer buffer, long virtualBaseOffset, int dataOffset) {
|
||||||
|
// System.out.println("Building Resource Entry buffer " + Name + "/" + Id + " @ " + buffer.position() + " (" + dataOffset + ")");
|
||||||
|
if (Name != null) {
|
||||||
|
buffer.putInt((int)(dataOffset | 0x80000000));
|
||||||
|
|
||||||
|
int stringoffset = dataOffset;
|
||||||
|
ByteBuffer strbuf = ByteBuffer.allocate(Name.length() * 2 + 2);
|
||||||
|
strbuf.order(ByteOrder.LITTLE_ENDIAN);
|
||||||
|
|
||||||
|
strbuf.putShort((short)Name.length());
|
||||||
|
for (int i = 0; i < Name.length(); i++) {
|
||||||
|
strbuf.putShort((short)Name.charAt(i));
|
||||||
|
}
|
||||||
|
strbuf.position(0);
|
||||||
|
|
||||||
|
long oldpos = buffer.position();
|
||||||
|
buffer.position(dataOffset);
|
||||||
|
buffer.put(strbuf);
|
||||||
|
dataOffset += Name.length() * 2 + 2;
|
||||||
|
if ((dataOffset % 4) != 0)
|
||||||
|
dataOffset += 4 - (dataOffset % 4);
|
||||||
|
buffer.position((int) oldpos);
|
||||||
|
} else {
|
||||||
|
buffer.putInt(Id);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (Directory != null) {
|
||||||
|
buffer.putInt((int)(dataOffset | 0x80000000));
|
||||||
|
|
||||||
|
int oldpos = buffer.position();
|
||||||
|
buffer.position(dataOffset);
|
||||||
|
int dirsize = Directory.buildBuffer(buffer, virtualBaseOffset);
|
||||||
|
dataOffset = dirsize;
|
||||||
|
buffer.position(oldpos);
|
||||||
|
|
||||||
|
} else if (Data != null) {
|
||||||
|
buffer.putInt(dataOffset);
|
||||||
|
int oldpos = buffer.position();
|
||||||
|
buffer.position(dataOffset);
|
||||||
|
dataOffset = Data.buildBuffer(buffer, virtualBaseOffset, dataOffset);
|
||||||
|
buffer.position(oldpos);
|
||||||
|
} else {
|
||||||
|
throw new RuntimeException("Directory and Data are both null!");
|
||||||
|
}
|
||||||
|
|
||||||
|
return dataOffset;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public class ImageResourceDirectory {
|
||||||
|
long Characteristics; // uint32_t
|
||||||
|
long TimeDateStamp; // uint32_t
|
||||||
|
int MajorVersion; // uint16_t
|
||||||
|
int MinorVersion; // uint16_t
|
||||||
|
int NumberOfNamedEntries; // uint16_t
|
||||||
|
int NumberOfIdEntries; // uint16_t
|
||||||
|
Vector NamedEntries = new Vector();
|
||||||
|
Vector IdEntries = new Vector();
|
||||||
|
|
||||||
|
public ImageResourceDirectory() {
|
||||||
|
}
|
||||||
|
|
||||||
|
public ImageResourceDirectory(FileChannel chan) throws IOException {
|
||||||
|
ByteBuffer header = ByteBuffer.allocate(16);
|
||||||
|
header.order(ByteOrder.LITTLE_ENDIAN);
|
||||||
|
chan.read(header);
|
||||||
|
header.position(0);
|
||||||
|
|
||||||
|
Characteristics = header.getInt();
|
||||||
|
TimeDateStamp = header.getInt();
|
||||||
|
MajorVersion = header.getShort();
|
||||||
|
MinorVersion = header.getShort();
|
||||||
|
NumberOfNamedEntries = header.getShort();
|
||||||
|
NumberOfIdEntries = header.getShort();
|
||||||
|
|
||||||
|
for (int i = 0; i < NumberOfNamedEntries; i++) {
|
||||||
|
ResourceEntry re = new ResourceEntry(chan);
|
||||||
|
NamedEntries.add(re);
|
||||||
|
}
|
||||||
|
for (int i = 0; i < NumberOfIdEntries; i++) {
|
||||||
|
ResourceEntry re = new ResourceEntry(chan);
|
||||||
|
IdEntries.add(re);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public void addNamedEntry(ResourceEntry entry) {
|
||||||
|
this.NamedEntries.add(entry);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void addIdEntry(ResourceEntry entry) {
|
||||||
|
this.IdEntries.add(entry);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void addEntry(ResourceEntry entry) {
|
||||||
|
if (entry.Name != null)
|
||||||
|
addNamedEntry(entry);
|
||||||
|
else
|
||||||
|
addIdEntry(entry);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void dump(PrintStream out, int level) {
|
||||||
|
indent(level, out);
|
||||||
|
out.println("Directory: ");
|
||||||
|
indent(level, out);
|
||||||
|
out.println("Characteristics=" + this.Characteristics);
|
||||||
|
indent(level, out);
|
||||||
|
out.println("TimeDateStamp=" + this.TimeDateStamp);
|
||||||
|
indent(level, out);
|
||||||
|
out.println("MajorVersion=" + this.MajorVersion);
|
||||||
|
indent(level, out);
|
||||||
|
out.println("MinorVersion=" + this.MinorVersion);
|
||||||
|
indent(level, out);
|
||||||
|
out.println("NumberOfNamedEntries=" + this.NumberOfNamedEntries);
|
||||||
|
indent(level, out);
|
||||||
|
out.println("NumberOfIdEntries=" + this.NumberOfIdEntries);
|
||||||
|
indent(level, out);
|
||||||
|
out.println("Named Entries:");
|
||||||
|
for (int i = 0; i < NumberOfNamedEntries; i++) {
|
||||||
|
ResourceEntry re = (ResourceEntry)NamedEntries.get(i);
|
||||||
|
re.dump(out, level + 1);
|
||||||
|
}
|
||||||
|
indent(level, out);
|
||||||
|
out.println("Id Entries:");
|
||||||
|
for (int i = 0; i < NumberOfIdEntries; i++) {
|
||||||
|
ResourceEntry re = (ResourceEntry)IdEntries.get(i);
|
||||||
|
re.dump(out, level + 1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void indent(int level, PrintStream out) {
|
||||||
|
for (int i = 0; i < level; i++)
|
||||||
|
out.print(" ");
|
||||||
|
}
|
||||||
|
|
||||||
|
public int diskSize() {
|
||||||
|
int size = 16;
|
||||||
|
for (int i = 0; i < this.NamedEntries.size(); i++) {
|
||||||
|
ResourceEntry re = (ResourceEntry)NamedEntries.get(i);
|
||||||
|
size += re.diskSize();
|
||||||
|
}
|
||||||
|
for (int i = 0; i < this.IdEntries.size(); i++) {
|
||||||
|
ResourceEntry re = (ResourceEntry)IdEntries.get(i);
|
||||||
|
size += re.diskSize();
|
||||||
|
}
|
||||||
|
|
||||||
|
if ((size % 4) > 0)
|
||||||
|
size += 4 - (size % 4);
|
||||||
|
|
||||||
|
return size;
|
||||||
|
}
|
||||||
|
|
||||||
|
public int buildBuffer(ByteBuffer buffer, long virtualBaseOffset) {
|
||||||
|
// System.out.println("Building Directory Entry buffer @ " + buffer.position());
|
||||||
|
|
||||||
|
buffer.putInt((int)this.Characteristics);
|
||||||
|
buffer.putInt((int)this.TimeDateStamp);
|
||||||
|
buffer.putShort((short)this.MajorVersion);
|
||||||
|
buffer.putShort((short)this.MinorVersion);
|
||||||
|
buffer.putShort((short)this.NamedEntries.size());
|
||||||
|
buffer.putShort((short)this.IdEntries.size());
|
||||||
|
|
||||||
|
int dataOffset = buffer.position() + (NamedEntries.size() * 8) + (IdEntries.size() * 8);
|
||||||
|
|
||||||
|
for (int i = 0; i < this.NamedEntries.size(); i++) {
|
||||||
|
ResourceEntry re = (ResourceEntry)this.NamedEntries.get(i);
|
||||||
|
dataOffset = re.buildBuffer(buffer, virtualBaseOffset, dataOffset);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (int i = 0; i < this.IdEntries.size(); i++) {
|
||||||
|
ResourceEntry re = (ResourceEntry)this.IdEntries.get(i);
|
||||||
|
dataOffset = re.buildBuffer(buffer, virtualBaseOffset, dataOffset);
|
||||||
|
}
|
||||||
|
|
||||||
|
buffer.position(dataOffset);
|
||||||
|
return dataOffset;
|
||||||
|
}
|
||||||
|
|
||||||
|
public ResourceEntry getResourceEntry(String name) {
|
||||||
|
// If name == null, get the first entry in lexical
|
||||||
|
// order. If no entry in lexical order, choose the
|
||||||
|
// lowest integer id entry.
|
||||||
|
if (name == null) {
|
||||||
|
if (NamedEntries.size() > 0) {
|
||||||
|
return (PEResourceDirectory.ResourceEntry)NamedEntries.get(0);
|
||||||
|
}
|
||||||
|
if (IdEntries.size() > 0) {
|
||||||
|
return (PEResourceDirectory.ResourceEntry)IdEntries.get(0);
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
if ((name.length() > 0) && (name.charAt(0) == '#')) {
|
||||||
|
try {
|
||||||
|
String nb = name.substring(1);
|
||||||
|
int i = Integer.parseInt(nb);
|
||||||
|
return getResourceEntry(i);
|
||||||
|
} catch (Exception exc) {
|
||||||
|
exc.printStackTrace();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for (Iterator i = this.NamedEntries.iterator(); i.hasNext();) {
|
||||||
|
ResourceEntry re = (ResourceEntry)i.next();
|
||||||
|
if (name.equals(re.Name)) {
|
||||||
|
return re;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
public ResourceEntry getResourceEntry(int id) {
|
||||||
|
for (Iterator i = this.IdEntries.iterator(); i.hasNext();) {
|
||||||
|
ResourceEntry re = (ResourceEntry)i.next();
|
||||||
|
if (id == re.Id) {
|
||||||
|
return re;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
PESection m_master;
|
||||||
|
PEFile m_file;
|
||||||
|
long m_offsetBase;
|
||||||
|
|
||||||
|
PEResourceDirectory.ImageResourceDirectory m_root;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Creates a new instance of PEResourceDirectory
|
||||||
|
*/
|
||||||
|
public PEResourceDirectory(PEFile file, PESection sect) throws IOException {
|
||||||
|
m_master = sect;
|
||||||
|
m_file = file;
|
||||||
|
m_offsetBase = sect.PointerToRawData;
|
||||||
|
init();
|
||||||
|
|
||||||
|
// System.out.println("--------\nTOTAL SIZE: " + m_root.diskSize());
|
||||||
|
|
||||||
|
// System.out.println("\n\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
public void init() throws IOException {
|
||||||
|
// / System.out.println("RESOURCE INIT");
|
||||||
|
// System.out.println(" Offset: " + m_master.PointerToRawData);
|
||||||
|
FileChannel chan = m_file.getChannel();
|
||||||
|
chan.position(m_master.PointerToRawData);
|
||||||
|
PEResourceDirectory.ImageResourceDirectory dir = new PEResourceDirectory.ImageResourceDirectory(chan);
|
||||||
|
// System.out.println("-----------------\nDUMP\n---------------");
|
||||||
|
m_root = dir;
|
||||||
|
|
||||||
|
// dir.dump(System.out, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void dump(PrintStream out) {
|
||||||
|
m_root.dump(out, 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
public int size() {
|
||||||
|
return m_root.diskSize();
|
||||||
|
}
|
||||||
|
|
||||||
|
public ByteBuffer buildResource(long virtualBaseOffset) {
|
||||||
|
// System.out.println("BUILDING RESOURCE / VIRTUAL: " + virtualBaseOffset);
|
||||||
|
int resourceSize = m_root.diskSize();
|
||||||
|
ByteBuffer resbuf = ByteBuffer.allocate(resourceSize);
|
||||||
|
resbuf.order(ByteOrder.LITTLE_ENDIAN);
|
||||||
|
resbuf.position(0);
|
||||||
|
m_root.buildBuffer(resbuf, virtualBaseOffset);
|
||||||
|
return resbuf;
|
||||||
|
}
|
||||||
|
|
||||||
|
public PEResourceDirectory.ImageResourceDirectory getRoot() {
|
||||||
|
return m_root;
|
||||||
|
}
|
||||||
|
|
||||||
|
public boolean replaceManifest(int resourceId, int langId, ByteBuffer data) {
|
||||||
|
ResourceEntry catEntry = m_root.getResourceEntry(24);
|
||||||
|
if ((catEntry != null) && (catEntry.Directory != null)) {
|
||||||
|
ResourceEntry identEntry = catEntry.Directory.getResourceEntry(resourceId);
|
||||||
|
if ((identEntry != null) && (identEntry.Directory != null)) {
|
||||||
|
ResourceEntry langEntry = identEntry.Directory.getResourceEntry(langId);
|
||||||
|
if ((langEntry != null) && (langEntry.Data != null)) {
|
||||||
|
DataEntry dataslot = langEntry.Data;
|
||||||
|
dataslot.setData(data);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
public boolean replaceResource(String catId, int resourceId, int langId, ByteBuffer data) {
|
||||||
|
ResourceEntry catEntry = m_root.getResourceEntry(catId);
|
||||||
|
if ((catEntry != null) && (catEntry.Directory != null)) {
|
||||||
|
ResourceEntry identEntry = catEntry.Directory.getResourceEntry(resourceId);
|
||||||
|
if ((identEntry != null) && (identEntry.Directory != null)) {
|
||||||
|
ResourceEntry langEntry = identEntry.Directory.getResourceEntry(langId);
|
||||||
|
if ((langEntry != null) && (langEntry.Data != null)) {
|
||||||
|
DataEntry dataslot = langEntry.Data;
|
||||||
|
dataslot.setData(data);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void addNewResource(String catId, String resourceId, String languageId, ByteBuffer data) {
|
||||||
|
DataEntry dataEntry = new DataEntry(data);
|
||||||
|
ResourceEntry languageEntry = buildResourceEntry(languageId, dataEntry);
|
||||||
|
ImageResourceDirectory languageDir = new ImageResourceDirectory();
|
||||||
|
|
||||||
|
languageDir.TimeDateStamp = 0x3F2CCF64;
|
||||||
|
languageDir.addEntry(languageEntry);
|
||||||
|
|
||||||
|
ResourceEntry identEntry = buildResourceEntry(resourceId, languageDir);
|
||||||
|
|
||||||
|
ImageResourceDirectory identDir = new ImageResourceDirectory();
|
||||||
|
identDir.TimeDateStamp = 0x3F2CCF64;
|
||||||
|
identDir.addEntry(identEntry);
|
||||||
|
|
||||||
|
ResourceEntry catEntry = buildResourceEntry(catId, identDir);
|
||||||
|
m_root.addEntry(catEntry);
|
||||||
|
}
|
||||||
|
|
||||||
|
public DataEntry getData(String catId, String resourceId, String langId) {
|
||||||
|
ResourceEntry catEntry = m_root.getResourceEntry(catId);
|
||||||
|
if ((catEntry != null) && (catEntry.Directory != null)) {
|
||||||
|
ResourceEntry identEntry = catEntry.Directory.getResourceEntry(resourceId);
|
||||||
|
if ((identEntry != null) && (identEntry.Directory != null)) {
|
||||||
|
ResourceEntry langEntry = identEntry.Directory.getResourceEntry(langId);
|
||||||
|
if ((langEntry != null) && (langEntry.Data != null)) {
|
||||||
|
DataEntry dataslot = langEntry.Data;
|
||||||
|
return dataslot;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
public ResourceEntry buildResourceEntry(String id, DataEntry data) {
|
||||||
|
if ((id.length() > 1) && (id.charAt(0) == '#')) {
|
||||||
|
int intid = Integer.parseInt(id.substring(1));
|
||||||
|
return new ResourceEntry(intid, data);
|
||||||
|
}
|
||||||
|
|
||||||
|
return new ResourceEntry(id, data);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
public ResourceEntry buildResourceEntry(String id, ImageResourceDirectory dir) {
|
||||||
|
if ((id.length() > 1) && (id.charAt(0) == '#')) {
|
||||||
|
int intid = Integer.parseInt(id.substring(1));
|
||||||
|
return new ResourceEntry(intid, dir);
|
||||||
|
}
|
||||||
|
|
||||||
|
return new ResourceEntry(id, dir);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
@ -0,0 +1,133 @@
|
|||||||
|
/*
|
||||||
|
JSmooth: a VM wrapper toolkit for Windows
|
||||||
|
Copyright (C) 2003 Rodrigo Reyes <reyes@charabia.net>
|
||||||
|
|
||||||
|
This program is free software; you can redistribute it and/or modify
|
||||||
|
it under the terms of the GNU General Public License as published by
|
||||||
|
the Free Software Foundation; either version 2 of the License, or
|
||||||
|
(at your option) any later version.
|
||||||
|
|
||||||
|
This program is distributed in the hope that it will be useful,
|
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
GNU General Public License for more details.
|
||||||
|
|
||||||
|
You should have received a copy of the GNU General Public License
|
||||||
|
along with this program; if not, write to the Free Software
|
||||||
|
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
||||||
|
|
||||||
|
*/
|
||||||
|
|
||||||
|
/*
|
||||||
|
* PESection.java
|
||||||
|
*
|
||||||
|
* Created on 29 juillet 2003, 21:34
|
||||||
|
*/
|
||||||
|
|
||||||
|
package net.charabia.jsmoothgen.pe;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.PrintStream;
|
||||||
|
import java.nio.ByteBuffer;
|
||||||
|
import java.nio.ByteOrder;
|
||||||
|
import java.nio.channels.FileChannel;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author Rodrigo
|
||||||
|
*/
|
||||||
|
public class PESection implements Cloneable {
|
||||||
|
byte[] ANSI_Name; // Name of the Section. Can be anything (0)(8BYTES)
|
||||||
|
long VirtualSize; // The size of the section when it is mapped to memory. Must be a multiple of 4096. (8)(DWORD)
|
||||||
|
long VirtualAddress; // An rva to where it should be mapped in memory. (12)(DWORD)
|
||||||
|
long SizeOfRawData; // The size of the section in the PE file. Must be a multiple of 512 (16)(DWORD)
|
||||||
|
long PointerToRawData; // A file based offset which points to the location of this sections data (20)(DWORD)
|
||||||
|
long PointerToRelocations; // In EXE's this field is meaningless, and is set 0 (24)(DWORD)
|
||||||
|
long PointerToLinenumbers; // This is the file-based offset of the line number table. This field is only used for debug purposes, and is usualy set to 0 (28)(DWORD)
|
||||||
|
int NumberOfRelocations; // In EXE's this field is meaningless, and is set 0 (32)(WORD)
|
||||||
|
int NumberOfLinenumbers; // The number of line numbers in the line number table for this section. This field is only used for debug purposes, and is usualy set to 0 (34)(WORD)
|
||||||
|
long Characteristics; // The kind of data stored in this section ie. Code, Data, Import data, Relocation data (36)(DWORD)
|
||||||
|
|
||||||
|
private long m_baseoffset;
|
||||||
|
private PEFile m_pe;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Creates a new instance of PESection
|
||||||
|
*/
|
||||||
|
public PESection(PEFile pef, long baseoffset) {
|
||||||
|
m_pe = pef;
|
||||||
|
m_baseoffset = baseoffset;
|
||||||
|
}
|
||||||
|
|
||||||
|
public Object clone() throws CloneNotSupportedException {
|
||||||
|
return super.clone();
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getName() {
|
||||||
|
StringBuffer buffer = new StringBuffer();
|
||||||
|
for (int i = 0; i < 8; i++)
|
||||||
|
buffer.append((char) ANSI_Name[i]);
|
||||||
|
return buffer.toString();
|
||||||
|
}
|
||||||
|
|
||||||
|
public void read() throws IOException {
|
||||||
|
FileChannel ch = m_pe.getChannel();
|
||||||
|
ByteBuffer head = ByteBuffer.allocate(40);
|
||||||
|
head.order(ByteOrder.LITTLE_ENDIAN);
|
||||||
|
ch.position(m_baseoffset);
|
||||||
|
ch.read(head);
|
||||||
|
head.position(0);
|
||||||
|
|
||||||
|
ANSI_Name = new byte[8];
|
||||||
|
for (int i = 0; i < 8; i++)
|
||||||
|
ANSI_Name[i] = head.get();
|
||||||
|
|
||||||
|
VirtualSize = head.getInt();
|
||||||
|
VirtualAddress = head.getInt();
|
||||||
|
SizeOfRawData = head.getInt();
|
||||||
|
PointerToRawData = head.getInt();
|
||||||
|
PointerToRelocations = head.getInt();
|
||||||
|
PointerToLinenumbers = head.getInt();
|
||||||
|
NumberOfRelocations = head.getShort();
|
||||||
|
NumberOfLinenumbers = head.getShort();
|
||||||
|
Characteristics = head.getInt();
|
||||||
|
}
|
||||||
|
|
||||||
|
public void dump(PrintStream out) {
|
||||||
|
out.println("SECTION:");
|
||||||
|
out.println(" Name= "+getName());
|
||||||
|
out.println(" VirtualSize= " + VirtualSize + " // The size of the section when it is mapped to memory. Must be a multiple of 4096. (8)(DWORD)");
|
||||||
|
out.println(" VirtualAddress= " + VirtualAddress + " // An rva to where it should be mapped in memory. (12)(DWORD)");
|
||||||
|
out.println(" SizeOfRawData= " + SizeOfRawData + " // The size of the section in the PE file. Must be a multiple of 512 (16)(DWORD)");
|
||||||
|
out.println(" PointerToRawData= " + PointerToRawData + " // A file based offset which points to the location of this sections data (20)(DWORD)");
|
||||||
|
out.println(" PointerToRelocations= " + PointerToRelocations + " // In EXE's this field is meaningless, and is set 0 (24)(DWORD)");
|
||||||
|
out.println(" PointerToLinenumbers= " + PointerToLinenumbers + " // This is the file-based offset of the line number table. This field is only used for debug purposes, and is usualy set to 0 (28)(DWORD)");
|
||||||
|
out.println(" NumberOfRelocations= " + NumberOfRelocations + " // In EXE's this field is meaningless, and is set 0 (32)(WORD)");
|
||||||
|
out.println(" NumberOfLinenumbers= " + NumberOfLinenumbers + " // The number of line numbers in the line number table for this section. This field is only used for debug purposes, and is usualy set to 0 (34)(WORD)");
|
||||||
|
out.println(" Characteristics= " + Characteristics + " // The kind of data stored in this section ie. Code, Data, Import data, Relocation data (36)(DWORD)");
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
public ByteBuffer get() {
|
||||||
|
ByteBuffer head = ByteBuffer.allocate(40);
|
||||||
|
head.order(ByteOrder.LITTLE_ENDIAN);
|
||||||
|
head.position(0);
|
||||||
|
|
||||||
|
for (int i = 0; i < 8; i++)
|
||||||
|
head.put((byte)ANSI_Name[i]);
|
||||||
|
|
||||||
|
head.putInt((int)VirtualSize);
|
||||||
|
head.putInt((int)VirtualAddress);
|
||||||
|
head.putInt((int)SizeOfRawData);
|
||||||
|
head.putInt((int)PointerToRawData);
|
||||||
|
head.putInt((int)PointerToRelocations);
|
||||||
|
head.putInt((int)PointerToLinenumbers);
|
||||||
|
head.putShort((short)NumberOfRelocations);
|
||||||
|
head.putShort((short)NumberOfLinenumbers);
|
||||||
|
head.putInt((int)Characteristics);
|
||||||
|
|
||||||
|
head.position(0);
|
||||||
|
return head;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user