Change MWR to FSecure in documentation

dependabot/npm_and_yarn/Src/WebController/UI/websocket-extensions-0.1.4
Grzegorz Rychlik 2020-03-05 17:09:11 +01:00
parent 28f464b282
commit 7e2586cbb5
3 changed files with 42 additions and 41 deletions

View File

@ -3,7 +3,7 @@
The following tutorials describe the process of C3 channel development in a step by step matter, along with detailed explanations and best practices. This section will demonstrate how to create a channel that utilizes regular files for communication. The following tutorials describe the process of C3 channel development in a step by step matter, along with detailed explanations and best practices. This section will demonstrate how to create a channel that utilizes regular files for communication.
## Basic Tutorial ## Basic Tutorial
This tutorial focuses only on a bare minimum effort required to develop a C3 Channel. First, open the C3 solution file in MS Visual Studio and add a new `*.cpp` file to the “Common” project. The Preferred location is: *“Src/Common/MWR/C3/Interfaces/Channels”* or its sub folder. This tutorial focuses only on a bare minimum effort required to develop a C3 Channel. First, open the C3 solution file in MS Visual Studio and add a new `*.cpp` file to the “Common” project. The Preferred location is: *“Src/Common/FSecure/C3/Interfaces/Channels”* or its sub folder.
![C3](Res/ContributionGuide/01.png) ![C3](Res/ContributionGuide/01.png)
@ -12,16 +12,16 @@ Next, open the newly created file and enter a type that will represent the Chann
#include "StdAfx.h" #include "StdAfx.h"
#include <fstream> #include <fstream>
namespace MWR::C3::Interfaces::Channels namespace FSecure::C3::Interfaces::Channels
{ {
class MyChannel : public MWR::C3::Interfaces::Channel<MyChannel> class MyChannel : public FSecure::C3::Interfaces::Channel<MyChannel>
{ {
public: public:
MyChannel(MWR::ByteView bv) : m_Filename("MyChannelFile") MyChannel(FSecure::ByteView bv) : m_Filename("MyChannelFile")
{ {
} }
size_t OnSendToChannel(MWR::ByteView packet) size_t OnSendToChannel(FSecure::ByteView packet)
{ {
// Create/open binary file with m_Filename // Create/open binary file with m_Filename
std::ofstream file(m_Filename, std::ios::binary); std::ofstream file(m_Filename, std::ios::binary);
@ -32,13 +32,13 @@ namespace MWR::C3::Interfaces::Channels
return packet.length(); return packet.length();
} }
MWR::ByteVector OnReceiveFromChannel() FSecure::ByteVector OnReceiveFromChannel()
{ {
// Open binary file with m_Filename // Open binary file with m_Filename
std::ifstream file{ m_Filename, std::ios::binary }; std::ifstream file{ m_Filename, std::ios::binary };
// read packet from file // read packet from file
return MWR::ByteVector{ std::istreambuf_iterator<char>{file}, {} }; return FSecure::ByteVector{ std::istreambuf_iterator<char>{file}, {} };
} }
private: private:
@ -52,7 +52,7 @@ A breakdown of this code is as such:
+ The first line is required by Visual Studio for projects using precompiled headers. + The first line is required by Visual Studio for projects using precompiled headers.
+ Channels must be defined in `Something::Interfaces::Channels` namespace. + Channels must be defined in `Something::Interfaces::Channels` namespace.
+ The `MyChannel` class definition publicly inherits from a *CRTP type* called `Channel<>`, which is required by the C3 framework in order to register the Channel. + The `MyChannel` class definition publicly inherits from a *CRTP type* called `Channel<>`, which is required by the C3 framework in order to register the Channel.
+ Next there are three required methods `OnSendToChannel`, `OnReceiveFromChannel` and a constructor that takes one `MWR::ByteView` argument. These methods are used to bind the channels functionality with the C3 framework. + Next there are three required methods `OnSendToChannel`, `OnReceiveFromChannel` and a constructor that takes one `FSecure::ByteView` argument. These methods are used to bind the channels functionality with the C3 framework.
+ For this basic tutorial the argument for the `MyChannel` constructor is ignored. Instead the file name member variable is initialised with a hardcoded value. + For this basic tutorial the argument for the `MyChannel` constructor is ignored. Instead the file name member variable is initialised with a hardcoded value.
+ `OnSendToChannel`'s implementation simply takes its parameter (packet) and writes it to the file. + `OnSendToChannel`'s implementation simply takes its parameter (packet) and writes it to the file.
+ `OnReceiveFromChannel` reads from the file and returns the buffer. + `OnReceiveFromChannel` reads from the file and returns the buffer.
@ -175,7 +175,7 @@ There are 2 parameters defined, both need to be at least 4 characters long and r
The next step in this tutorial involves updating `MyChannels` constructor to read the new arguments: The next step in this tutorial involves updating `MyChannels` constructor to read the new arguments:
```cpp ```cpp
MyChannel(MWR::ByteView bv) MyChannel(FSecure::ByteView bv)
: m_InFile(bv.Read<std::string>()) : m_InFile(bv.Read<std::string>())
, m_OutFile(bv.Read<std::string>()) , m_OutFile(bv.Read<std::string>())
{ {
@ -186,7 +186,7 @@ MyChannel(MWR::ByteView bv)
**Note:** The Read method “moves” `ByteView` according to the length of data that was read and allows for multiple parameters to be read at once (e.g. `auto[integer, text, real] = args.Read<int8_t, std::string, float>()`). **Note:** The Read method “moves” `ByteView` according to the length of data that was read and allows for multiple parameters to be read at once (e.g. `auto[integer, text, real] = args.Read<int8_t, std::string, float>()`).
**Note:** `std::string` and `MWR::ByteVector` types are stored in a buffer in a way that keeps track of their length (by preceding them with 4 byte-long size). **Note:** `std::string` and `FSecure::ByteVector` types are stored in a buffer in a way that keeps track of their length (by preceding them with 4 byte-long size).
**Note:** Arithmetic types are always stored in the little-endian byte order. **Note:** Arithmetic types are always stored in the little-endian byte order.
@ -264,7 +264,7 @@ ByteVector TestCommand(ByteView args)
} }
``` ```
The `TestCommand` method adds a debug text entry to C3s *Log*. `DebugInformation` is used which is rendered in a Relays console window only when compiled in debug mode. The message consists of two elements: a hard-coded text "Custom Command : " and a string that is provided by a user in the UI. Note the use of the `OBF` macro, this is used to encrypt strings at compile-time, which are decrypted at run-time only when required. Also of note is that the `TestCommand` handler returns an empty `MWR::ByteVector` as theres nothing to send back to the Gateway. The `TestCommand` method adds a debug text entry to C3s *Log*. `DebugInformation` is used which is rendered in a Relays console window only when compiled in debug mode. The message consists of two elements: a hard-coded text "Custom Command : " and a string that is provided by a user in the UI. Note the use of the `OBF` macro, this is used to encrypt strings at compile-time, which are decrypted at run-time only when required. Also of note is that the `TestCommand` handler returns an empty `FSecure::ByteVector` as theres nothing to send back to the Gateway.
**Note:** The `GetCapability` method is compiled only when building Gateways executable and therefore its not required to use the `OBF` macro to obfuscate its strings. **Note:** The `GetCapability` method is compiled only when building Gateways executable and therefore its not required to use the `OBF` macro to obfuscate its strings.
@ -277,29 +277,29 @@ After rebuilding the solution once again and launching C3 the Test Command becom
### Putting it all Together ### Putting it all Together
After developing a functionally working channel, the next stage is to refactor the code. The `MyChannel` type is still quite small, but for larger classes its common to split the implementation between a header and a compilation unit file. For `MyChannel` this involves adding a new *MyChannel.h* file in the same location as *MyChannel.cpp*. This also involves embracing the channel with the appropriate namespace declaration - `MWR::C3::Interfaces::Channels`. Finally, doxygen-style documentation is added to improve readability. *MyChannel.h* is then as such: After developing a functionally working channel, the next stage is to refactor the code. The `MyChannel` type is still quite small, but for larger classes its common to split the implementation between a header and a compilation unit file. For `MyChannel` this involves adding a new *MyChannel.h* file in the same location as *MyChannel.cpp*. This also involves embracing the channel with the appropriate namespace declaration - `FSecure::C3::Interfaces::Channels`. Finally, doxygen-style documentation is added to improve readability. *MyChannel.h* is then as such:
```cpp ```cpp
#pragma once #pragma once
namespace MWR::C3::Interfaces::Channels namespace FSecure::C3::Interfaces::Channels
{ {
/// Implementation of a File Channel. /// Implementation of a File Channel.
class MyChannel : public MWR::C3::Interfaces::Channel<MyChannel> class MyChannel : public FSecure::C3::Interfaces::Channel<MyChannel>
{ {
public: public:
/// Public constructor. /// Public constructor.
/// @param arguments factory arguments. /// @param arguments factory arguments.
MyChannel(MWR::ByteView arguments); MyChannel(FSecure::ByteView arguments);
/// OnSend callback implementation. Called every time attached Relay wants to send a packet through this Channel Device. @see Device::OnSendToChannelInternal. /// OnSend callback implementation. Called every time attached Relay wants to send a packet through this Channel Device. @see Device::OnSendToChannelInternal.
/// @param packet data to send through the Channel. /// @param packet data to send through the Channel.
/// @return number of bytes successfully sent through the Channel. One call does not have to send all data. In that case chunking will take place, Chunks must be at least 64 bytes or equal to packet.size() to be accepted, /// @return number of bytes successfully sent through the Channel. One call does not have to send all data. In that case chunking will take place, Chunks must be at least 64 bytes or equal to packet.size() to be accepted,
size_t OnSendToChannel(MWR::ByteView packet); size_t OnSendToChannel(FSecure::ByteView packet);
/// Reads a single C3 packet from Channel. Periodically called by attached Relay. Implementation should read the data (or return an empty buffer if there's nothing in the Channel waiting to read) and leave as soon as possible. /// Reads a single C3 packet from Channel. Periodically called by attached Relay. Implementation should read the data (or return an empty buffer if there's nothing in the Channel waiting to read) and leave as soon as possible.
/// @return ByteVector that contains a single packet retrieved from Channel. /// @return ByteVector that contains a single packet retrieved from Channel.
MWR::ByteVector OnReceiveFromChannel(); FSecure::ByteVector OnReceiveFromChannel();
/// Processes Commands addressed to this Channel. /// Processes Commands addressed to this Channel.
/// @param command a buffer containing whole command and it's parameters. /// @param command a buffer containing whole command and it's parameters.
@ -337,15 +337,15 @@ And the implementation within MyChannel.cpp:
#include <fstream> #include <fstream>
#include <filesystem> #include <filesystem>
namespace MWR::C3::Interfaces::Channels namespace FSecure::C3::Interfaces::Channels
{ {
MyChannel::MyChannel(MWR::ByteView bv) MyChannel::MyChannel(FSecure::ByteView bv)
: m_InFile(bv.Read<std::string>()) : m_InFile(bv.Read<std::string>())
, m_OutFile(bv.Read<std::string>()) , m_OutFile(bv.Read<std::string>())
{ {
} }
size_t MyChannel::OnSendToChannel(MWR::ByteView packet) size_t MyChannel::OnSendToChannel(FSecure::ByteView packet)
{ {
// Create/open binary file with m_OutFile name. // Create/open binary file with m_OutFile name.
std::ofstream file(m_OutFile, std::ios::binary); std::ofstream file(m_OutFile, std::ios::binary);
@ -356,12 +356,12 @@ namespace MWR::C3::Interfaces::Channels
return packet.length(); return packet.length();
} }
MWR::ByteVector MyChannel::OnReceiveFromChannel() FSecure::ByteVector MyChannel::OnReceiveFromChannel()
{ {
// Open binary file with m_InFile name. // Open binary file with m_InFile name.
std::ifstream file{ m_InFile, std::ios::binary }; std::ifstream file{ m_InFile, std::ios::binary };
// read packet from file // read packet from file
auto packet = MWR::ByteVector{ std::istreambuf_iterator<char>{file}, {} }; auto packet = FSecure::ByteVector{ std::istreambuf_iterator<char>{file}, {} };
file.close(); file.close();
std::filesystem::remove(m_OutFile); std::filesystem::remove(m_OutFile);
return packet; return packet;
@ -441,7 +441,7 @@ The tutorials provided here have discussed all the elements required to develop
+ `OnReceiveFromChannel` is called periodically, while `OnSendToChannel` is called only if theres something to send through the Channel. Bear in mind that calls to both methods could be made at the same time from two different threads. Implement appropriate synchronization mechanisms if they share any non-atomic objects. + `OnReceiveFromChannel` is called periodically, while `OnSendToChannel` is called only if theres something to send through the Channel. Bear in mind that calls to both methods could be made at the same time from two different threads. Implement appropriate synchronization mechanisms if they share any non-atomic objects.
+ `OnSendToChannel` returns the number of bytes successfully sent to the channel. If its less than the size of the provided packet, but more than 63, *automatic chunking* kicks in. If a channel sends less than 64 bytes, C3 will attempt to resend the packet and therefore call `OnSendToChannel` with the same buffer again, while the complementary Channel automatically rejects the packet. It follows from the above that it is necessary for every C3 Channel to be able to send chunks of size 64 bytes and more. For Channels that fail to meet this requirement, C3 will be stuck useless. This "magic" value of 64 is called *C3 Minimal MTU*. + `OnSendToChannel` returns the number of bytes successfully sent to the channel. If its less than the size of the provided packet, but more than 63, *automatic chunking* kicks in. If a channel sends less than 64 bytes, C3 will attempt to resend the packet and therefore call `OnSendToChannel` with the same buffer again, while the complementary Channel automatically rejects the packet. It follows from the above that it is necessary for every C3 Channel to be able to send chunks of size 64 bytes and more. For Channels that fail to meet this requirement, C3 will be stuck useless. This "magic" value of 64 is called *C3 Minimal MTU*.
+ `OnReceiveFromChannel` can have one of two signatures: + `OnReceiveFromChannel` can have one of two signatures:
+ `MWR::ByteVector OnReceiveFromChannel();` if it can retrieve only one packet at a time. + `FSecure::ByteVector OnReceiveFromChannel();` if it can retrieve only one packet at a time.
+ `std::vector<MWR::ByteVector> OnReceiveFromChannel();` if it can retrieve multiple packets at a time (eg. a whole page of messages in a single HTTP request) + `std::vector<FSecure::ByteVector> OnReceiveFromChannel();` if it can retrieve multiple packets at a time (eg. a whole page of messages in a single HTTP request)
In case there is nothing to read Channel should return an empty container. In case there is nothing to read Channel should return an empty container.
+ Both methods should perform the minimal number of operations and leave as soon as possible. There is a possibility to add single-threaded Relays to C3 in the future, which means that only one thread of execution will be shared between all interfaces and all their methods. Bear in mind that one stalling Channel will block the whole Relay, until this method returns. Thats why if a Channels methods (especially updates - `OnReceiveFromChannel`) require more time to execute, the implementation should be split into smaller pieces, *Update Delay Jitter* should be increased, and those pieces invoked in an interlaced manner. + Both methods should perform the minimal number of operations and leave as soon as possible. There is a possibility to add single-threaded Relays to C3 in the future, which means that only one thread of execution will be shared between all interfaces and all their methods. Bear in mind that one stalling Channel will block the whole Relay, until this method returns. Thats why if a Channels methods (especially updates - `OnReceiveFromChannel`) require more time to execute, the implementation should be split into smaller pieces, *Update Delay Jitter* should be increased, and those pieces invoked in an interlaced manner.

View File

@ -37,6 +37,7 @@ The most commonly used terms in C3:
BSD 3-Clause License BSD 3-Clause License
Copyright (c) 2019-2020, F-Secure
Copyright (c) 2018-2019, MWR Infosecurity Copyright (c) 2018-2019, MWR Infosecurity
All rights reserved. All rights reserved.

View File

@ -28,12 +28,12 @@ At a high level, the objective of integration would result in the communication
## Getting Started - Code Reuse is Good ## Getting Started - Code Reuse is Good
The first step for integrating Covenant involves creating the new source files within C3. Initially this is performed by copying the the code from `/src/Common/MWR/C3/Interfaces/Connectors/TeamServer.cpp` to `Covenant.cpp`. The first step for integrating Covenant involves creating the new source files within C3. Initially this is performed by copying the the code from `/src/Common/FSecure/C3/Interfaces/Connectors/TeamServer.cpp` to `Covenant.cpp`.
After replacing references to TeamServer, we end up with a number of methods that we simply need to redesign to work with Covenant. Highlighted below are the main functions which require alteration. After replacing references to TeamServer, we end up with a number of methods that we simply need to redesign to work with Covenant. Highlighted below are the main functions which require alteration.
```c++ ```c++
namespace MWR::C3::Interfaces::Connectors namespace FSecure::C3::Interfaces::Connectors
{ {
/// A class representing communication with Covenant. /// A class representing communication with Covenant.
struct Covenant : Connector<Covenant> struct Covenant : Connector<Covenant>
@ -46,17 +46,17 @@ namespace MWR::C3::Interfaces::Connectors
<...Snipped...> <...Snipped...>
} }
MWR::C3::Interfaces::Connectors::Covenant::Covenant(ByteView arguments) FSecure::C3::Interfaces::Connectors::Covenant::Covenant(ByteView arguments)
{ {
<...Snipped...> <...Snipped...>
} }
MWR::ByteVector MWR::C3::Interfaces::Connectors::Covenant::GeneratePayload(ByteView binderId, std::string pipename, uint32_t delay, uint32_t jitter, uint32_t connectAttempts) FSecure::ByteVector FSecure::C3::Interfaces::Connectors::Covenant::GeneratePayload(ByteView binderId, std::string pipename, uint32_t delay, uint32_t jitter, uint32_t connectAttempts)
{ {
<...Snipped...> <...Snipped...>
} }
MWR::ByteView MWR::C3::Interfaces::Connectors::Covenant::GetCapability() FSecure::ByteView FSecure::C3::Interfaces::Connectors::Covenant::GetCapability()
{ {
<...Snipped...> <...Snipped...>
} }
@ -78,7 +78,7 @@ The first function to be implemented is `GetCapability`, this is the form that u
This is achieved through the code below: This is achieved through the code below:
```c++ ```c++
MWR::ByteView MWR::C3::Interfaces::Connectors::Covenant::GetCapability() FSecure::ByteView FSecure::C3::Interfaces::Connectors::Covenant::GetCapability()
{ {
return R"( return R"(
{ {
@ -144,7 +144,7 @@ The resulting form is shown in the next figure. Note that no changes were made t
Once this form is submitted the only thing we need to know is that the connector's constructor is called and passed a `ByteView` containing the user supplied arguments. As well as adding the form above, we also made sure to create member variables for the Covenant connector object that will store the user supplied arguments. As such, when the constructor is called we are able to retrieve and store the arguments by simply doing: Once this form is submitted the only thing we need to know is that the connector's constructor is called and passed a `ByteView` containing the user supplied arguments. As well as adding the form above, we also made sure to create member variables for the Covenant connector object that will store the user supplied arguments. As such, when the constructor is called we are able to retrieve and store the arguments by simply doing:
```c++ ```c++
MWR::C3::Interfaces::Connectors::Covenant::Covenant(ByteView arguments) FSecure::C3::Interfaces::Connectors::Covenant::Covenant(ByteView arguments)
{ {
std::tie(m_ListeningPostPort, m_webHost, m_username, m_password) = arguments.Read<uint16_t, std::string, std::string, std::string>(); std::tie(m_ListeningPostPort, m_webHost, m_username, m_password) = arguments.Read<uint16_t, std::string, std::string, std::string>();
<...Snipped...> <...Snipped...>
@ -164,7 +164,7 @@ The code below demonstrates how we authenticate to Covenant within the construct
3. Make the request and if successful, retrieve the authentication token. 3. Make the request and if successful, retrieve the authentication token.
```c++ ```c++
MWR::C3::Interfaces::Connectors::Covenant::Covenant(ByteView arguments) FSecure::C3::Interfaces::Connectors::Covenant::Covenant(ByteView arguments)
{ {
json postData; json postData;
json response; json response;
@ -289,7 +289,7 @@ The final 2 lines of the Covenant connector's constructor may cause confusion. T
Payload generation is implemented in the `GeneratePayload` method, which in-turn is called from the `PeripheralCreationCommand` method as shown below. Payload generation is implemented in the `GeneratePayload` method, which in-turn is called from the `PeripheralCreationCommand` method as shown below.
```c++ ```c++
MWR::ByteVector MWR::C3::Interfaces::Connectors::Covenant::PeripheralCreationCommand(ByteView connectionId, ByteView data, bool isX64) FSecure::ByteVector FSecure::C3::Interfaces::Connectors::Covenant::PeripheralCreationCommand(ByteView connectionId, ByteView data, bool isX64)
{ {
auto [pipeName, delay, jitter, connectAttempts] = data.Read<std::string, uint32_t, uint32_t, uint32_t>(); auto [pipeName, delay, jitter, connectAttempts] = data.Read<std::string, uint32_t, uint32_t, uint32_t>();
@ -315,7 +315,7 @@ It is worth noting that we do not need to know where the payload is being return
The logic for the above actions is as such: The logic for the above actions is as such:
```c++ ```c++
MWR::ByteVector MWR::C3::Interfaces::Connectors::Covenant::GeneratePayload(ByteView binderId, std::string pipename, uint32_t delay, uint32_t jitter, uint32_t connectAttempts) FSecure::ByteVector FSecure::C3::Interfaces::Connectors::Covenant::GeneratePayload(ByteView binderId, std::string pipename, uint32_t delay, uint32_t jitter, uint32_t connectAttempts)
{ {
if (binderId.empty() || pipename.empty()) if (binderId.empty() || pipename.empty())
throw std::runtime_error{ OBF("Wrong parameters, cannot create payload") }; throw std::runtime_error{ OBF("Wrong parameters, cannot create payload") };
@ -389,7 +389,7 @@ if (binderId.empty() || pipename.empty())
## Adding the Grunt Peripheral ## Adding the Grunt Peripheral
As described, when testing the code above we cheated by adding a call the `GeneratePayload` in the connector's constructor. What we actually need to do is add a Peripheral called `Grunt` to C3. In similar fashion to how the Covenant.cpp connector code was developed, we start by copying `/Src/Common/MWR/C3/Interfaces/Peripherals/Beacon.cpp` to `Src/Common/MWR/C3/Interfaces/Peripherals/Grunt.cpp` - the same is done for the header files. As described, when testing the code above we cheated by adding a call the `GeneratePayload` in the connector's constructor. What we actually need to do is add a Peripheral called `Grunt` to C3. In similar fashion to how the Covenant.cpp connector code was developed, we start by copying `/Src/Common/FSecure/C3/Interfaces/Peripherals/Beacon.cpp` to `Src/Common/FSecure/C3/Interfaces/Peripherals/Grunt.cpp` - the same is done for the header files.
**Stage 1 - Getting User Input** **Stage 1 - Getting User Input**
@ -397,7 +397,7 @@ As described, when testing the code above we cheated by adding a call the `Gener
In order to get user input that is passed to the Covenant connector `GeneratePayload` method, we need to edit the Grunt Peripheral's `GetCapability` function: In order to get user input that is passed to the Covenant connector `GeneratePayload` method, we need to edit the Grunt Peripheral's `GetCapability` function:
```c++ ```c++
MWR::ByteView MWR::C3::Interfaces::Peripherals::Grunt::GetCapability() FSecure::ByteView FSecure::C3::Interfaces::Peripherals::Grunt::GetCapability()
{ {
return R"( return R"(
{ {
@ -458,7 +458,7 @@ First and foremost, when the user submits the form above and a payload is genera
The constructor for a Grunt is slightly complex, essentially we execute the .NET assembly in our own CLR instance running in a seperate thread. Note the `arguments.Read` call below, these are passed in the same order they are presented in the web form. The constructor for a Grunt is slightly complex, essentially we execute the .NET assembly in our own CLR instance running in a seperate thread. Note the `arguments.Read` call below, these are passed in the same order they are presented in the web form.
```c++ ```c++
MWR::C3::Interfaces::Peripherals::Grunt::Grunt(ByteView arguments) FSecure::C3::Interfaces::Peripherals::Grunt::Grunt(ByteView arguments)
{ {
auto [pipeName, payload, connectAttempts] = arguments.Read<std::string, ByteVector, uint32_t>(); auto [pipeName, payload, connectAttempts] = arguments.Read<std::string, ByteVector, uint32_t>();
@ -467,7 +467,7 @@ MWR::C3::Interfaces::Peripherals::Grunt::Grunt(ByteView arguments)
SIZE_T len = payload.size(); SIZE_T len = payload.size();
//Setup the arguments to run the .NET assembly in a seperate thread. //Setup the arguments to run the .NET assembly in a seperate thread.
namespace SEH = MWR::WinTools::StructuredExceptionHandling; namespace SEH = FSecure::WinTools::StructuredExceptionHandling;
SEH::gruntArgs args; SEH::gruntArgs args;
args.gruntStager = x; args.gruntStager = x;
args.len = len; args.len = len;
@ -523,10 +523,10 @@ public static void Write(PipeStream pipe, byte[] bytes)
} }
``` ```
In order for the C3 Relay to read this data, we must alter the named pipe communications. This requires creating a custom named pipe communication specific to Covenant. Essentially, we create a new read method `ReadCov` in `/Src/Common/MWR/WinTools/Pipe.cpp` as part of the `AlternatingPipe` class. This method reflects how an SMB Grunt writes data. In order for the C3 Relay to read this data, we must alter the named pipe communications. This requires creating a custom named pipe communication specific to Covenant. Essentially, we create a new read method `ReadCov` in `/Src/Common/FSecure/WinTools/Pipe.cpp` as part of the `AlternatingPipe` class. This method reflects how an SMB Grunt writes data.
```c++ ```c++
MWR::ByteVector MWR::WinTools::AlternatingPipe::ReadCov() FSecure::ByteVector FSecure::WinTools::AlternatingPipe::ReadCov()
{ {
DWORD temp = 0, total = 0; DWORD temp = 0, total = 0;
if (WaitForSingleObject(m_Event.get(), 0) != WAIT_OBJECT_0) if (WaitForSingleObject(m_Event.get(), 0) != WAIT_OBJECT_0)
@ -563,7 +563,7 @@ MWR::ByteVector MWR::WinTools::AlternatingPipe::ReadCov()
Back in the Grunt.cpp code, we alter the `OnReceiveFromPeripheral` method to make use of this new logic: Back in the Grunt.cpp code, we alter the `OnReceiveFromPeripheral` method to make use of this new logic:
```c++ ```c++
MWR::ByteVector MWR::C3::Interfaces::Peripherals::Grunt::OnReceiveFromPeripheral() FSecure::ByteVector FSecure::C3::Interfaces::Peripherals::Grunt::OnReceiveFromPeripheral()
{ {
std::unique_lock<std::mutex> lock{ m_Mutex }; std::unique_lock<std::mutex> lock{ m_Mutex };
m_ConditionalVariable.wait(lock, [this]() { return m_ReadingState || m_Close; }); m_ConditionalVariable.wait(lock, [this]() { return m_ReadingState || m_Close; });