NetD : Allow passing in interface names for wifi/data app restriction

[android-x86/system-netd.git] / server / CommandListener.cpp

diff --git a/server/CommandListener.cpp b/server/CommandListener.cpp
index 5b284c4..b16da18 100755 (executable)
--- a/server/CommandListener.cpp
+++ b/server/CommandListener.cpp
@@ -1281,38 +1281,38 @@ int CommandListener::BandwidthControlCmd::runCommand(SocketClient *cli, int argc
     }
 
     if (!strcmp(argv[1], "addrestrictappsondata")) {
-        if (argc < 3) {
-            sendGenericSyntaxError(cli, "addrestrictappsondata <appUid> ...");
+        if (argc < 4) {
+            sendGenericSyntaxError(cli, "addrestrictappsondata <interface> <appUid> ...");
             return 0;
         }
-        int rc = gCtls->bandwidthCtrl.addRestrictAppsOnData(argc - 2, argv + 2);
+        int rc = gCtls->bandwidthCtrl.addRestrictAppsOnData(argv[2], argc - 3, argv + 3);
         sendGenericOkFail(cli, rc);
         return 0;
     }
     if (!strcmp(argv[1], "removerestrictappsondata")) {
-        if (argc < 3) {
-            sendGenericSyntaxError(cli, "removerestrictappsondata <appUid> ...");
+        if (argc < 4) {
+            sendGenericSyntaxError(cli, "removerestrictappsondata <interface> <appUid> ...");
             return 0;
         }
-        int rc = gCtls->bandwidthCtrl.removeRestrictAppsOnData(argc - 2, argv + 2);
+        int rc = gCtls->bandwidthCtrl.removeRestrictAppsOnData(argv[2], argc - 3, argv + 3);
         sendGenericOkFail(cli, rc);
         return 0;
     }
     if (!strcmp(argv[1], "addrestrictappsonwlan")) {
-        if (argc < 3) {
-            sendGenericSyntaxError(cli, "addrestrictappsonwlan <appUid> ...");
+        if (argc < 4) {
+            sendGenericSyntaxError(cli, "addrestrictappsonwlan <interface> <appUid> ...");
             return 0;
         }
-        int rc = gCtls->bandwidthCtrl.addRestrictAppsOnWlan(argc - 2, argv + 2);
+        int rc = gCtls->bandwidthCtrl.addRestrictAppsOnWlan(argv[2], argc - 3, argv + 3);
         sendGenericOkFail(cli, rc);
         return 0;
     }
     if (!strcmp(argv[1], "removerestrictappsonwlan")) {
-        if (argc < 3) {
-            sendGenericSyntaxError(cli, "removerestrictappsonwlan <appUid> ...");
+        if (argc < 4) {
+            sendGenericSyntaxError(cli, "removerestrictappsonwlan <inteface> <appUid> ...");
             return 0;
         }
-        int rc = gCtls->bandwidthCtrl.removeRestrictAppsOnWlan(argc - 2, argv + 2);
+        int rc = gCtls->bandwidthCtrl.removeRestrictAppsOnWlan(argv[2], argc - 3, argv + 3);
         sendGenericOkFail(cli, rc);
         return 0;
     }