Method PerformCreateCommand

Src/Core/Profiler.cpp:675–731

///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
void FSecure::C3::Core::Profiler::Agent::PerformCreateCommand(json const& jCommandElement)
{
	auto profiler = m_Owner.lock();
	if (!profiler)
		return; // probably shutting down

	auto gateRelay = profiler->m_Gateway->m_Gateway.lock();
	if (!gateRelay)
		return; // probably shutting down

	auto route = gateRelay->FindRoute(m_Id);
	if (!route)
		throw std::runtime_error("Failed to find route to agent id = " + m_Id.ToString());

	auto commandWithArguments = base64::decode<ByteVector>(jCommandElement["Command"]["ByteForm"].get<std::string>());

	// check if command is a create device command -> then we need to assign deviceId and retreive its hash and repack commandWithArguments
	// [CommandId{CreateAndAttachDevice}(u16)][NewDeviceId(u16)][NewDeviceTypeHash(HashT)][arguments]
	auto commandReadView = ByteView{ commandWithArguments };
	auto commandId = commandReadView.Read<std::uint16_t>();
	auto const& createCommands = profiler->m_Gateway->m_CreateCommands;
	auto command = std::find_if(createCommands.cbegin(), createCommands.cend(), [commandId](Gateway::CreateCommand const& cc) { return cc.m_IsDevice && cc.m_Id == commandId; });

	// this function will only suport create commands
	if (command == createCommands.cend())
		return;

	// it is a create command
	DeviceId newDeviceId = ++m_LastDeviceId;
	ByteVector repacked;
	repacked.Write(Command::AddDevice, newDeviceId, command->m_IsNegotiableChannel, command->m_Hash);
	if (auto binder = profiler->GetBinderTo(command->m_Hash); binder && command->m_IsDevice) // peripheral, check if payload is needed.
	{
		auto connector = profiler->m_Gateway->m_Gateway.lock()->GetConnector(binder);
		if (!connector)
			throw std::runtime_error{ "Connector for requested peripheral is closed" };

		auto updatedArguments = connector->PeripheralCreationCommand(ByteVector::Create(RouteId{ route->m_RouteId.GetAgentId(), newDeviceId }), commandReadView, m_IsX64);
		repacked.Concat(updatedArguments);
	}
	else
	{
		repacked.Concat(commandReadView);
	}

	commandWithArguments = repacked;

	auto outgoingChannel = route->m_Channel.lock();
	if (!outgoingChannel)
		throw std::runtime_error("Tried to send command through dead channel"); // TODO maybe try through different route

	// push json with startup arguments to some container, use it if channel is created.
	AddScheduledDevice(newDeviceId, jCommandElement["Command"]);

	auto query = ProceduresG2X::RunCommandOnAgentQuery::Create(route->m_RouteId, gateRelay->m_Signature, m_EncryptionKey, gateRelay->m_DecryptionKey, commandWithArguments);
	gateRelay->LockAndSendPacket(query->ComposeQueryPacket(), outgoingChannel);
}