Add chat-bot PSE

Author: cytopia

pse/README.md

@@ -32,6 +32,12 @@ These scripts currently only serve as a way to give you an idea about how this c
    <td>✔</td>
    <td>✔</td>
   </tr>
+  <tr>
+   <td><a href="chat-bot">chat-bot</a></td>
+   <td>Basic <i>dummy</i> chat bot that wants you to greet it, tell it your name and will then ask you a couple of questions.</td>
+   <td>✔</td>
+   <td>✔</td>
+  </tr>
  </tbody>
 </table>
 

pse/chat-bot/README.md

@@ -0,0 +1,41 @@
+# PSE: Chat bot
+
+A simple script that makes the remote peer site behave like a bot replying to your questions.
+
+## Usage
+
+Use high verbosity to see how the data gets transformed.
+
+### Server
+
+**Note:** The server side is using `/bin/cat` as its command in order to echo the messages back to the client. Then we hook into the sending part and simply overwrite the replies based on the received input.
+```bash
+pwncat -vvvv -l localhost 4444 \
+  --exec /bin/cat \
+  --script-send pse/chat-bot/pse-chat_bot.py
+```
+
+### Client
+```bash
+pwncat localhost 4444
+```
+
+### In action
+
+The following shows it in action. `>` represents sending data to the server and `<` are the replies.
+```
+> some data
+< Be polite and greet first.
+> hi
+< Nice to meet you, what is your name?
+> cytopia
+< OK, I call you: cytopia
+> that's right
+< Hi cytopia, How is the weather?
+> it's pretty good
+< Hi cytopia, Why are you chatting to a bot?
+> I don't know
+< Hi cytopia, Are you bored?
+> Maybe
+< Hi cytopia, Have you had dinner?
+```

pse/chat-bot/pse-chat_bot.py

@@ -0,0 +1,39 @@
+"""Greet Bot."""
+
+PSE_BOT_QUESTIONS = [
+    "How is the weather?",
+    "How old are you?",
+    "What do you do today?",
+    "Do you like Python?",
+    "Have you had dinner?",
+    "Can you guess my name?",
+    "Are you bored?",
+    "Why are you chatting to a bot?",
+    "Do you expect me to answer properly?",
+]
+
+
+def transform(data, pse):
+    """Transform."""
+    global PSE_BOT_QUESTIONS
+
+    # Setup custom data structure
+    if pse.store is None:
+        pse.store = {}
+    if "bot" not in pse.store:
+        pse.store["bot"] = {}
+
+    from random import randrange
+
+    if "greet" not in pse.store["bot"]:
+        if any(x in data for x in ["hi", "helo", "halo", "hallo", "hello"]):
+            pse.store["bot"]["greet"] = True
+            return "Nice to meet you, what is your name?\n"
+        return "Be polite and greet first.\n"
+
+    if "name" not in pse.store["bot"]:
+        pse.store["bot"]["name"] = data.rstrip()
+        return "OK, I call you: {}".format(data)
+
+    index = randrange(0, len(PSE_BOT_QUESTIONS)-1)
+    return "Hi {}, {}\n".format(pse.store["bot"]["name"], PSE_BOT_QUESTIONS[index])