android网络防火墙的实现iptables解决方案

Android 网络防火墙的实现 Iptables解决方案       

通过对Android SDK帮助文档的阅读，我没有发现Android的高层提供的API，于是通过更底层考虑，我发现了可以采用Iptables实现防火墙的功能。而且linux下主流的防火墙也是Iptables。

Iptables的介绍：

iptables是与最新的 2.6.x
版本 Linux 内核集成的 IP
信息包过滤系统。如果 Linux 系统连接到因特网或 LAN、服务器或连接 LAN
和因特网的代理服务器， 则该系统有利于在 Linux
系统上更好地控制 IP 信息包过滤和防火墙配置。
其工作原理:
netfilter/iptables IP 信息包过滤系统是一种功能强大的工具，
可用于添加、编辑和除去规则，这些规则是在做信息包过滤决定时，防火墙所遵循和组成的规则。这些规则存储在专用的信息包过滤表中，
而这些表集成在 Linux 内核中。
在信息包过滤表中，规则被分组放在我们所谓的 链（chain）中。我马上会详细讨论这些规则以及如何建立这些规则并将它们分组在链中。
虽然 netfilter/iptables IP
信息包过滤系统被称为单个实体，但它实际上由两个组件 netfilter和 iptables
组成。
netfilter 组件也称为
内核空间（kernelspace），是内核的一部分，由一些信息包过滤表组成，
这些表包含内核用来控制信息包过滤处理的规则集。
iptables组件是一种工具，也称为
用户空间（userspace），它使插入、修改和除去信息包过滤表中的规则变得容易。
除非您正在使用 Red Hat Linux 7.1 或更高版本，否则需要从 netfilter.org
下载该工具并安装使用它。
通过使用用户空间，可以构建自己的定制规则，这些规则存储在内核空间的信息包过滤表中。
这些规则具有 目标，它们告诉内核对来自某些源、前往某些目的地或具有某些协议类型的信息包做些什么。
如果某个信息包与规则匹配，那么使用目标 ACCEPT 允许该信息包通过。还可以使用目标 DROP
或 REJECT 来阻塞并杀死信息包。对于可对信息包执行的其它操作，还有许多其它目标。
根据规则所处理的信息包的类型，可以将规则分组在链中。处理入站信息包的规则被添加到 INPUT
链中。处理出站信息包的规则被添加到 OUTPUT 链中。处理正在转发的信息包的规则被添加到 FORWARD
链中。这三个链是基本信息包过滤表中内置的缺省主链。 另外，还有其它许多可用的链的类型（如 PREROUTING
和 POSTROUTING ），
以及提供用户定义的链。每个链都可以有一个 策略，
它定义“缺省目标”，也就是要执行的缺省操作，当信息包与链中的任何规则都不匹配时，执行此操作。
建立规则并将链放在适当的位置之后，就可以开始进行真正的信息包过滤工作了。
这时内核空间从用户空间接管工作。当信息包到达防火墙时，内核先 检查信息包的头信息，尤其是信息包的目的地。 我们将这个过程称为
路由。
如果信息包源自外界并前往系统，而且防火墙是打开的，那么内核将它传递到内核空间信息包过滤表的 INPUT
链。如果信息包源自系统内部或系统所连接的内部网上的其它源，并且此信息包要前往另一个外部系统， 那么信息包被传递到 OUTPUT
链。类似的，源自外部系统并前往外部系统的信息包被传递到 FORWARD
链。
接下来，将信息包的头信息与它所传递到的链中的每条规则进行比较，看它是否与某条规则完全匹配。
如果信息包与某条规则匹配，那么内核就对该信息包执行由该规则的目标指定的操作。 但是，如果信息包与这条规则不匹配，那么它将与链中的下一条规则进行比较。
最后，如果信息包与链中的任何规则都不匹配，那么内核将参考该链的策略来决定如何处理该信息包。 理想的策略应该告诉内核 DROP
该信息包。下图用图形说明了这个信息包过滤过程。

使用Iptables进行防火墙软件设计的解决方案。

由于Iptables已经有了完善的防火墙规则，我们只需要设计一个基于Iptables的Android前台即可，通过运行脚本，调用iptables设置防火墙规则即可。

大家可以学习下Droid wall的源代码

[java]
view plaincopyprint?

package com.googlecode.droidwall;

/**
* Contains shared programming interfaces.

* All iptables "communication" is handled by this class.

*
* Copyright (C) 2009 Rodrigo Zechin Rosauro

*
* 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 3 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, see <http://www.gnu.org/licenses/>.

*
* @author Rodrigo Zechin Rosauro

* @version 1.0
*/
import java.io.IOException;
import java.io.InputStreamReader;

import java.io.OutputStreamWriter;

import java.util.Arrays;

import java.util.HashMap;
import java.util.LinkedList;

import java.util.List;
import java.util.StringTokenizer;

import android.app.AlertDialog;

import android.content.Context;

import android.content.SharedPreferences;

import android.content.SharedPreferences.Editor;

import android.content.pm.ApplicationInfo;

import android.content.pm.PackageManager;

import android.util.Log;
/**
* Contains shared programming interfaces.

* All iptables "communication" is handled by this class.

*/
public final
class Api {
public static
final String VERSION = "1.3.6";

// Preferences
public static
final String PREFS_NAME = "DroidWallPrefs";

public static
final String PREF_ALLOWEDUIDS =
"AllowedUids";
public static
final String PREF_PASSWORD = "Password";

public static
final String PREF_MODE = "BlockMode";

public static
final String PREF_ITFS = "Interfaces";

// Modes
public static
final String MODE_WHITELIST = "whitelist";

public static
final String MODE_BLACKLIST = "blacklist";

// Interfaces
public static
final String ITF_3G = "2G/3G";

public static
final String ITF_WIFI = "Wi-fi";

// Cached applications

public static DroidApp applications[] =
null;
// Do we have "Wireless Tether for Root Users" installed?

public static String hastether =
null;
// Do we have root access?

private static
boolean hasroot = false;

/**
* Display a simple alert box
* @param ctx context
* @param msg message
*/
public static
void alert(Context ctx, CharSequence msg) {

if (ctx != null) {

new AlertDialog.Builder(ctx)

.setNeutralButton(android.R.string.ok,
null)
.setMessage(msg)
.show();
}
}
/**
* Purge and re-add all rules (internal implementation).

* @param ctx application context (mandatory)

* @param uids list of selected uids to allow or disallow (depending on the working mode)

* @param showErrors indicates if errors should be alerted

*/
private static
boolean applyIptablesRulesImpl(Context ctx, List<Integer> uids,
boolean showErrors) {
if (ctx == null) {

return false;

}
final SharedPreferences prefs = ctx.getSharedPreferences(PREFS_NAME,
0);
final boolean whitelist = prefs.getString(PREF_MODE, MODE_WHITELIST).equals(MODE_WHITELIST);

boolean wifi = false;
// Wi-fi selected ?
final String itfs = prefs.getString(PREF_ITFS, ITF_3G);

String itfFilter;
if (itfs.indexOf("|") != -1) {

itfFilter = ""; // Block all interfaces

wifi = true;
} else if (itfs.indexOf(ITF_3G) != -1) {

itfFilter = "-o rmnet+";;
// Block all rmnet interfaces
} else {
itfFilter = "-o tiwlan+";;
// Block all tiwlan interfaces
wifi = true;
}
final StringBuilder script =
new StringBuilder();
try {
int code;
script.append("iptables -F || exit/n");

final String targetRule = (whitelist ?
"ACCEPT" : "REJECT");

if (whitelist && wifi) {

// When "white listing" Wi-fi, we need ensure that the dhcp and wifi users are allowed

int uid = android.os.Process.getUidForName("dhcp");

if (uid != -1) script.append("iptables -A OUTPUT " + itfFilter +
" -m owner --uid-owner " + uid +
" -j ACCEPT || exit/n");
uid = android.os.Process.getUidForName("wifi");

if (uid != -1) script.append("iptables -A OUTPUT " + itfFilter +
" -m owner --uid-owner " + uid +
" -j ACCEPT || exit/n");
}
for (Integer uid : uids) {
script.append("iptables -A OUTPUT " + itfFilter +
" -m owner --uid-owner " + uid +
" -j " + targetRule + " || exit/n");

}
if (whitelist) {

script.append("iptables -A OUTPUT " + itfFilter +
" -j REJECT || exit/n");
}
StringBuilder res = new StringBuilder();

code = runScriptAsRoot(script.toString(), res);
if (showErrors && code != 0) {

String msg = res.toString();
Log.e("DroidWall", msg);
// Search for common error messages

if (msg.indexOf("Couldn't find match `owner'") != -1 || msg.indexOf("no chain/target match")
!= -1) {
alert(ctx, "Error applying iptables rules./nExit code: " + code +
"/n/n" +
"It seems your Linux kernel was not compiled with the netfilter /"owner/" module enabled, which is required for Droid Wall to work properly./n/n" +

"You should check if there is an updated version of your Android ROM compiled with this kernel module.");

} else {
// Remove unnecessary help message from output

if (msg.indexOf("/nTry `iptables -h' or 'iptables --help' for more information.") != -1) {

msg = msg.replace("/nTry `iptables -h' or 'iptables --help' for more information.",
"");
}
// Try `iptables -h' or 'iptables --help' for more information.

alert(ctx, "Error applying iptables rules. Exit code: " + code +
"/n/n" + msg.trim());
}
} else {
return true;

}
} catch (Exception e) {

if (showErrors) alert(ctx, "error refreshing iptables: " + e);

}
return false;

}
/**
* Purge and re-add all saved rules (not in-memory ones).

* This is much faster than just calling "applyIptablesRules", since it don't need to read installed applications.

* @param ctx application context (mandatory)

* @param showErrors indicates if errors should be alerted

*/
public static
boolean applySavedIptablesRules(Context ctx,
boolean showErrors) {
if (ctx == null) {

return false;

}
final String savedNames = ctx.getSharedPreferences(PREFS_NAME,
0).getString(PREF_ALLOWEDUIDS, "");

List<Integer> uids = new LinkedList<Integer>();

if (savedNames.length() > 0) {

// Check which applications are allowed

final StringTokenizer tok = new StringTokenizer(savedNames,
"|");
while (tok.hasMoreTokens()) {

uids.add(android.os.Process.getUidForName(tok.nextToken()));
}
}
return applyIptablesRulesImpl(ctx, uids, showErrors);

}
/**
* Purge and re-add all rules.
* @param ctx application context (mandatory)

* @param showErrors indicates if errors should be alerted

*/
public static
boolean applyIptablesRules(Context ctx,
boolean showErrors) {
if (ctx == null) {

return false;

}
final SharedPreferences prefs = ctx.getSharedPreferences(PREFS_NAME,
0);
List<Integer> uidsToApply = new LinkedList<Integer>();

final DroidApp[] apps = getApps(ctx);

// Builds a pipe-separated list of names

final StringBuilder newnames =
new StringBuilder();
for (int i=0; i<apps.length; i++) {

if (apps[i].selected) {
if (newnames.length() !=
0) newnames.append('|');

newnames.append(apps[i].username);
uidsToApply.add(apps[i].uid);
}
}
// save the new list of names if necessary

if (!newnames.toString().equals(prefs.getString(PREF_ALLOWEDUIDS,
""))) {
Editor edit = prefs.edit();
edit.putString(PREF_ALLOWEDUIDS, newnames.toString());

edit.commit();
}
return applyIptablesRulesImpl(ctx, uidsToApply, showErrors);

}
/**
* Purge all iptables rules.

* @param ctx context optional context for alert messages

* @return true if the rules were purged

*/
public static
boolean purgeIptables(Context ctx) {

StringBuilder res = new StringBuilder();

try {
int code = runScriptAsRoot("iptables -F || exit/n", res);

if (code != 0) {

alert(ctx, "error purging iptables. exit code: " + code +
"/n" + res);
return false;

}
return true;

} catch (Exception e) {
alert(ctx, "error purging iptables: " + e);

return false;

}
}
/**
* Display iptables rules output
* @param ctx application context

*/
public static
void showIptablesRules(Context ctx) {

try {
final StringBuilder res =
new StringBuilder();
runScriptAsRoot("iptables -L/n", res);

alert(ctx, res);
} catch (Exception e) {
alert(ctx, "error: " + e);

}
}
/**
* @param ctx application context (mandatory)

* @return a list of applications
*/
public static DroidApp[] getApps(Context ctx) {

if (applications !=
null) {
// return cached instance
return applications;

}
hastether = null;

// allowed application names separated by pipe '|' (persisted)

final String savedNames = ctx.getSharedPreferences(PREFS_NAME,
0).getString(PREF_ALLOWEDUIDS, "");

String allowed[];
if (savedNames.length() >
0) {
// Check which applications are allowed

final StringTokenizer tok =
new StringTokenizer(savedNames,
"|");
allowed = new String[tok.countTokens()];

for (int i=0; i<allowed.length; i++) {

allowed[i] = tok.nextToken();
}
// Sort the array to allow using "Arrays.binarySearch" later

Arrays.sort(allowed);
} else {
allowed = new String[0];

}
try {
final PackageManager pkgmanager = ctx.getPackageManager();

final List<ApplicationInfo> installed = pkgmanager.getInstalledApplications(0);

final HashMap<Integer, DroidApp> map =
new HashMap<Integer, DroidApp>();

String name;
DroidApp app;
for (final ApplicationInfo apinfo : installed) {

app = map.get(apinfo.uid);
name = pkgmanager.getApplicationLabel(apinfo).toString();

// Check for the tethering application (which causes conflicts with Droid Wall)

if (apinfo.packageName.equals("android.tether")) {

hastether = name;
}
if (app == null) {

app = new DroidApp();

app.uid = apinfo.uid;
app.username = pkgmanager.getNameForUid(apinfo.uid);

app.names = new String[] { name };

map.put(apinfo.uid, app);
} else {
final String newnames[] =
new String[app.names.length + 1];

System.arraycopy(app.names, 0, newnames,
0, app.names.length);
newnames[app.names.length] = name;
app.names = newnames;
}
// check if this application is allowed

if (!app.selected && Arrays.binarySearch(allowed, app.username) >=
0) {
app.selected = true;
}
}
/* add special applications to the list */

final DroidApp special[] = {

new DroidApp(android.os.Process.getUidForName("root"),
"root", "(Applications running as root)",
false),
new DroidApp(android.os.Process.getUidForName("media"),
"media", "Media server",
false),
};
for (int i=0; i<special.length; i++) {

app = special[i];
if (app.uid != -1 && !map.containsKey(app.uid)) {

// check if this application is allowed

if (Arrays.binarySearch(allowed, app.username) >=
0) {
app.selected = true;

}
map.put(app.uid, app);
}
}
applications = new DroidApp[map.size()];

int index = 0;

for (DroidApp application : map.values()) applications[index++] = application;

return applications;

} catch (Exception e) {
alert(ctx, "error: " + e);

}
return null;

}
/**
* Check if we have root access
* @param ctx optional context to display alert messages

* @return boolean true if we have root

*/
public static
boolean hasRootAccess(Context ctx) {

if (hasroot) return
true;
try {
// Run an empty script just to check root access

if (runScriptAsRoot("exit 0",
null, 20000) ==
0) {
hasroot = true;
return true;

}
} catch (Exception e) {
}
alert(ctx, "Could not acquire root access./n" +

"You need a rooted phone to run Droid Wall./n/n" +

"If this phone is already rooted, please make sure Droid Wall has enough permissions to execute the /"su/" command.");

return false;

}
/**
* Runs a script as root (multiple commands separated by "/n").

*
* @param script the script to be executed

* @param res the script output response (stdout + stderr)

* @param timeout timeout in milliseconds (-1 for none)

* @return the script exit code

*/
public static
int runScriptAsRoot(String script, StringBuilder res,
final long timeout) {

final ScriptRunner runner = new ScriptRunner(script, res);

runner.start();
try {
if (timeout > 0) {

runner.join(timeout);
} else {
runner.join();
}
if (runner.isAlive()) {
// Timed-out
runner.interrupt();
runner.destroy();
runner.join(50);
}
} catch (InterruptedException ex) {}

return runner.exitcode;

}
/**
* Runs a script as root (multiple commands separated by "/n") with a default timeout of 5 seconds.

*
* @param script the script to be executed

* @param res the script output response (stdout + stderr)

* @param timeout timeout in milliseconds (-1 for none)

* @return the script exit code

* @throws IOException on any error executing the script, or writing it to disk

*/
public static
int runScriptAsRoot(String script, StringBuilder res)
throws IOException {
return runScriptAsRoot(script, res,
15000);
}
/**
* Small structure to hold an application info

*/
public static
final class DroidApp {

/** linux user id */

int uid;
/** application user name (Android actually uses a package name to identify) */

String username;
/** application names belonging to this user id */

String names[];
/** indicates if this application is selected (checked) */

boolean selected;
/** toString cache */

String tostr;
public DroidApp() {

}
public DroidApp(int uid, String username, String name,
boolean selected) {
this.uid = uid;
this.username = username;

this.names = new String[] {name};

this.selected = selected;

}
/**
* Screen representation of this application

*/
@Override
public String toString() {

if (tostr == null) {

final StringBuilder s =
new StringBuilder(uid + ": ");

for (int i=0; i<names.length; i++) {

if (i != 0) s.append(", ");

s.append(names[i]);
}
tostr = s.toString();
}
return tostr;
}
}
/**
* Internal thread used to execute scripts as root.

*/
private static
final class ScriptRunner
extends Thread {
private final String script;

private final StringBuilder res;

public int exitcode = -1;

private Process exec;
/**
* Creates a new script runner.
* @param script script to run

* @param res response output
*/
public ScriptRunner(String script, StringBuilder res) {

this.script = script;

this.res = res;
}
@Override
public void run() {

try {
// Create the "su" request to run the command

// note that this will create a shell that we must interact to (using stdin/stdout)

exec = Runtime.getRuntime().exec("su");

final OutputStreamWriter out =
new OutputStreamWriter(exec.getOutputStream());

// Write the script to be executed

out.write(script);
// Ensure that the last character is an "enter"

if (!script.endsWith("/n")) out.write("/n");

out.flush();
// Terminate the "su" process

out.write("exit/n");

out.flush();
final char buf[] =
new char[1024];

// Consume the "stdout"
InputStreamReader r = new InputStreamReader(exec.getInputStream());

int read=0;

while ((read=r.read(buf)) != -1) {

if (res != null) res.append(buf,
0, read);
}
// Consume the "stderr"
r = new InputStreamReader(exec.getErrorStream());

read=0;
while ((read=r.read(buf)) != -1) {

if (res != null) res.append(buf,
0, read);
}
// get the process exit code

if (exec != null)
this.exitcode = exec.waitFor();

} catch (InterruptedException ex) {

if (res != null) res.append("/nOperation timed-out");

} catch (Exception ex) {
if (res != null) res.append("/n" + ex);

} finally {
destroy();
}
}
/**
* Destroy this script runner

*/
public synchronized
void destroy() {
if (exec != null) exec.destroy();

exec = null;
}
}
}

package com.googlecode.droidwall;
/**
* Contains shared programming interfaces.
* All iptables "communication" is handled by this class.
*
* Copyright (C) 2009 Rodrigo Zechin Rosauro
*
* 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 3 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, see <http://www.gnu.org/licenses/>.
*
* @author Rodrigo Zechin Rosauro
* @version 1.0
*/
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.Arrays;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.StringTokenizer;
import android.app.AlertDialog;
import android.content.Context;
import android.content.SharedPreferences;
import android.content.SharedPreferences.Editor;
import android.content.pm.ApplicationInfo;
import android.content.pm.PackageManager;
import android.util.Log;
/**
* Contains shared programming interfaces.
* All iptables "communication" is handled by this class.
*/
public final class Api {
public static final String VERSION = "1.3.6";
// Preferences
public static final String PREFS_NAME = "DroidWallPrefs";
public static final String PREF_ALLOWEDUIDS = "AllowedUids";
public static final String PREF_PASSWORD = "Password";
public static final String PREF_MODE = "BlockMode";
public static final String PREF_ITFS = "Interfaces";
// Modes
public static final String MODE_WHITELIST = "whitelist";
public static final String MODE_BLACKLIST = "blacklist";
// Interfaces
public static final String ITF_3G = "2G/3G";
public static final String ITF_WIFI = "Wi-fi";
// Cached applications
public static DroidApp applications[] = null;
// Do we have "Wireless Tether for Root Users" installed?
public static String hastether = null;
// Do we have root access?
private static boolean hasroot = false;
/**
* Display a simple alert box
* @param ctx context
* @param msg message
*/
public static void alert(Context ctx, CharSequence msg) {
if (ctx != null) {
new AlertDialog.Builder(ctx)
.setNeutralButton(android.R.string.ok, null)
.setMessage(msg)
.show();
}
}
/**
* Purge and re-add all rules (internal implementation).
* @param ctx application context (mandatory)
* @param uids list of selected uids to allow or disallow (depending on the working mode)
* @param showErrors indicates if errors should be alerted
*/
private static boolean applyIptablesRulesImpl(Context ctx, List<Integer> uids, boolean showErrors) {
if (ctx == null) {
return false;
}
final SharedPreferences prefs = ctx.getSharedPreferences(PREFS_NAME, 0);
final boolean whitelist = prefs.getString(PREF_MODE, MODE_WHITELIST).equals(MODE_WHITELIST);
boolean wifi = false; // Wi-fi selected ?
final String itfs = prefs.getString(PREF_ITFS, ITF_3G);
String itfFilter;
if (itfs.indexOf("|") != -1) {
itfFilter = ""; // Block all interfaces
wifi = true;
} else if (itfs.indexOf(ITF_3G) != -1) {
itfFilter = "-o rmnet+";; // Block all rmnet interfaces
} else {
itfFilter = "-o tiwlan+";; // Block all tiwlan interfaces
wifi = true;
}
final StringBuilder script = new StringBuilder();
try {
int code;
script.append("iptables -F || exit/n");
final String targetRule = (whitelist ? "ACCEPT" : "REJECT");
if (whitelist && wifi) {
// When "white listing" Wi-fi, we need ensure that the dhcp and wifi users are allowed
int uid = android.os.Process.getUidForName("dhcp");
if (uid != -1) script.append("iptables -A OUTPUT " + itfFilter + " -m owner --uid-owner " + uid + " -j ACCEPT || exit/n");
uid = android.os.Process.getUidForName("wifi");
if (uid != -1) script.append("iptables -A OUTPUT " + itfFilter + " -m owner --uid-owner " + uid + " -j ACCEPT || exit/n");
}
for (Integer uid : uids) {
script.append("iptables -A OUTPUT " + itfFilter + " -m owner --uid-owner " + uid + " -j " + targetRule + " || exit/n");
}
if (whitelist) {
script.append("iptables -A OUTPUT " + itfFilter + " -j REJECT || exit/n");
}
StringBuilder res = new StringBuilder();
code = runScriptAsRoot(script.toString(), res);
if (showErrors && code != 0) {
String msg = res.toString();
Log.e("DroidWall", msg);
// Search for common error messages
if (msg.indexOf("Couldn't find match `owner'") != -1 || msg.indexOf("no chain/target match") != -1) {
alert(ctx, "Error applying iptables rules./nExit code: " + code + "/n/n" +
"It seems your Linux kernel was not compiled with the netfilter /"owner/" module enabled, which is required for Droid Wall to work properly./n/n" +
"You should check if there is an updated version of your Android ROM compiled with this kernel module.");
} else {
// Remove unnecessary help message from output
if (msg.indexOf("/nTry `iptables -h' or 'iptables --help' for more information.") != -1) {
msg = msg.replace("/nTry `iptables -h' or 'iptables --help' for more information.", "");
}
// Try `iptables -h' or 'iptables --help' for more information.
alert(ctx, "Error applying iptables rules. Exit code: " + code + "/n/n" + msg.trim());
}
} else {
return true;
}
} catch (Exception e) {
if (showErrors) alert(ctx, "error refreshing iptables: " + e);
}
return false;
}
/**
* Purge and re-add all saved rules (not in-memory ones).
* This is much faster than just calling "applyIptablesRules", since it don't need to read installed applications.
* @param ctx application context (mandatory)
* @param showErrors indicates if errors should be alerted
*/
public static boolean applySavedIptablesRules(Context ctx, boolean showErrors) {
if (ctx == null) {
return false;
}
final String savedNames = ctx.getSharedPreferences(PREFS_NAME, 0).getString(PREF_ALLOWEDUIDS, "");
List<Integer> uids = new LinkedList<Integer>();
if (savedNames.length() > 0) {
// Check which applications are allowed
final StringTokenizer tok = new StringTokenizer(savedNames, "|");
while (tok.hasMoreTokens()) {
uids.add(android.os.Process.getUidForName(tok.nextToken()));
}
}
return applyIptablesRulesImpl(ctx, uids, showErrors);
}
/**
* Purge and re-add all rules.
* @param ctx application context (mandatory)
* @param showErrors indicates if errors should be alerted
*/
public static boolean applyIptablesRules(Context ctx, boolean showErrors) {
if (ctx == null) {
return false;
}
final SharedPreferences prefs = ctx.getSharedPreferences(PREFS_NAME, 0);
List<Integer> uidsToApply = new LinkedList<Integer>();
final DroidApp[] apps = getApps(ctx);
// Builds a pipe-separated list of names
final StringBuilder newnames = new StringBuilder();
for (int i=0; i<apps.length; i++) {
if (apps[i].selected) {
if (newnames.length() != 0) newnames.append('|');
newnames.append(apps[i].username);
uidsToApply.add(apps[i].uid);
}
}
// save the new list of names if necessary
if (!newnames.toString().equals(prefs.getString(PREF_ALLOWEDUIDS, ""))) {
Editor edit = prefs.edit();
edit.putString(PREF_ALLOWEDUIDS, newnames.toString());
edit.commit();
}
return applyIptablesRulesImpl(ctx, uidsToApply, showErrors);
}
/**
* Purge all iptables rules.
* @param ctx context optional context for alert messages
* @return true if the rules were purged
*/
public static boolean purgeIptables(Context ctx) {
StringBuilder res = new StringBuilder();
try {
int code = runScriptAsRoot("iptables -F || exit/n", res);
if (code != 0) {
alert(ctx, "error purging iptables. exit code: " + code + "/n" + res);
return false;
}
return true;
} catch (Exception e) {
alert(ctx, "error purging iptables: " + e);
return false;
}
}
/**
* Display iptables rules output
* @param ctx application context
*/
public static void showIptablesRules(Context ctx) {
try {
final StringBuilder res = new StringBuilder();
runScriptAsRoot("iptables -L/n", res);
alert(ctx, res);
} catch (Exception e) {
alert(ctx, "error: " + e);
}
}
/**
* @param ctx application context (mandatory)
* @return a list of applications
*/
public static DroidApp[] getApps(Context ctx) {
if (applications != null) {
// return cached instance
return applications;
}
hastether = null;
// allowed application names separated by pipe '|' (persisted)
final String savedNames = ctx.getSharedPreferences(PREFS_NAME, 0).getString(PREF_ALLOWEDUIDS, "");
String allowed[];
if (savedNames.length() > 0) {
// Check which applications are allowed
final StringTokenizer tok = new StringTokenizer(savedNames, "|");
allowed = new String[tok.countTokens()];
for (int i=0; i<allowed.length; i++) {
allowed[i] = tok.nextToken();
}
// Sort the array to allow using "Arrays.binarySearch" later
Arrays.sort(allowed);
} else {
allowed = new String[0];
}
try {
final PackageManager pkgmanager = ctx.getPackageManager();
final List<ApplicationInfo> installed = pkgmanager.getInstalledApplications(0);
final HashMap<Integer, DroidApp> map = new HashMap<Integer, DroidApp>();
String name;
DroidApp app;
for (final ApplicationInfo apinfo : installed) {
app = map.get(apinfo.uid);
name = pkgmanager.getApplicationLabel(apinfo).toString();
// Check for the tethering application (which causes conflicts with Droid Wall)
if (apinfo.packageName.equals("android.tether")) {
hastether = name;
}
if (app == null) {
app = new DroidApp();
app.uid = apinfo.uid;
app.username = pkgmanager.getNameForUid(apinfo.uid);
app.names = new String[] { name };
map.put(apinfo.uid, app);
} else {
final String newnames[] = new String[app.names.length + 1];
System.arraycopy(app.names, 0, newnames, 0, app.names.length);
newnames[app.names.length] = name;
app.names = newnames;
}
// check if this application is allowed
if (!app.selected && Arrays.binarySearch(allowed, app.username) >= 0) {
app.selected = true;
}
}
/* add special applications to the list */
final DroidApp special[] = {
new DroidApp(android.os.Process.getUidForName("root"), "root", "(Applications running as root)", false),
new DroidApp(android.os.Process.getUidForName("media"), "media", "Media server", false),
};
for (int i=0; i<special.length; i++) {
app = special[i];
if (app.uid != -1 && !map.containsKey(app.uid)) {
// check if this application is allowed
if (Arrays.binarySearch(allowed, app.username) >= 0) {
app.selected = true;
}
map.put(app.uid, app);
}
}
applications = new DroidApp[map.size()];
int index = 0;
for (DroidApp application : map.values()) applications[index++] = application;
return applications;
} catch (Exception e) {
alert(ctx, "error: " + e);
}
return null;
}
/**
* Check if we have root access
* @param ctx optional context to display alert messages
* @return boolean true if we have root
*/
public static boolean hasRootAccess(Context ctx) {
if (hasroot) return true;
try {
// Run an empty script just to check root access
if (runScriptAsRoot("exit 0", null, 20000) == 0) {
hasroot = true;
return true;
}
} catch (Exception e) {
}
alert(ctx, "Could not acquire root access./n" +
"You need a rooted phone to run Droid Wall./n/n" +
"If this phone is already rooted, please make sure Droid Wall has enough permissions to execute the /"su/" command.");
return false;
}
/**
* Runs a script as root (multiple commands separated by "/n").
*
* @param script the script to be executed
* @param res the script output response (stdout + stderr)
* @param timeout timeout in milliseconds (-1 for none)
* @return the script exit code
*/
public static int runScriptAsRoot(String script, StringBuilder res, final long timeout) {
final ScriptRunner runner = new ScriptRunner(script, res);
runner.start();
try {
if (timeout > 0) {
runner.join(timeout);
} else {
runner.join();
}
if (runner.isAlive()) {
// Timed-out
runner.interrupt();
runner.destroy();
runner.join(50);
}
} catch (InterruptedException ex) {}
return runner.exitcode;
}
/**
* Runs a script as root (multiple commands separated by "/n") with a default timeout of 5 seconds.
*
* @param script the script to be executed
* @param res the script output response (stdout + stderr)
* @param timeout timeout in milliseconds (-1 for none)
* @return the script exit code
* @throws IOException on any error executing the script, or writing it to disk
*/
public static int runScriptAsRoot(String script, StringBuilder res) throws IOException {
return runScriptAsRoot(script, res, 15000);
}
/**
* Small structure to hold an application info
*/
public static final class DroidApp {
/** linux user id */
int uid;
/** application user name (Android actually uses a package name to identify) */
String username;
/** application names belonging to this user id */
String names[];
/** indicates if this application is selected (checked) */
boolean selected;
/** toString cache */
String tostr;
public DroidApp() {
}
public DroidApp(int uid, String username, String name, boolean selected) {
this.uid = uid;
this.username = username;
this.names = new String[] {name};
this.selected = selected;
}
/**
* Screen representation of this application
*/
@Override
public String toString() {
if (tostr == null) {
final StringBuilder s = new StringBuilder(uid + ": ");
for (int i=0; i<names.length; i++) {
if (i != 0) s.append(", ");
s.append(names[i]);
}
tostr = s.toString();
}
return tostr;
}
}
/**
* Internal thread used to execute scripts as root.
*/
private static final class ScriptRunner extends Thread {
private final String script;
private final StringBuilder res;
public int exitcode = -1;
private Process exec;
/**
* Creates a new script runner.
* @param script script to run
* @param res response output
*/
public ScriptRunner(String script, StringBuilder res) {
this.script = script;
this.res = res;
}
@Override
public void run() {
try {
// Create the "su" request to run the command
// note that this will create a shell that we must interact to (using stdin/stdout)
exec = Runtime.getRuntime().exec("su");
final OutputStreamWriter out = new OutputStreamWriter(exec.getOutputStream());
// Write the script to be executed
out.write(script);
// Ensure that the last character is an "enter"
if (!script.endsWith("/n")) out.write("/n");
out.flush();
// Terminate the "su" process
out.write("exit/n");
out.flush();
final char buf[] = new char[1024];
// Consume the "stdout"
InputStreamReader r = new InputStreamReader(exec.getInputStream());
int read=0;
while ((read=r.read(buf)) != -1) {
if (res != null) res.append(buf, 0, read);
}
// Consume the "stderr"
r = new InputStreamReader(exec.getErrorStream());
read=0;
while ((read=r.read(buf)) != -1) {
if (res != null) res.append(buf, 0, read);
}
// get the process exit code
if (exec != null) this.exitcode = exec.waitFor();
} catch (InterruptedException ex) {
if (res != null) res.append("/nOperation timed-out");
} catch (Exception ex) {
if (res != null) res.append("/n" + ex);
} finally {
destroy();
}
}
/**
* Destroy this script runner
*/
public synchronized void destroy() {
if (exec != null) exec.destroy();
exec = null;
}
}
}

实质就是通过运行IPTABLES的脚本来实现防火墙规则的定制