78 lines
2.0 KiB
Go
78 lines
2.0 KiB
Go
|
//go:build android && cgo
|
||
|
|
|
||
|
|
package main
|
||
|
|
|
||
|
|
/*
|
||
|
|
#include <stdlib.h>
|
||
|
|
|
||
|
|
typedef void (*release_object_func)(void *obj);
|
||
|
|
|
||
|
|
typedef void (*protect_func)(void *tun_interface, int fd);
|
||
|
|
|
||
|
|
typedef const char* (*resolve_process_func)(void *tun_interface, int protocol, const char *source, const char *target, int uid);
|
||
|
|
|
||
|
|
static void protect(protect_func fn, void *tun_interface, int fd) {
|
||
|
|
if (fn) {
|
||
|
|
fn(tun_interface, fd);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
static const char* resolve_process(resolve_process_func fn, void *tun_interface, int protocol, const char *source, const char *target, int uid) {
|
||
|
|
if (fn) {
|
||
|
|
return fn(tun_interface, protocol, source, target, uid);
|
||
|
|
}
|
||
|
|
return "";
|
||
|
|
}
|
||
|
|
|
||
|
|
static void release_object(release_object_func fn, void *obj) {
|
||
|
|
if (fn) {
|
||
|
|
return fn(obj);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
*/
|
||
|
|
import "C"
|
||
|
|
import (
|
||
|
|
"unsafe"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
globalCallbacks struct {
|
||
|
|
releaseObjectFunc C.release_object_func
|
||
|
|
protectFunc C.protect_func
|
||
|
|
resolveProcessFunc C.resolve_process_func
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
func protect(callback unsafe.Pointer, fd int) {
|
||
|
|
if globalCallbacks.protectFunc != nil {
|
||
|
|
C.protect(globalCallbacks.protectFunc, callback, C.int(fd))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func resolveProcess(callback unsafe.Pointer, protocol int, source, target string, uid int) string {
|
||
|
|
if globalCallbacks.resolveProcessFunc == nil {
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
s := C.CString(source)
|
||
|
|
defer C.free(unsafe.Pointer(s))
|
||
|
|
t := C.CString(target)
|
||
|
|
defer C.free(unsafe.Pointer(t))
|
||
|
|
res := C.resolve_process(globalCallbacks.resolveProcessFunc, callback, C.int(protocol), s, t, C.int(uid))
|
||
|
|
defer C.free(unsafe.Pointer(res))
|
||
|
|
return C.GoString(res)
|
||
|
|
}
|
||
|
|
|
||
|
|
func releaseObject(callback unsafe.Pointer) {
|
||
|
|
if globalCallbacks.releaseObjectFunc == nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
C.release_object(globalCallbacks.releaseObjectFunc, callback)
|
||
|
|
}
|
||
|
|
|
||
|
|
//export registerCallbacks
|
||
|
|
func registerCallbacks(markSocketFunc C.protect_func, resolveProcessFunc C.resolve_process_func, releaseObjectFunc C.release_object_func) {
|
||
|
|
globalCallbacks.protectFunc = markSocketFunc
|
||
|
|
globalCallbacks.resolveProcessFunc = resolveProcessFunc
|
||
|
|
globalCallbacks.releaseObjectFunc = releaseObjectFunc
|
||
|
|
}
|