Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 3 additions & 2 deletions src/brand/bhyve/Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,9 @@
#

BRAND = bhyve
FILES = boot config.xml createzone init platform.xml socat support uninstall \
bootlib.py bundle.py uefi/__init__.py uefi/align.py uefi/vars.py
FILES = boot config.xml createzone init platform.xml qemu-monitor-cmd socat \
support uninstall bootlib.py bundle.py \
uefi/__init__.py uefi/align.py uefi/vars.py
BINS = init boot

PYMODULES = $(PYVERSIONS:%=modules/%)
Expand Down
202 changes: 202 additions & 0 deletions src/brand/bhyve/qemu-monitor-command
Original file line number Diff line number Diff line change
@@ -0,0 +1,202 @@
#!/bin/ksh -p
#
# This file and its contents are supplied under the terms of the
# Common Development and Distribution License ("CDDL"), version 1.0.
# You may only use this file in accordance with the terms of version
# 1.0 of the CDDL.
#
# A full copy of the text of the CDDL should have accompanied this
# source. A copy of the CDDL is also available via the Internet at
# http://www.illumos.org/license/CDDL.

# Copyright 2026 EFit Partners

_json_val() {
typeset v="$1"
case "$v" in
true|false|null)
print -- "$v" ;;
+([0-9])|+([0-9])\.+([0-9]))
print -- "$v" ;;
*)
v="${v//\\/\\\\}"; v="${v//\"/\\\"}"
print -- "\"${v}\"" ;;
esac
}

dot_to_json() {
typeset prefix="${1:-arguments}"
shift

typeset -A val cnt
typeset -a keys args
typeset arg key v json
typeset i=0 j=0 n c first=1

args=("$@")
n=${#args[@]}

while (( i < n )); do
arg="${args[i]}"
if [[ "$arg" == "--${prefix}."* ]]; then
key="${arg#--${prefix}.}"
(( i++ ))
if [[ -z "${cnt[$key]}" ]]; then
cnt[$key]=0 # explicit init before arithmetic
keys[${#keys[@]}]="$key"
fi
val["${key}.${cnt[$key]}"]="${args[i]}" # no :-0 needed anymore
(( cnt[$key] += 1 )) # clean arithmetic, no ${}
fi
(( i++ ))
done

json="{"
for key in "${keys[@]}"; do
c=${cnt[$key]}
(( first )) || json+=","
first=0
v="$key"; v="${v//\\/\\\\}"; v="${v//\"/\\\"}"
json+="\"${v}\":"
if (( c == 1 )); then
json+="$(_json_val "${val[${key}.0]}")"
else
json+="["
j=0
while (( j < c )); do
(( j )) && json+=","
json+="$(_json_val "${val[${key}.${j}]}")"
(( j++ ))
done
json+="]"
fi
done
json+="}"
print -- "$json"
}

get_zone_path() {
typeset zone="$1"
typeset zone_path
zone_path=$(zonecfg -z "${zone}" info zonepath 2>/dev/null | awk '{print $2}')
[[ -z ${zone_path} ]] && return 1
typeset socket_path
socket_path=$(zonecfg -z "${zone}" info attr name=com2 2>/dev/null | awk '/value/ {print $2}')
[[ -z ${socket_path} ]] && return 2
print "${zone_path}/root${socket_path#socket,}"
}

print_error() {
typeset zone=$1
typeset err_msg=$2
print -r -- "{\"error\":{\"class\":\"QA_CMD\",\"desc\":\"${1}: ${2}\"}}"
}

query_socket() {
typeset zone="$1"
typeset command=${2:-"guest-info"}
typeset args_json="${3:-}"
typeset socket_path
typeset retval

socket_path=$(get_zone_path "${zone}")
retval=$?
(( retval == 1 )) && print_error "${zone}" "No such zone configured" && return 3
(( retval == 2 )) && print_error "${zone}" "No com2 serial configured" && return 4

if [[ -n "$args_json" ]]; then
command='{"execute":"guest-'${command#guest-}'","arguments":'${args_json}'}'
else
command='{"execute":"guest-'${command#guest-}'"}'
fi

zoneadm -z "${zone}" list -v | grep running >/dev/null
(( $? != 0 )) && print_error "${zone}" "Not running" && return 5

echo "$command" | /usr/lib/brand/bhyve/socat "$socket_path" 0 -
}

usage() {
print "Usage: $0 -z|--zone <zone> -c|--command <command> (default: info)"
print " [-j|--json <json>] [--arguments.<key> <value> ...]"
print " [-f|--format]"
exit 2
}

ZONE=""
COMMAND=""
FORMAT=0
ARGS_JSON=""
typeset -a DOT_ARGS
typeset -a SILENT_COMMANDS=(shutdown suspend-disk suspend-ram suspend-hybrid)

while (( $# > 0 )); do
case "$1" in
-z|--zone)
[[ $# -ge 2 ]] || { print "Missing value for $1"; usage; }
ZONE="$2"
shift 2
;;
-c|--command)
[[ $# -ge 2 ]] || { print "Missing value for $1"; usage; }
COMMAND="$2"
shift 2
;;
-j|--json)
[[ $# -ge 2 ]] || { print "Missing value for $1"; usage; }
ARGS_JSON="$2"
shift 2
;;
--arguments.*)
[[ $# -ge 2 ]] || { print "Missing value for $1"; usage; }
DOT_ARGS+=("$1" "$2")
shift 2
;;
-f|--format)
if [[ -x /opt/ooce/bin/jq ]]; then
FORMAT=1
else
print "Please install ooce/util/jq to allow format option"
fi
shift
;;
-h|--help)
usage
;;
--)
shift
break
;;
-*)
print "Unknown option: $1"
usage
;;
*)
break
;;
esac
done

[[ -z "$ZONE" ]] && usage

# Dot notation takes precedence over --json if both are given
if (( ${#DOT_ARGS[@]} > 0 )); then
[[ -n "$ARGS_JSON" ]] && print "Warning: --json ignored, using --arguments.* dot notation"
ARGS_JSON=$(dot_to_json arguments "${DOT_ARGS[@]}")
fi

RESULT=$(query_socket "$ZONE" "$COMMAND" "$ARGS_JSON")
retval=$?

if [[ -n "${RESULT//[$'\r\n\t ']/}" ]]; then
if (( FORMAT == 0 )); then
print "$RESULT"
else
print "$RESULT" | /opt/ooce/bin/jq
fi
elif [[ " ${SILENT_COMMANDS[*]} " != *" ${COMMAND#guest-} "* ]]; then
print_error "${ZONE}" "Qemu Guest Agent does not seem to be running or configured"
exit 6
fi

exit ${retval}
146 changes: 92 additions & 54 deletions src/brand/bhyve/socat
100755 → 100644
Original file line number Diff line number Diff line change
Expand Up @@ -3,69 +3,107 @@
use strict;
use warnings;

use IO::Handle;
use IO::Socket::INET;
use IO::Socket::UNIX qw(SOCK_STREAM);
use IO::Select;

my ($sock, $port) = @ARGV;
my ($sock, $port, $client_input) = @ARGV;
$port //= 5900;
my $ip = '0.0.0.0';

my $iosel = IO::Select->new;
my %connection = ();
my $socket;
my $client;
if ($port <= 0 && !defined $client_input) {
die "ERROR: client_input is mandatory when port <= 0\n";
}

if ($port <= 0) {
my $socket = IO::Socket::UNIX->new(
Type => SOCK_STREAM,
Peer => $sock,
) or die "ERROR: cannot connect to $sock: $!\n";

my $sel = IO::Select->new($socket);

if ($client_input eq '-') {
# Bidirectional relay between STDIN and socket, like socat
my $stdin = IO::Handle->new->fdopen(fileno(STDIN), 'r');
$sel->add($stdin);

my $server = IO::Socket::INET->new(
LocalAddr => $ip,
LocalPort => $port,
ReuseAddr => 1,
Listen => 1,
while ($sel->count && (my @ready = $sel->can_read(1))) {
for my $fh (@ready) {
my $n = $fh->sysread(my $buf, 4096);
if (!$n) {
$sel->remove($fh);
next;
}
if ($fh == $stdin) {
$socket->syswrite($buf);
} else {
STDOUT->syswrite($buf);
}
}
}
} else {
# Send client_input then read response
$socket->syswrite($client_input);

while ($sel->can_read(1)) {
my $n = $socket->sysread(my $buf, 4096);
last unless $n;
STDOUT->syswrite($buf);
}
}

# 4. Close connection
$socket->close;
} else {
my $iosel = IO::Select->new;
my %connection = ();

my $ip = '0.0.0.0';
my $server = IO::Socket::INET->new(
LocalAddr => $ip,
LocalPort => $port,
ReuseAddr => 1,
Listen => 1,
) or die "ERROR: cannot listen on $ip:$port: $!\n";
$iosel->add($server);
print "Listening on $ip:$port...\n";

print "Listening on $ip:$port...\n";
$iosel->add($server);

while (1)
{
for my $ready ($iosel->can_read)
{
if ($ready == $server)
{
$socket = IO::Socket::UNIX->new(
Type => SOCK_STREAM,
Peer => $sock,
);
if (!$socket)
{
print "ERROR: cannot open socket $sock: $!\n";
next;
}
$iosel->add($socket);
while (1) {
for my $ready ($iosel->can_read) {
if ($ready == $server) {
my $client = $server->accept;
$iosel->add($client);

$client = $server->accept;
$iosel->add($client);
$connection{$client} = $socket;
$connection{$socket} = $client;
}
else
{
next if !exists $connection{$ready};
my $buffer;
if ($ready->sysread($buffer, 4096))
{
$connection{$ready}->syswrite($buffer);
}
else
{
$iosel->remove($client);
$iosel->remove($socket);
delete $connection{$client};
delete $connection{$socket};
$client->close;
$socket->close;
}
}
}
my $socket = IO::Socket::UNIX->new(
Type => SOCK_STREAM,
Peer => $sock,
);
if (!$socket) {
print STDERR "ERROR: cannot open socket $sock: $!\n";
$iosel->remove($client);
$client->close;
next;
}
$iosel->add($socket);
$connection{$client} = $socket;
$connection{$socket} = $client;
} else {
next if !exists $connection{$ready};
my $buffer;
if ($ready->sysread($buffer, 4096)) {
$connection{$ready}->syswrite($buffer);
} else {
my $peer = $connection{$ready};
$iosel->remove($ready);
$iosel->remove($peer);
delete $connection{$peer};
delete $connection{$ready};
$ready->close;
$peer->close;
}
}
}
}
}

1 change: 1 addition & 0 deletions src/pkg/manifests/system:zones:brand:bhyve.p5m
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,7 @@ file path=usr/lib/brand/bhyve/config.xml mode=0444
file path=usr/lib/brand/bhyve/createzone mode=0555
file path=usr/lib/brand/bhyve/init mode=0555
file path=usr/lib/brand/bhyve/platform.xml mode=0444
file path=usr/lib/brand/bhyve/qemu-monitor-cmd mode=0555
file path=usr/lib/brand/bhyve/socat mode=0555 pkg.depend.bypass-generate=.*
file path=usr/lib/brand/bhyve/support mode=0555
dir path=usr/lib/brand/bhyve/uefi
Expand Down