Add missing information of the agent

wazuh · Nov 15, 2024 · 9e924d4 · 9e924d4
1 parent 10f7e3b
commit 9e924d4
Show file tree

Hide file tree

Showing 9 changed files with 35 additions and 0 deletions.
diff --git a/ecs/states-fim/event-generator/event_generator.py b/ecs/states-fim/event-generator/event_generator.py
@@ -35,6 +35,9 @@ def generate_random_date():
 def generate_random_agent():
     agent = {
         'id': f'agent{random.randint(0, 99)}',
+        'name': f'Agent{random.randint(0, 99)}',
+        'type': random.choice(['filebeat', 'windows', 'linux', 'macos']),
+        'version': f'v{random.randint(0, 9)}-stable',
         'groups': [f'group{random.randint(0, 99)}', f'group{random.randint(0, 99)}'],
         'host': generate_random_host()
     }

diff --git a/ecs/states-inventory-hardware/event-generator/event_generator.py b/ecs/states-inventory-hardware/event-generator/event_generator.py
@@ -34,6 +34,10 @@ def generate_random_date():
 
 def generate_random_agent():
     agent = {
+        'id': f'agent{random.randint(0, 99)}',
+        'name': f'Agent{random.randint(0, 99)}',
+        'type': random.choice(['filebeat', 'windows', 'linux', 'macos']),
+        'version': f'v{random.randint(0, 9)}-stable',
         'groups': [f'group{random.randint(0, 99)}', f'group{random.randint(0, 99)}'],
         'host': generate_random_host()
     }

diff --git a/ecs/states-inventory-hotfixes/event-generator/event_generator.py b/ecs/states-inventory-hotfixes/event-generator/event_generator.py
@@ -34,6 +34,10 @@ def generate_random_date():
 
 def generate_random_agent():
     agent = {
+        'id': f'agent{random.randint(0, 99)}',
+        'name': f'Agent{random.randint(0, 99)}',
+        'type': random.choice(['filebeat', 'windows', 'linux', 'macos']),
+        'version': f'v{random.randint(0, 9)}-stable',
         'groups': [f'group{random.randint(0, 99)}', f'group{random.randint(0, 99)}'],
         'host': generate_random_host()
     }

diff --git a/ecs/states-inventory-networks/event-generator/event_generator.py b/ecs/states-inventory-networks/event-generator/event_generator.py
@@ -34,6 +34,10 @@ def generate_random_date():
 
 def generate_random_agent():
     agent = {
+        'id': f'agent{random.randint(0, 99)}',
+        'name': f'Agent{random.randint(0, 99)}',
+        'type': random.choice(['filebeat', 'windows', 'linux', 'macos']),
+        'version': f'v{random.randint(0, 9)}-stable',
         'groups': [f'group{random.randint(0, 99)}', f'group{random.randint(0, 99)}'],
         'host': generate_random_host()
     }

diff --git a/ecs/states-inventory-packages/event-generator/event_generator.py b/ecs/states-inventory-packages/event-generator/event_generator.py
@@ -34,6 +34,10 @@ def generate_random_date():
 
 def generate_random_agent():
     agent = {
+        'id': f'agent{random.randint(0, 99)}',
+        'name': f'Agent{random.randint(0, 99)}',
+        'type': random.choice(['filebeat', 'windows', 'linux', 'macos']),
+        'version': f'v{random.randint(0, 9)}-stable',
         'groups': [f'group{random.randint(0, 99)}', f'group{random.randint(0, 99)}'],
         'host': generate_random_host()
     }

diff --git a/ecs/states-inventory-ports/event-generator/event_generator.py b/ecs/states-inventory-ports/event-generator/event_generator.py
@@ -34,6 +34,10 @@ def generate_random_date():
 
 def generate_random_agent():
     agent = {
+        'id': f'agent{random.randint(0, 99)}',
+        'name': f'Agent{random.randint(0, 99)}',
+        'type': random.choice(['filebeat', 'windows', 'linux', 'macos']),
+        'version': f'v{random.randint(0, 9)}-stable',
         'groups': [f'group{random.randint(0, 99)}', f'group{random.randint(0, 99)}'],
         'host': generate_random_host()
     }

diff --git a/ecs/states-inventory-processes/event-generator/event_generator.py b/ecs/states-inventory-processes/event-generator/event_generator.py
@@ -34,6 +34,10 @@ def generate_random_date():
 
 def generate_random_agent():
     agent = {
+        'id': f'agent{random.randint(0, 99)}',
+        'name': f'Agent{random.randint(0, 99)}',
+        'type': random.choice(['filebeat', 'windows', 'linux', 'macos']),
+        'version': f'v{random.randint(0, 9)}-stable',
         'groups': [f'group{random.randint(0, 99)}', f'group{random.randint(0, 99)}'],
         'host': generate_random_host()
     }

diff --git a/ecs/states-inventory-system/event-generator/event_generator.py b/ecs/states-inventory-system/event-generator/event_generator.py
@@ -34,6 +34,10 @@ def generate_random_date():
 
 def generate_random_agent():
     agent = {
+        'id': f'agent{random.randint(0, 99)}',
+        'name': f'Agent{random.randint(0, 99)}',
+        'type': random.choice(['filebeat', 'windows', 'linux', 'macos']),
+        'version': f'v{random.randint(0, 9)}-stable',
         'groups': [f'group{random.randint(0, 99)}', f'group{random.randint(0, 99)}'],
         'host': generate_random_host()
     }

diff --git a/ecs/states-vulnerabilities/event-generator/event_generator.py b/ecs/states-vulnerabilities/event-generator/event_generator.py
@@ -34,6 +34,10 @@ def generate_random_date():
 
 def generate_random_agent():
     agent = {
+        'id': f'agent{random.randint(0, 99)}',
+        'name': f'Agent{random.randint(0, 99)}',
+        'type': random.choice(['filebeat', 'windows', 'linux', 'macos']),
+        'version': f'v{random.randint(0, 9)}-stable',
         'groups': [f'group{random.randint(0, 99)}', f'group{random.randint(0, 99)}'],
         'host': generate_random_host()
     }