Merge "Protect mtpd socket from VPN"

2 files changed, 6 insertions, 0 deletions

diff --git a/Android.bp b/Android.bp
index 7af119f..b865eec 100644
--- a/Android.bp
+++ b/Android.bp
@@ -23,6 +23,7 @@ cc_binary {
         "libcrypto",
         "libcutils",
         "liblog",
+        "libnetd_client",
     ],
     cflags: ["-DANDROID_CHANGES"],
     init_rc: ["mtpd.rc"],
diff --git a/mtpd.c b/mtpd.c
index c005567..ae0a1e6 100644
--- a/mtpd.c
+++ b/mtpd.c
@@ -35,6 +35,7 @@
 #endif
 
 #include "mtpd.h"
+#include "NetdClient.h"
 
 int the_socket = -1;
 
@@ -295,6 +296,10 @@ void create_socket(int family, int type, char *server, char *port)
         exit(NETWORK_ERROR);
     }
 
+#ifdef ANDROID_CHANGES
+    protectFromVpn(the_socket);
+#endif
+
     fcntl(the_socket, F_SETFD, FD_CLOEXEC);
     log_print(INFO, "Connection established (socket = %d)", the_socket);
 }