Compare commits
1 Commits
dbus
...
67ea99cd38
| Author | SHA1 | Date | |
|---|---|---|---|
|
67ea99cd38
|
16
lib/args.nim
16
lib/args.nim
@@ -4,11 +4,11 @@ import os
|
|||||||
|
|
||||||
type Args* = object
|
type Args* = object
|
||||||
name*: Option[string]
|
name*: Option[string]
|
||||||
cmd*: Option[string]
|
cmd*: Option[seq[string]]
|
||||||
profile*: Option[string]
|
profile*: Option[string]
|
||||||
|
|
||||||
proc getCmd*(args: Args): string =
|
proc getCmd*(args: Args): seq[string] =
|
||||||
return args.cmd.get(getEnv("SHELL", "/bin/bash"))
|
return args.cmd.get(@[getEnv("SHELL", "/bin/bash")])
|
||||||
|
|
||||||
proc getProfile*(args: Args): string =
|
proc getProfile*(args: Args): string =
|
||||||
if args.profile.isSome:
|
if args.profile.isSome:
|
||||||
@@ -18,8 +18,8 @@ proc getProfile*(args: Args): string =
|
|||||||
|
|
||||||
proc parseOpt(args: var Args, key: string, value: string): bool =
|
proc parseOpt(args: var Args, key: string, value: string): bool =
|
||||||
case key
|
case key
|
||||||
of "command", "c":
|
of "name", "n":
|
||||||
args.cmd = some(value)
|
args.name = some(value)
|
||||||
of "profile", "p":
|
of "profile", "p":
|
||||||
args.profile = some(value)
|
args.profile = some(value)
|
||||||
else:
|
else:
|
||||||
@@ -30,6 +30,7 @@ proc parseOpt(args: var Args, key: string, value: string): bool =
|
|||||||
proc parseArgs*(): Option[Args] =
|
proc parseArgs*(): Option[Args] =
|
||||||
var p = initOptParser()
|
var p = initOptParser()
|
||||||
var args = Args()
|
var args = Args()
|
||||||
|
var command = newSeq[string]()
|
||||||
|
|
||||||
while true:
|
while true:
|
||||||
p.next()
|
p.next()
|
||||||
@@ -40,6 +41,9 @@ proc parseArgs*(): Option[Args] =
|
|||||||
echo "Invalid argument ", p.val
|
echo "Invalid argument ", p.val
|
||||||
return
|
return
|
||||||
of cmdArgument:
|
of cmdArgument:
|
||||||
args.name = some(p.key.string)
|
command.add(p.key.string)
|
||||||
|
|
||||||
|
if command.len > 0:
|
||||||
|
args.cmd = some(command)
|
||||||
|
|
||||||
return some(args)
|
return some(args)
|
||||||
@@ -16,13 +16,6 @@ type Config* = object
|
|||||||
mountcwd*: Option[bool]
|
mountcwd*: Option[bool]
|
||||||
privileged*: Option[bool]
|
privileged*: Option[bool]
|
||||||
sethostname*: Option[bool]
|
sethostname*: Option[bool]
|
||||||
allowdri*: Option[bool]
|
|
||||||
dbus*: Option[bool]
|
|
||||||
dbussee*: Option[seq[string]]
|
|
||||||
dbustalk*: Option[seq[string]]
|
|
||||||
dbusown*: Option[seq[string]]
|
|
||||||
dbuscall*: Option[seq[string]]
|
|
||||||
dbusbroadcast*: Option[seq[string]]
|
|
||||||
|
|
||||||
proc applyConfig*(call: var BwrapCall, config: Config) =
|
proc applyConfig*(call: var BwrapCall, config: Config) =
|
||||||
for mount in config.mount.get(@[]):
|
for mount in config.mount.get(@[]):
|
||||||
@@ -46,19 +39,10 @@ proc extendConfig*(config: var Config): Config {.discardable.} =
|
|||||||
var eConf = loadConfig(getProfilePath(config.extends.unsafeGet))
|
var eConf = loadConfig(getProfilePath(config.extends.unsafeGet))
|
||||||
eConf.extendConfig()
|
eConf.extendConfig()
|
||||||
|
|
||||||
# todo: replace using macro / templates
|
|
||||||
config.mount = some(config.mount.get(@[]).concat(eConf.mount.get(@[])))
|
config.mount = some(config.mount.get(@[]).concat(eConf.mount.get(@[])))
|
||||||
config.romount = some(config.romount.get(@[]).concat(eConf.romount.get(@[])))
|
config.romount = some(config.romount.get(@[]).concat(eConf.romount.get(@[])))
|
||||||
config.symlinks = some(config.symlinks.get(@[]).concat(eConf.symlinks.get(@[])))
|
config.symlinks = some(config.symlinks.get(@[]).concat(eConf.symlinks.get(@[])))
|
||||||
config.mountcwd = some(config.mountcwd.get(eConf.mountcwd.get(false)))
|
config.mountcwd = some(config.mountcwd.get(eConf.mountcwd.get(false)))
|
||||||
config.sethostname = some(config.sethostname.get(eConf.sethostname.get(false)))
|
config.sethostname = some(config.sethostname.get(eConf.sethostname.get(false)))
|
||||||
config.allowdri = some(config.allowdri.get(eConf.allowdri.get(false)))
|
|
||||||
|
|
||||||
config.dbus = some(config.dbus.get(eConf.dbus.get(false)))
|
|
||||||
config.dbussee = some(config.dbussee.get(@[]).concat(eConf.dbussee.get(@[])))
|
|
||||||
config.dbustalk = some(config.dbustalk.get(@[]).concat(eConf.dbustalk.get(@[])))
|
|
||||||
config.dbusown = some(config.dbusown.get(@[]).concat(eConf.dbusown.get(@[])))
|
|
||||||
config.dbuscall = some(config.dbuscall.get(@[]).concat(eConf.dbuscall.get(@[])))
|
|
||||||
config.dbusbroadcast = some(config.dbusbroadcast.get(@[]).concat(eConf.dbusbroadcast.get(@[])))
|
|
||||||
|
|
||||||
return config
|
return config
|
||||||
|
|||||||
54
lib/dbus.nim
54
lib/dbus.nim
@@ -1,54 +0,0 @@
|
|||||||
import strformat
|
|
||||||
import options
|
|
||||||
import config
|
|
||||||
import osproc
|
|
||||||
import random
|
|
||||||
import os
|
|
||||||
|
|
||||||
type DbusProxy* = object
|
|
||||||
process*: Process
|
|
||||||
socket*: string
|
|
||||||
args: seq[string]
|
|
||||||
|
|
||||||
proc exec*(proxy: var DbusProxy) =
|
|
||||||
# todo: start dbus proxy in bwrap
|
|
||||||
# todo: pass arguments as fd
|
|
||||||
proxy.process = startProcess("xdg-dbus-proxy", args = proxy.args,
|
|
||||||
options = {poEchoCmd, poParentStreams, poUsePath})
|
|
||||||
|
|
||||||
proc startDBusProxy*(config: Config, hostname: string): DbusProxy =
|
|
||||||
let busPath = getEnv("DBUS_SESSION_BUS_ADDRESS")
|
|
||||||
let runtimeDir = getEnv("XDG_RUNTIME_DIR")
|
|
||||||
|
|
||||||
if busPath == "" or runtimeDir == "":
|
|
||||||
raise newException(IOError, "DBUS_SESSION_BUS_ADDRESS and XDG_RUNTIME_DIR are required")
|
|
||||||
|
|
||||||
let id = rand(1000)
|
|
||||||
let filterName = &"dbus-proxy-{hostname}-{id}"
|
|
||||||
|
|
||||||
var proxy = DbusProxy()
|
|
||||||
proxy.socket = &"{runtimeDir}/{filterName}"
|
|
||||||
|
|
||||||
proxy.args.add(busPath)
|
|
||||||
proxy.args.add(proxy.socket)
|
|
||||||
|
|
||||||
for name in config.dbussee.get(@[]):
|
|
||||||
proxy.args.add(&"--see={name}")
|
|
||||||
|
|
||||||
for name in config.dbustalk.get(@[]):
|
|
||||||
proxy.args.add(&"--talk={name}")
|
|
||||||
|
|
||||||
for name in config.dbuscall.get(@[]):
|
|
||||||
proxy.args.add(&"--call={name}")
|
|
||||||
|
|
||||||
for name in config.dbusown.get(@[]):
|
|
||||||
proxy.args.add(&"--own={name}")
|
|
||||||
|
|
||||||
for name in config.dbusbroadcast.get(@[]):
|
|
||||||
proxy.args.add(&"--broadcast={name}")
|
|
||||||
|
|
||||||
proxy.args.add("--filter")
|
|
||||||
proxy.args.add("--log")
|
|
||||||
proxy.exec()
|
|
||||||
|
|
||||||
proxy
|
|
||||||
@@ -1,12 +1,10 @@
|
|||||||
import strutils
|
import os
|
||||||
import options
|
|
||||||
import config
|
|
||||||
import utils
|
|
||||||
import bwrap
|
|
||||||
import args
|
import args
|
||||||
import json
|
import json
|
||||||
import dbus
|
import utils
|
||||||
import os
|
import bwrap
|
||||||
|
import config
|
||||||
|
import options
|
||||||
|
|
||||||
proc sandboxExec*(args: Args) =
|
proc sandboxExec*(args: Args) =
|
||||||
var call = BwrapCall()
|
var call = BwrapCall()
|
||||||
@@ -36,11 +34,10 @@ proc sandboxExec*(args: Args) =
|
|||||||
config.extendConfig()
|
config.extendConfig()
|
||||||
|
|
||||||
call
|
call
|
||||||
.addArg("--dev", "/dev")
|
.addMount("--dev-bind", "/dev/null")
|
||||||
.addMount("--dev-bind", "/dev/random")
|
.addMount("--dev-bind", "/dev/random")
|
||||||
.addMount("--dev-bind", "/dev/urandom")
|
.addMount("--dev-bind", "/dev/urandom")
|
||||||
.addArg("--tmpfs", "/tmp")
|
.addArg("--tmpfs", "/tmp")
|
||||||
.addArg("--tmpfs", "/dev/shm")
|
|
||||||
.addArg("--proc", "/proc")
|
.addArg("--proc", "/proc")
|
||||||
.addArg("--unshare-all")
|
.addArg("--unshare-all")
|
||||||
.addArg("--share-net")
|
.addArg("--share-net")
|
||||||
@@ -48,18 +45,6 @@ proc sandboxExec*(args: Args) =
|
|||||||
.addArg("--setenv", "BWSANDBOX", "1")
|
.addArg("--setenv", "BWSANDBOX", "1")
|
||||||
.applyConfig(config)
|
.applyConfig(config)
|
||||||
|
|
||||||
if config.dbus.get(false):
|
|
||||||
# todo: handle process and cleanup later
|
|
||||||
let proxy = startDBusProxy(config, hostname)
|
|
||||||
call.addArg("--ro-bind", proxy.socket,
|
|
||||||
getEnv("DBUS_SESSION_BUS_ADDRESS").split('=')[1])
|
|
||||||
|
|
||||||
# todo: use fd signaling instead of this
|
|
||||||
sleep(100)
|
|
||||||
|
|
||||||
if config.allowdri.get(false):
|
|
||||||
enableDri(call)
|
|
||||||
|
|
||||||
if config.mountcwd.get(false):
|
if config.mountcwd.get(false):
|
||||||
call
|
call
|
||||||
.addMount("--bind", getCurrentDir())
|
.addMount("--bind", getCurrentDir())
|
||||||
|
|||||||
@@ -1,8 +1,5 @@
|
|||||||
import strformat
|
|
||||||
import posix
|
|
||||||
import bwrap
|
|
||||||
import args
|
|
||||||
import os
|
import os
|
||||||
|
import args
|
||||||
|
|
||||||
const APP_NAME = "bwsandbox"
|
const APP_NAME = "bwsandbox"
|
||||||
|
|
||||||
@@ -26,26 +23,3 @@ proc getSandboxPath*(name: string): string =
|
|||||||
getDataDir()
|
getDataDir()
|
||||||
.joinPath(APP_NAME)
|
.joinPath(APP_NAME)
|
||||||
.joinPath(name)
|
.joinPath(name)
|
||||||
|
|
||||||
proc deviceExists(path: string): bool =
|
|
||||||
var res: Stat
|
|
||||||
return stat(path, res) >= 0 and S_ISCHR(res.st_mode)
|
|
||||||
|
|
||||||
# https://github.com/flatpak/flatpak/blob/1bdbb80ac57df437e46fce2cdd63e4ff7704718b/common/flatpak-run.c#L1496
|
|
||||||
proc enableDri*(call: var BwrapCall) =
|
|
||||||
const mounts = [
|
|
||||||
"/dev/dri", # general
|
|
||||||
"/dev/mali", "/dev/mali0", "/dev/umplock", # mali
|
|
||||||
"/dev/nvidiactl", "/dev/nvidia-modeset", # nvidia
|
|
||||||
"/dev/nvidia-uvm", "/dev/nvidia-uvm-tools" # nvidia OpenCl/CUDA
|
|
||||||
]
|
|
||||||
|
|
||||||
for mount in mounts:
|
|
||||||
if deviceExists(mount):
|
|
||||||
call.addMount("--dev-bind", mount)
|
|
||||||
|
|
||||||
for i in 0..20:
|
|
||||||
let device = &"/dev/nvidia{i}"
|
|
||||||
|
|
||||||
if deviceExists(device):
|
|
||||||
call.addMount("--dev-bind", device)
|
|
||||||
5
main.nim
5
main.nim
@@ -1,16 +1,15 @@
|
|||||||
import lib/sandbox
|
import lib/sandbox
|
||||||
import lib/args
|
import lib/args
|
||||||
import options
|
import options
|
||||||
import random
|
|
||||||
|
|
||||||
proc main(): int =
|
proc main(): int =
|
||||||
let args = parseArgs()
|
let args = parseArgs()
|
||||||
|
echo args
|
||||||
|
|
||||||
if args.isNone:
|
if args.isNone:
|
||||||
echo "Usage: bwshell --command=cmd --profile=profile <sandbox_name>"
|
echo "Usage: bwshell --name=sandbox_name --profile=profile <sandbox_cmd>"
|
||||||
return 1
|
return 1
|
||||||
else:
|
else:
|
||||||
randomize()
|
|
||||||
sandboxExec(args.unsafeGet)
|
sandboxExec(args.unsafeGet)
|
||||||
|
|
||||||
quit(main())
|
quit(main())
|
||||||
|
|||||||
24
scripts/applications.sh
Executable file
24
scripts/applications.sh
Executable file
@@ -0,0 +1,24 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
if [ $# -ne 1 ]; then
|
||||||
|
echo "Usage: $0 <target_dir>"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
check_dir() {
|
||||||
|
local dir=$1
|
||||||
|
local file
|
||||||
|
|
||||||
|
for application in "$dir/"*; do
|
||||||
|
file="$(basename "$application")"
|
||||||
|
|
||||||
|
sed "s/Exec=/Exec=bwrap --name='$file' --profile=gui /gi" "$application" > "$target/$file"
|
||||||
|
done
|
||||||
|
}
|
||||||
|
|
||||||
|
dirs=("/usr/share/applications" "$HOME/.local/share/applications")
|
||||||
|
target="$1"
|
||||||
|
|
||||||
|
for dir in "${dirs[@]}"; do
|
||||||
|
check_dir "$dir"
|
||||||
|
done
|
||||||
Reference in New Issue
Block a user