WIP: add cap to grpc

2 files changed, 45 insertions, 0 deletions

diff --git a/tunnel/src/main/java/com/wireguard/android/backend/GoBackend.java b/tunnel/src/main/java/com/wireguard/android/backend/GoBackend.java
index 79b724e8..2af43fb2 100644
--- a/tunnel/src/main/java/com/wireguard/android/backend/GoBackend.java
+++ b/tunnel/src/main/java/com/wireguard/android/backend/GoBackend.java
@@ -34,6 +34,8 @@ import com.google.protobuf.Empty;
 
 import com.wireguard.android.backend.BackendException.Reason;
 import com.wireguard.android.backend.Tunnel.State;
+import com.wireguard.android.backend.gen.CapabilitiesChangedRequest;
+import com.wireguard.android.backend.gen.CapabilitiesChangedResponse;
 import com.wireguard.android.backend.gen.DhcpRequest;
 import com.wireguard.android.backend.gen.DhcpResponse;
 import com.wireguard.android.backend.gen.GetConnectionOwnerUidResponse;
@@ -110,6 +112,7 @@ public final class GoBackend implements Backend {
     private static final String TAG = "WireGuard/GoBackend";
     private static final int STATS_TAG = 2;
     private static final int MSG_DHCP_EXPIRE = 1;
+    private static final int MSG_CAPABILITIES_CHANGED = 2;
     @Nullable private static AlwaysOnCallback alwaysOnCallback;
     private static GhettoCompletableFuture<VpnService> vpnService = new GhettoCompletableFuture<>();
     private final Context context;
@@ -125,6 +128,7 @@ public final class GoBackend implements Backend {
     @Nullable private Bgp bgp;
     private HandlerThread thread;
     private Handler handler;
+    private NetworkCapabilities activeNetworkCapabilities;
 
     /**
      * Public constructor for GoBackend.
@@ -363,6 +367,18 @@ public final class GoBackend implements Backend {
         return buf.toString();
     }
 
+    private void capabilitiesChanged(NetworkCapabilities capabilities) {
+        LibwgGrpc.LibwgBlockingStub stub = LibwgGrpc.newBlockingStub(channel);
+        CapabilitiesChangedRequest.Builder reqBuilder = CapabilitiesChangedRequest.newBuilder();
+
+        if (capabilities != null && capabilities.hasCapability(NetworkCapabilities.NET_CAPABILITY_NOT_METERED)) {
+            reqBuilder.addCapabilities(CapabilitiesChangedRequest.Capability.NOT_METERED);
+        }
+
+        CapabilitiesChangedResponse resp = stub.capabilitiesChanged(reqBuilder.build());
+        Log.i(TAG, "Capabilities change: " + resp.getError().getMessage());
+    }
+
     private int startHttpProxy(String pacFile) {
         LibwgGrpc.LibwgStub asyncStub = LibwgGrpc.newStub(channel);
         LibwgGrpc.LibwgBlockingStub stub = LibwgGrpc.newBlockingStub(channel);
@@ -386,6 +402,7 @@ public final class GoBackend implements Backend {
         StartHttpProxyRequest req = reqBuilder.build();
         StartHttpProxyResponse resp = stub.startHttpProxy(req);
         Log.i(TAG, "Start http proxy listen_port:" + resp.getListenPort() + ", error:" + resp.getError().getMessage());
+        capabilitiesChanged(activeNetworkCapabilities);
         streamer.start();
         return resp.getListenPort();
     }
@@ -755,6 +772,10 @@ public final class GoBackend implements Backend {
                                 Log.e(TAG, "DHCP failed: " + ex);
                             }
                             break;
+                        case MSG_CAPABILITIES_CHANGED:
+                            Log.w(TAG, "Msg: capabilities changed: " + activeNetworkCapabilities);
+                            capabilitiesChanged(activeNetworkCapabilities);
+                            break;
                         default:
                             Log.w(TAG, "Unknown message: " + msg.what);
                             break;
@@ -972,6 +993,16 @@ public final class GoBackend implements Backend {
                 }
             }
         }
+
+        @Override
+        public void onCapabilitiesChanged (Network network, NetworkCapabilities networkCapabilities) {
+            boolean hasCapNotMetered = networkCapabilities.hasCapability(NetworkCapabilities.NET_CAPABILITY_NOT_METERED);
+            Log.w(TAG, "onCapabilitiesChanged: " + network + " is not metered:" + (hasCapNotMetered));
+            if (network.equals(activeNetwork) && handler != null) {
+                activeNetworkCapabilities = networkCapabilities;
+                handler.sendEmptyMessage(MSG_CAPABILITIES_CHANGED);
+            }
+        }
     }
 
     private AlarmManager.OnAlarmListener alarmListener = new AlarmManager.OnAlarmListener() {
diff --git a/tunnel/src/main/proto/libwg.proto b/tunnel/src/main/proto/libwg.proto
index 3195690d..bf471599 100644
--- a/tunnel/src/main/proto/libwg.proto
+++ b/tunnel/src/main/proto/libwg.proto
@@ -18,6 +18,7 @@ service Libwg {
 	rpc Reverse(stream ReverseRequest) returns (stream ReverseResponse);
 	rpc IpcSet(IpcSetRequest) returns (IpcSetResponse);
 	rpc Dhcp(DhcpRequest) returns (DhcpResponse);
+        rpc CapabilitiesChanged(CapabilitiesChangedRequest) returns (CapabilitiesChangedResponse);
 }
 
 message TunnelHandle { int32 handle = 1; }
@@ -123,3 +124,16 @@ message DhcpResponse {
        Error error = 1;
        repeated Lease leases = 2;
 }
+
+message CapabilitiesChangedRequest {
+        enum Capability {
+                NONE = 0;
+                NOT_METERED = 11;
+        };
+
+        repeated Capability capabilities = 1;
+}
+
+message CapabilitiesChangedResponse {
+        Error error = 1;
+}