Skip to content

Commit

Permalink
Merge pull request #30 from mysteriumnetwork/10-tunnel-status
Browse files Browse the repository at this point in the history 
10 tunnel status
  • Loading branch information
@tadaskay
tadaskay authored Aug 4, 2023
2 parents b05768c + 3c347c1 commit 6b36902
Show file tree
Hide file tree
Showing 27 changed files with 443 additions and 377 deletions.
2 changes: 2 additions & 0 deletions .gitignore
View file
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,8 @@ build/
# If you're building an application, you may want to check-in your pubspec.lock
pubspec.lock

.idea

# Directory created by dartdoc
# If you don't generate documentation locally you can remove this line.
doc/api/
Expand Down
15 changes: 15 additions & 0 deletions android/src/main/kotlin/network/mysterium/wireguard_dart/ConnectionStatus.kt
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
package network.mysterium.wireguard_dart

import com.wireguard.android.backend.Tunnel

enum class ConnectionStatus {
disconnected, connected, connecting, disconnecting, unknown;

companion object {
fun fromTunnelState(state: Tunnel.State?): ConnectionStatus = when (state) {
Tunnel.State.UP -> connected
Tunnel.State.DOWN -> disconnected
else -> unknown
}
}
}
25 changes: 25 additions & 0 deletions android/src/main/kotlin/network/mysterium/wireguard_dart/ConnectionStatusBroadcaster.kt
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,25 @@
package network.mysterium.wireguard_dart

import android.os.Handler
import android.os.Looper
import io.flutter.plugin.common.EventChannel
import io.flutter.plugin.common.EventChannel.EventSink
import io.flutter.plugin.common.EventChannel.StreamHandler

class ConnectionStatusBroadcaster : StreamHandler {

private var eventSink: EventSink? = null
override fun onListen(arguments: Any?, events: EventChannel.EventSink?) {
eventSink = events
}

override fun onCancel(arguments: Any?) {
eventSink = null
}

fun send(status: ConnectionStatus) {
Handler(Looper.getMainLooper()).post {
eventSink?.success(hashMapOf("status" to status.name))
}
}
}
194 changes: 95 additions & 99 deletions android/src/main/kotlin/network/mysterium/wireguard_dart/WireguardDartPlugin.kt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -17,28 +17,22 @@ import android.content.Context
import android.util.Log
import com.beust.klaxon.Klaxon
import com.wireguard.android.backend.*
import com.wireguard.crypto.Key
import com.wireguard.crypto.KeyPair
import io.flutter.plugin.common.EventChannel
import kotlinx.coroutines.*
import java.util.*


import kotlinx.coroutines.launch
import java.io.ByteArrayInputStream

/** WireguardDartPlugin */

const val PERMISSIONS_REQUEST_CODE = 10014
const val METHOD_CHANNEL_NAME = "wireguard_dart"

class WireguardDartPlugin : FlutterPlugin, MethodCallHandler, ActivityAware,
PluginRegistry.ActivityResultListener {
/// The MethodChannel that will the communication between Flutter and native Android
///
/// This local reference serves to register the plugin with the Flutter Engine and unregister it
/// when the Flutter Engine is detached from the Activity
PluginRegistry.ActivityResultListener {

private lateinit var channel: MethodChannel
private lateinit var tunnelName: String
private lateinit var statusChannel: EventChannel
private lateinit var statusBroadcaster: ConnectionStatusBroadcaster
private val futureBackend = CompletableDeferred<Backend>()
private val scope = CoroutineScope(Job() + Dispatchers.Main.immediate)
private var backend: Backend? = null
Expand All @@ -47,15 +41,21 @@ class WireguardDartPlugin : FlutterPlugin, MethodCallHandler, ActivityAware,
private var activity: Activity? = null
private var config: com.wireguard.config.Config? = null
private var tunnel: WireguardTunnel? = null

private var status: ConnectionStatus = ConnectionStatus.disconnected
set(value) {
field = value
if (::statusBroadcaster.isInitialized) {
statusBroadcaster.send(value)
}
}

companion object {
const val TAG = "MainActivity"
}

override fun onActivityResult(requestCode: Int, resultCode: Int, data: Intent?): Boolean {
havePermission =
(requestCode == PERMISSIONS_REQUEST_CODE) && (resultCode == Activity.RESULT_OK)
(requestCode == PERMISSIONS_REQUEST_CODE) && (resultCode == Activity.RESULT_OK)
return havePermission
}

Expand All @@ -76,7 +76,10 @@ class WireguardDartPlugin : FlutterPlugin, MethodCallHandler, ActivityAware,
}

override fun onAttachedToEngine(flutterPluginBinding: FlutterPlugin.FlutterPluginBinding) {
channel = MethodChannel(flutterPluginBinding.binaryMessenger, METHOD_CHANNEL_NAME)
channel = MethodChannel(flutterPluginBinding.binaryMessenger, "wireguard_dart")
statusChannel = EventChannel(flutterPluginBinding.binaryMessenger, "wireguard_dart.status")
statusBroadcaster = ConnectionStatusBroadcaster()
statusChannel.setStreamHandler(statusBroadcaster)
context = flutterPluginBinding.applicationContext

scope.launch(Dispatchers.IO) {
Expand Down Expand Up @@ -121,115 +124,109 @@ class WireguardDartPlugin : FlutterPlugin, MethodCallHandler, ActivityAware,
"setupTunnel" -> setupTunnel(call.argument<String>("bundleId").toString(), result)
"connect" -> connect(call.argument<String>("cfg").toString(), result)
"disconnect" -> disconnect(result)
"getStats" -> handleGetStats(call.arguments, result)
"status" -> status(result)
"statistics" -> statistics(result)
else -> flutterNotImplemented(result)
}
}

private fun handleGetStats(arguments: Any?, result: Result) {
val tunnelName = arguments?.toString()
if (tunnelName.isNullOrEmpty()) {
flutterError(result, "Tunnel has not been initialized")
return
}
private fun generateKeyPair(result: Result) {
val keyPair = KeyPair()
result.success(
hashMapOf(
"privateKey" to keyPair.privateKey.toBase64(),
"publicKey" to keyPair.publicKey.toBase64()
)
)
}

private fun setupTunnel(bundleId: String, result: Result) {
scope.launch(Dispatchers.IO) {
if (Tunnel.isNameInvalid(bundleId)) {
flutterError(result, "Invalid Name")
return@launch
}
checkPermission()
tunnel = WireguardTunnel(bundleId) { state ->
status = ConnectionStatus.fromTunnelState(state)
}
status = ConnectionStatus.fromTunnelState(backend?.getState(tunnel!!))
result.success(null)
}
}

private fun connect(cfg: String, result: Result) {
val tun = tunnel ?: run {
result.error("err_setup_tunnel", "Tunnel is not initialized", null)
return
}
status = ConnectionStatus.connecting
scope.launch(Dispatchers.IO) {
try {
val statistics = futureBackend.await().getStatistics(tunnel(tunnelName))
val stats = Stats(statistics.totalRx(), statistics.totalTx())

flutterSuccess(
result, Klaxon().toJsonString(
stats
)
)
Log.i(TAG, "Statistics - ${stats.totalDownload} ${stats.totalUpload}")

} catch (e: BackendException) {
Log.e(TAG, "Statistics - BackendException - ERROR - ${e.reason} ", e)
flutterError(result, e.reason.toString())
if (!havePermission) {
checkPermission()
throw Exception("Permissions are not given")
}
val inputStream = ByteArrayInputStream(cfg.toByteArray())
config = com.wireguard.config.Config.parse(inputStream)
futureBackend.await().setState(tun, Tunnel.State.UP, config);
flutterSuccess(result, "")
} catch (e: Throwable) {
Log.e(TAG, "Statistics - Can't get stats: ${e.message}", e)
Log.e(TAG, "Connect - Can't connect to tunnel: $e", e)
status = queryStatus()
flutterError(result, e.message.toString())
}
}
}

private fun disconnect(result: Result) {

val tun = tunnel ?: run {
result.error("err_setup_tunnel", "Tunnel is not initialized", null)
return
}
scope.launch(Dispatchers.IO) {
try {
if (futureBackend.await().runningTunnelNames.isEmpty()) {
throw Exception("Tunnel is not running")
}

futureBackend.await().setState(
tunnel(tunnelName) { state ->
scope.launch(Dispatchers.Main) {
Log.i(TAG, "onStateChange - $state")
channel.invokeMethod(
"onStateChange", state == Tunnel.State.UP
)
}
}, Tunnel.State.DOWN, config
)
Log.i(TAG, "Disconnect - success!")
futureBackend.await().setState(tun, Tunnel.State.DOWN, config)
flutterSuccess(result, "")
} catch (e: BackendException) {
Log.e(TAG, "Disconnect - BackendException - ERROR - ${e.reason}", e)
flutterError(result, e.reason.toString())
} catch (e: Throwable) {
Log.e(TAG, "Disconnect - Can't disconnect from tunnel: ${e.message}")
status = queryStatus()
flutterError(result, e.message.toString())
}
}
}

private fun connect(cfg: String, result: Result) {

private fun statistics(result: Result) {
val tun = tunnel ?: run {
result.error("err_setup_tunnel", "Tunnel is not initialized", null)
return
}
scope.launch(Dispatchers.IO) {
try {
if (!havePermission) {
checkPermission()
throw Exception("Permissions are not given")
}
val inputStream = ByteArrayInputStream(cfg.toByteArray())
config = com.wireguard.config.Config.parse(inputStream)
futureBackend.await().setState(
tunnel(tunnelName) { state ->
scope.launch(Dispatchers.Main) {
Log.i(TAG, "onStateChange - $state")
channel.invokeMethod(
"onStateChange", state == Tunnel.State.UP
)
}
}, Tunnel.State.UP, config
val statistics = futureBackend.await().getStatistics(tun)
val stats = Stats(statistics.totalRx(), statistics.totalTx())

flutterSuccess(
result, Klaxon().toJsonString(
stats
)
Log.i(TAG, "Connect - success!")
flutterSuccess(result, "")
)
Log.i(TAG, "Statistics - ${stats.totalDownload} ${stats.totalUpload}")

} catch (e: BackendException) {
Log.e(TAG, "Connect - BackendException - ERROR - ${e.reason}", e)
Log.e(TAG, "Statistics - BackendException - ERROR - ${e.reason} ", e)
flutterError(result, e.reason.toString())
} catch (e: Throwable) {
Log.e(TAG, "Connect - Can't connect to tunnel: $e", e)
Log.e(TAG, "Statistics - Can't get stats: ${e.message}", e)
flutterError(result, e.message.toString())
}
}
}

private fun setupTunnel(bundleId: String, result: Result) {
scope.launch(Dispatchers.IO) {
if (Tunnel.isNameInvalid(bundleId)) {
flutterError(result, "Invalid Name")
return@launch
}
tunnelName = bundleId
checkPermission()
result.success(null)
}
}

private fun checkPermission() {
val intent = GoBackend.VpnService.prepare(this.activity)
if (intent != null) {
Expand All @@ -240,32 +237,31 @@ class WireguardDartPlugin : FlutterPlugin, MethodCallHandler, ActivityAware,
}
}

private fun generateKeyPair(result: Result) {
val keyPair = KeyPair()
val privateKey = keyPair.privateKey.toBase64()
val publicKey = KeyPair(Key.fromBase64(privateKey)).publicKey.toBase64()
val map: HashMap<String, String> =
hashMapOf("privateKey" to privateKey, "publicKey" to publicKey)
result.success(map)
return
}

override fun onDetachedFromEngine(binding: FlutterPlugin.FlutterPluginBinding) {
channel.setMethodCallHandler(null)
}

private fun tunnel(name: String, callback: StateChangeCallback? = null): WireguardTunnel {
private fun status(result: Result) {
val status = queryStatus()
result.success(hashMapOf("status" to status.name))
}

private fun queryStatus(): ConnectionStatus {
if (tunnel == null) {
tunnel = WireguardTunnel(name, callback)
return ConnectionStatus.unknown
}
return when (backend?.getState(tunnel!!)) {
Tunnel.State.DOWN -> ConnectionStatus.disconnected
Tunnel.State.UP -> ConnectionStatus.connected
else -> ConnectionStatus.unknown
}
return tunnel as WireguardTunnel
}
}

typealias StateChangeCallback = (Tunnel.State) -> Unit

class WireguardTunnel(
private val name: String, private val onStateChanged: StateChangeCallback? = null
private val name: String, private val onStateChanged: StateChangeCallback? = null
) : Tunnel {

override fun getName() = name
Expand All @@ -277,8 +273,8 @@ class WireguardTunnel(
}

class Stats(
val totalDownload: Long,
val totalUpload: Long,
val totalDownload: Long,
val totalUpload: Long,
)


0 ios/.gitignore → darwin/.gitignore
View file
File renamed without changes.
0 ios/Assets/.gitkeep → darwin/Assets/.gitkeep
View file
File renamed without changes.
Loading

0 comments on commit 6b36902

Please sign in to comment.