1--
2-- SPDX-License-Identifier: BSD-2-Clause-FreeBSD
3--
4-- Copyright (c) 2018 Kyle Evans <kevans@FreeBSD.org>
5-- All rights reserved.
6--
7-- Redistribution and use in source and binary forms, with or without
8-- modification, are permitted provided that the following conditions
9-- are met:
10-- 1. Redistributions of source code must retain the above copyright
11--    notice, this list of conditions and the following disclaimer.
12-- 2. Redistributions in binary form must reproduce the above copyright
13--    notice, this list of conditions and the following disclaimer in the
14--    documentation and/or other materials provided with the distribution.
15--
16-- THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17-- ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18-- IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19-- ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20-- FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21-- DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22-- OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23-- HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24-- LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25-- OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26-- SUCH DAMAGE.
27--
28-- $FreeBSD: stable/11/stand/lua/cli.lua 360596 2020-05-03 03:53:38Z kevans $
29--
30
31local config = require("config")
32local core = require("core")
33
34local cli = {}
35
36-- Internal function
37-- Parses arguments to boot and returns two values: kernel_name, argstr
38-- Defaults to nil and "" respectively.
39-- This will also parse arguments to autoboot, but the with_kernel argument
40-- will need to be explicitly overwritten to false
41local function parseBootArgs(argv, with_kernel)
42	if with_kernel == nil then
43		with_kernel = true
44	end
45	if #argv == 0 then
46		if with_kernel then
47			return nil, ""
48		else
49			return ""
50		end
51	end
52	local kernel_name
53	local argstr = ""
54
55	for _, v in ipairs(argv) do
56		if with_kernel and v:sub(1,1) ~= "-" then
57			kernel_name = v
58		else
59			argstr = argstr .. " " .. v
60		end
61	end
62	if with_kernel then
63		return kernel_name, argstr
64	else
65		return argstr
66	end
67end
68
69-- Declares a global function cli_execute that attempts to dispatch the
70-- arguments passed as a lua function. This gives lua a chance to intercept
71-- builtin CLI commands like "boot"
72-- This function intentionally does not follow our general naming guideline for
73-- functions. This is global pollution, but the clearly separated 'cli' looks
74-- more like a module indicator to serve as a hint of where to look for the
75-- corresponding definition.
76function cli_execute(...)
77	local argv = {...}
78	-- Just in case...
79	if #argv == 0 then
80		return loader.command(...)
81	end
82
83	local cmd_name = argv[1]
84	local cmd = cli[cmd_name]
85	if cmd ~= nil and type(cmd) == "function" then
86		-- Pass argv wholesale into cmd. We could omit argv[0] since the
87		-- traditional reasons for including it don't necessarily apply,
88		-- it may not be totally redundant if we want to have one global
89		-- handling multiple commands
90		return cmd(...)
91	else
92		return loader.command(...)
93	end
94
95end
96
97function cli_execute_unparsed(str)
98	return cli_execute(loader.parse(str))
99end
100
101-- Module exports
102
103function cli.boot(...)
104	local _, argv = cli.arguments(...)
105	local kernel, argstr = parseBootArgs(argv)
106	if kernel ~= nil then
107		loader.perform("unload")
108		config.selectKernel(kernel)
109	end
110	core.boot(argstr)
111end
112
113function cli.autoboot(...)
114	local _, argv = cli.arguments(...)
115	local argstr = parseBootArgs(argv, false)
116	core.autoboot(argstr)
117end
118
119cli['boot-conf'] = function(...)
120	local _, argv = cli.arguments(...)
121	local kernel, argstr = parseBootArgs(argv)
122	if kernel ~= nil then
123		loader.perform("unload")
124		config.selectKernel(kernel)
125	end
126	core.autoboot(argstr)
127end
128
129cli['read-conf'] = function(...)
130	local _, argv = cli.arguments(...)
131	config.readConf(assert(core.popFrontTable(argv)))
132end
133
134cli['reload-conf'] = function(...)
135	config.reload()
136end
137
138-- Used for splitting cli varargs into cmd_name and the rest of argv
139function cli.arguments(...)
140	local argv = {...}
141	local cmd_name
142	cmd_name, argv = core.popFrontTable(argv)
143	return cmd_name, argv
144end
145
146return cli
147