From a97b8706ec39fb7192a9a78a91b1481c3e051129 Mon Sep 17 00:00:00 2001
From: Ilija Lazoroski <ilija.la@live.com>
Date: Wed, 16 Feb 2022 17:29:21 +0100
Subject: [PATCH] Agent: Add SSH keypair credential type

---
 .../credential_components/ssh_keypair.py                 | 9 +++++++++
 .../i_puppet/credential_collection/credential_type.py    | 1 +
 2 files changed, 10 insertions(+)
 create mode 100644 monkey/infection_monkey/credential_collectors/credential_components/ssh_keypair.py

diff --git a/monkey/infection_monkey/credential_collectors/credential_components/ssh_keypair.py b/monkey/infection_monkey/credential_collectors/credential_components/ssh_keypair.py
new file mode 100644
index 000000000..c23833681
--- /dev/null
+++ b/monkey/infection_monkey/credential_collectors/credential_components/ssh_keypair.py
@@ -0,0 +1,9 @@
+from dataclasses import dataclass, field
+
+from infection_monkey.i_puppet import CredentialType, ICredentialComponent
+
+
+@dataclass(frozen=True)
+class SSHKeypair(ICredentialComponent):
+    credential_type: CredentialType = field(default=CredentialType.SSH_KEYPAIR, init=False)
+    content: dict
diff --git a/monkey/infection_monkey/i_puppet/credential_collection/credential_type.py b/monkey/infection_monkey/i_puppet/credential_collection/credential_type.py
index 98e6c0097..ef00f3732 100644
--- a/monkey/infection_monkey/i_puppet/credential_collection/credential_type.py
+++ b/monkey/infection_monkey/i_puppet/credential_collection/credential_type.py
@@ -6,3 +6,4 @@ class CredentialType(Enum):
     PASSWORD = 2
     NT_HASH = 3
     LM_HASH = 4
+    SSH_KEYPAIR = 5