[Redesign add agent] Add register agent command generator (#5469)

(cherry picked from commit 106cda1)
wazuh · Jun 13, 2023 · a587436 · a587436
1 parent 2c316e9
commit a587436
Show file tree

Hide file tree

Showing 15 changed files with 2,488 additions and 0 deletions.
diff --git a/public/controllers/register-agent/config/os-commands-definitions.ts b/public/controllers/register-agent/config/os-commands-definitions.ts
@@ -0,0 +1,148 @@
+import { IOSDefinition, tOptionalParams } from '../core/register-commands/types';
+
+// Defined OS combinations
+export interface ILinuxOSTypes {
+  name: 'linux';
+  architecture: 'x64' | 'x86';
+  extension: 'rpm' | 'deb';
+}
+export interface IWindowsOSTypes {
+  name: 'windows';
+  architecture: 'x86';
+  extension: 'msi';
+}
+
+export interface IMacOSTypes {
+  name: 'mac';
+  architecture: '32/64';
+  extension: 'pkg';
+}
+
+export type tOperatingSystem = ILinuxOSTypes | IMacOSTypes | IWindowsOSTypes;
+
+
+export type tOptionalParameters = 'server_address' | 'agent_name' | 'agent_group' | 'protocol' | 'wazuh_password';
+
+///////////////////////////////////////////////////////////////////
+/// Operating system commands definitions
+///////////////////////////////////////////////////////////////////
+
+const linuxDefinition: IOSDefinition<tOperatingSystem, tOptionalParameters> = {
+  name: 'linux',
+  options: [
+    {
+      extension: 'deb',
+      architecture: '32/64',
+      urlPackage: props =>
+        `https://packages.wazuh.com/4.x/yum/wazuh-agent-${props.wazuhVersion}-1.x86_64.${props.extension}`,
+      installCommand: props =>
+        `sudo yum install -y ${props.urlPackage}`,
+      startCommand: props => `sudo systemctl start wazuh-agent`,
+    },
+    {
+      extension: 'deb',
+      architecture: 'x64',
+      urlPackage: props =>
+        `https://packages.wazuh.com/4.x/apt/pool/main/w/wazuh-agent/ wazuh-agent_${props.wazuhVersion}-1_${props.architecture}.${props.extension}`,
+      installCommand: props =>
+        `curl -so wazuh-agent.deb ${props.urlPackage} && sudo dpkg -i ./wazuh-agent.deb`,
+      startCommand: props => `sudo systemctl start wazuh-agent`,
+    },
+    {
+      extension: 'rpm',
+      architecture: '32/64',
+      urlPackage: props =>
+        `https://packages.wazuh.com/4.x/yum/wazuh-agent-${props.wazuhVersion}-1.x86_64.${props.extension}`,
+      installCommand: props =>
+        `sudo yum install -y ${props.urlPackage}`,
+      startCommand: props => `sudo systemctl start wazuh-agent`,
+    },
+    {
+      extension: 'deb',
+      architecture: 'x64',
+      urlPackage: props =>
+        `https://packages.wazuh.com/4.x/apt/pool/main/w/wazuh-agent/wazuh-agent_${props.wazuhVersion}-1_amd64.${props.extension}`,
+      installCommand: props =>
+        `curl -so wazuh-agent.deb ${props.urlPackage} && sudo dpkg -i ./wazuh-agent.deb`,
+      startCommand: props => `sudo systemctl start wazuh-agent`,
+    },
+  ],
+};
+
+const windowsDefinition: IOSDefinition<tOperatingSystem, tOptionalParameters> = {
+  name: 'windows',
+  options: [
+    {
+      extension: 'msi',
+      architecture: 'x86',
+      urlPackage: props =>
+        `https://packages.wazuh.com/4.x/windows/wazuh-agent-${props.wazuhVersion}-1.${props.extension}`,
+      installCommand: props =>
+        `Invoke-WebRequest -Uri ${props.urlPackage} -OutFile \${env.tmp}\\wazuh-agent.${props.extension}; msiexec.exe /i \${env.tmp}\\wazuh-agent.${props.extension} /q`,
+      startCommand: props => `Start-Service -Name wazuh-agent`,
+    },
+  ],
+};
+
+const macDefinition: IOSDefinition<tOperatingSystem, tOptionalParameters> = {
+  name: 'mac',
+  options: [
+    {
+      extension: 'pkg',
+      architecture: '32/64',
+      urlPackage: props =>
+        `https://packages.wazuh.com/4.x/macos/wazuh-agent-${props.wazuhVersion}-1.${props.extension}`,
+      installCommand: props =>
+        `mac -so wazuh-agent.pkg ${props.urlPackage} && sudo launchctl setenv && sudo installer -pkg ./wazuh-agent.pkg -target /`,
+      startCommand: props => `sudo /Library/Ossec/bin/wazuh-control start`,
+    },
+  ],
+};
+
+export const osCommandsDefinitions = [
+  linuxDefinition,
+  windowsDefinition,
+  macDefinition,
+];
+
+///////////////////////////////////////////////////////////////////
+/// Optional parameters definitions
+///////////////////////////////////////////////////////////////////
+
+export const optionalParamsDefinitions: tOptionalParams<tOptionalParameters> = {
+  server_address: {
+      property: 'WAZUH_MANAGER',
+      getParamCommand:  props => {
+        const { property, value } = props;
+        return `${property}=${value}`;
+    }
+  },
+  agent_name: {
+      property: 'WAZUH_AGENT_NAME',
+      getParamCommand: props => {
+          const { property, value } = props;
+          return `${property}=${value}`;
+      }
+  },
+  protocol: {
+      property: 'WAZUH_MANAGER_PROTOCOL',
+      getParamCommand: props => {
+        const { property, value } = props;
+        return `${property}=${value}`;
+    }
+  },
+  agent_group: {
+      property: 'WAZUH_AGENT_GROUP',
+      getParamCommand:  props => {
+        const { property, value } = props;
+        return `${property}=${value}`;
+    }
+  },
+  wazuh_password: {
+      property: 'WAZUH_PASSWORD',
+      getParamCommand:  props => {
+        const { property, value } = props;
+        return `${property}=${value}`;
+    }
+  }
+}