mirror of
https://github.com/akuker/RASCSI.git
synced 2024-10-11 05:23:39 +00:00
b7cb23e391
* Add statistics and make scsictl accept generic key/value parameters
388 lines
8.5 KiB
C++
388 lines
8.5 KiB
C++
//---------------------------------------------------------------------------
|
|
//
|
|
// SCSI Target Emulator PiSCSI
|
|
// for Raspberry Pi
|
|
//
|
|
// Copyright (C) 2021-2023 Uwe Seimet
|
|
//
|
|
//---------------------------------------------------------------------------
|
|
|
|
#include "shared/network_util.h"
|
|
#include "shared/piscsi_util.h"
|
|
#include "shared/protobuf_util.h"
|
|
#include "shared/piscsi_exceptions.h"
|
|
#include "scsictl_commands.h"
|
|
#include <sys/socket.h>
|
|
#include <arpa/inet.h>
|
|
#include <netinet/in.h>
|
|
#include <unistd.h>
|
|
#include <netdb.h>
|
|
#include <iostream>
|
|
#include <vector>
|
|
|
|
using namespace std;
|
|
using namespace piscsi_interface;
|
|
using namespace network_util;
|
|
using namespace piscsi_util;
|
|
using namespace protobuf_util;
|
|
|
|
bool ScsictlCommands::Execute(string_view log_level, string_view default_folder, string_view reserved_ids,
|
|
string_view image_params, string_view filename)
|
|
{
|
|
switch(command.operation()) {
|
|
case LOG_LEVEL:
|
|
return CommandLogLevel(log_level);
|
|
|
|
case DEFAULT_FOLDER:
|
|
return CommandDefaultImageFolder(default_folder);
|
|
|
|
case RESERVE_IDS:
|
|
return CommandReserveIds(reserved_ids);
|
|
|
|
case CREATE_IMAGE:
|
|
return CommandCreateImage(image_params);
|
|
|
|
case DELETE_IMAGE:
|
|
return CommandDeleteImage(image_params);
|
|
|
|
case RENAME_IMAGE:
|
|
return CommandRenameImage(image_params);
|
|
|
|
case COPY_IMAGE:
|
|
return CommandCopyImage(image_params);
|
|
|
|
case DEVICES_INFO:
|
|
return CommandDeviceInfo();
|
|
|
|
case DEVICE_TYPES_INFO:
|
|
return CommandDeviceTypesInfo();
|
|
|
|
case VERSION_INFO:
|
|
return CommandVersionInfo();
|
|
|
|
case SERVER_INFO:
|
|
return CommandServerInfo();
|
|
|
|
case DEFAULT_IMAGE_FILES_INFO:
|
|
return CommandDefaultImageFilesInfo();
|
|
|
|
case IMAGE_FILE_INFO:
|
|
return CommandImageFileInfo(filename);
|
|
|
|
case NETWORK_INTERFACES_INFO:
|
|
return CommandNetworkInterfacesInfo();
|
|
|
|
case LOG_LEVEL_INFO:
|
|
return CommandLogLevelInfo();
|
|
|
|
case RESERVED_IDS_INFO:
|
|
return CommandReservedIdsInfo();
|
|
|
|
case MAPPING_INFO:
|
|
return CommandMappingInfo();
|
|
|
|
case STATISTICS_INFO:
|
|
return CommandStatisticsInfo();
|
|
|
|
case OPERATION_INFO:
|
|
return CommandOperationInfo();
|
|
|
|
case NO_OPERATION:
|
|
return false;
|
|
|
|
default:
|
|
return SendCommand();
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool ScsictlCommands::SendCommand()
|
|
{
|
|
sockaddr_in server_addr = {};
|
|
if (!ResolveHostName(hostname, &server_addr)) {
|
|
throw io_exception("Can't resolve hostname '" + hostname + "'");
|
|
}
|
|
|
|
const int fd = socket(AF_INET, SOCK_STREAM, 0);
|
|
if (fd == -1) {
|
|
throw io_exception("Can't create socket: " + string(strerror(errno)));
|
|
}
|
|
|
|
server_addr.sin_port = htons(uint16_t(port));
|
|
if (connect(fd, (sockaddr *)&server_addr, sizeof(server_addr)) < 0) {
|
|
close(fd);
|
|
|
|
throw io_exception("Can't connect to piscsi on host '" + hostname + "', port " + to_string(port)
|
|
+ ": " + strerror(errno));
|
|
}
|
|
|
|
if (write(fd, "RASCSI", 6) != 6) {
|
|
close(fd);
|
|
|
|
throw io_exception("Can't write magic");
|
|
}
|
|
|
|
SerializeMessage(fd, command);
|
|
DeserializeMessage(fd, result);
|
|
|
|
close(fd);
|
|
|
|
if (!result.status()) {
|
|
throw io_exception(result.msg());
|
|
}
|
|
|
|
if (!result.msg().empty()) {
|
|
cout << result.msg() << endl;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ScsictlCommands::CommandDevicesInfo()
|
|
{
|
|
SendCommand();
|
|
|
|
cout << scsictl_display.DisplayDevicesInfo(result.devices_info()) << flush;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ScsictlCommands::CommandLogLevel(string_view log_level)
|
|
{
|
|
SetParam(command, "level", log_level);
|
|
|
|
return SendCommand();
|
|
}
|
|
|
|
bool ScsictlCommands::CommandReserveIds(string_view reserved_ids)
|
|
{
|
|
SetParam(command, "ids", reserved_ids);
|
|
|
|
return SendCommand();
|
|
}
|
|
|
|
bool ScsictlCommands::CommandCreateImage(string_view image_params)
|
|
{
|
|
if (!EvaluateParams(image_params, "file", "size")) {
|
|
cerr << "Error: Invalid file descriptor '" << image_params << "', format is NAME:SIZE" << endl;
|
|
|
|
return false;
|
|
}
|
|
|
|
SetParam(command, "read_only", "false");
|
|
|
|
return SendCommand();
|
|
}
|
|
|
|
bool ScsictlCommands::CommandDeleteImage(string_view filename)
|
|
{
|
|
SetParam(command, "file", filename);
|
|
|
|
return SendCommand();
|
|
}
|
|
|
|
bool ScsictlCommands::CommandRenameImage(string_view image_params)
|
|
{
|
|
if (!EvaluateParams(image_params, "from", "to")) {
|
|
cerr << "Error: Invalid file descriptor '" << image_params << "', format is CURRENT_NAME:NEW_NAME" << endl;
|
|
|
|
return false;
|
|
}
|
|
|
|
return SendCommand();
|
|
}
|
|
|
|
bool ScsictlCommands::CommandCopyImage(string_view image_params)
|
|
{
|
|
if (!EvaluateParams(image_params, "from", "to")) {
|
|
cerr << "Error: Invalid file descriptor '" << image_params << "', format is CURRENT_NAME:NEW_NAME" << endl;
|
|
|
|
return false;
|
|
}
|
|
|
|
return SendCommand();
|
|
}
|
|
|
|
bool ScsictlCommands::CommandDefaultImageFolder(string_view folder)
|
|
{
|
|
SetParam(command, "folder", folder);
|
|
|
|
return SendCommand();
|
|
}
|
|
|
|
bool ScsictlCommands::CommandDeviceInfo()
|
|
{
|
|
SendCommand();
|
|
|
|
for (const auto& device : result.devices_info().devices()) {
|
|
cout << scsictl_display.DisplayDeviceInfo(device);
|
|
}
|
|
|
|
cout << flush;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ScsictlCommands::CommandDeviceTypesInfo()
|
|
{
|
|
SendCommand();
|
|
|
|
cout << scsictl_display.DisplayDeviceTypesInfo(result.device_types_info()) << flush;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ScsictlCommands::CommandVersionInfo()
|
|
{
|
|
SendCommand();
|
|
|
|
cout << scsictl_display.DisplayVersionInfo(result.version_info()) << flush;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ScsictlCommands::CommandServerInfo()
|
|
{
|
|
SendCommand();
|
|
|
|
PbServerInfo server_info = result.server_info();
|
|
|
|
if (server_info.has_version_info()) {
|
|
cout << scsictl_display.DisplayVersionInfo(server_info.version_info());
|
|
}
|
|
|
|
if (server_info.has_log_level_info()) {
|
|
cout << scsictl_display.DisplayLogLevelInfo(server_info.log_level_info());
|
|
}
|
|
|
|
if (server_info.has_image_files_info()) {
|
|
cout << scsictl_display.DisplayImageFilesInfo(server_info.image_files_info());
|
|
}
|
|
|
|
if (server_info.has_mapping_info()) {
|
|
cout << scsictl_display.DisplayMappingInfo(server_info.mapping_info());
|
|
}
|
|
|
|
if (server_info.has_network_interfaces_info()) {
|
|
cout << scsictl_display.DisplayNetworkInterfaces(server_info.network_interfaces_info());
|
|
}
|
|
|
|
if (server_info.has_device_types_info()) {
|
|
cout << scsictl_display.DisplayDeviceTypesInfo(server_info.device_types_info());
|
|
}
|
|
|
|
if (server_info.has_reserved_ids_info()) {
|
|
cout << scsictl_display.DisplayReservedIdsInfo(server_info.reserved_ids_info());
|
|
}
|
|
|
|
if (server_info.has_statistics_info()) {
|
|
cout << scsictl_display.DisplayStatisticsInfo(server_info.statistics_info());
|
|
}
|
|
|
|
if (server_info.has_operation_info()) {
|
|
cout << scsictl_display.DisplayOperationInfo(server_info.operation_info());
|
|
}
|
|
|
|
if (server_info.has_devices_info() && server_info.devices_info().devices_size()) {
|
|
vector<PbDevice> sorted_devices = { server_info.devices_info().devices().begin(), server_info.devices_info().devices().end() };
|
|
ranges::sort(sorted_devices, [](const auto& a, const auto& b) { return a.id() < b.id() || a.unit() < b.unit(); });
|
|
|
|
cout << "Attached devices:\n";
|
|
|
|
for (const auto& device : sorted_devices) {
|
|
cout << scsictl_display.DisplayDeviceInfo(device);
|
|
}
|
|
}
|
|
|
|
cout << flush;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ScsictlCommands::CommandDefaultImageFilesInfo()
|
|
{
|
|
SendCommand();
|
|
|
|
cout << scsictl_display.DisplayImageFilesInfo(result.image_files_info()) << flush;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ScsictlCommands::CommandImageFileInfo(string_view filename)
|
|
{
|
|
SetParam(command, "file", filename);
|
|
|
|
SendCommand();
|
|
|
|
cout << scsictl_display.DisplayImageFile(result.image_file_info()) << flush;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ScsictlCommands::CommandNetworkInterfacesInfo()
|
|
{
|
|
SendCommand();
|
|
|
|
cout << scsictl_display.DisplayNetworkInterfaces(result.network_interfaces_info()) << flush;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ScsictlCommands::CommandLogLevelInfo()
|
|
{
|
|
SendCommand();
|
|
|
|
cout << scsictl_display.DisplayLogLevelInfo(result.log_level_info()) << flush;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ScsictlCommands::CommandReservedIdsInfo()
|
|
{
|
|
SendCommand();
|
|
|
|
cout << scsictl_display.DisplayReservedIdsInfo(result.reserved_ids_info()) << flush;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ScsictlCommands::CommandMappingInfo()
|
|
{
|
|
SendCommand();
|
|
|
|
cout << scsictl_display.DisplayMappingInfo(result.mapping_info()) << flush;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ScsictlCommands::CommandStatisticsInfo()
|
|
{
|
|
SendCommand();
|
|
|
|
cout << scsictl_display.DisplayStatisticsInfo(result.statistics_info()) << flush;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ScsictlCommands::CommandOperationInfo()
|
|
{
|
|
SendCommand();
|
|
|
|
cout << scsictl_display.DisplayOperationInfo(result.operation_info()) << flush;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ScsictlCommands::EvaluateParams(string_view image_params, const string& key1, const string& key2)
|
|
{
|
|
if (const auto& components = Split(string(image_params), COMPONENT_SEPARATOR, 2); components.size() == 2) {
|
|
SetParam(command, key1, components[0]);
|
|
SetParam(command, key2, components[1]);
|
|
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|