From a17935c1e53f6d22517aec4cc4b877828b6bec9d Mon Sep 17 00:00:00 2001
From: jLynx <admin@jlynx.net>
Date: Thu, 19 Nov 2020 13:54:12 +1300
Subject: [PATCH 1/3] Update handshakes-dl.py

Added support to download other files such as:
- .pcap
- .2500
- .16800
---
 handshakes-dl.py | 46 ++++++++++++++++++++++++++++++++--------------
 1 file changed, 32 insertions(+), 14 deletions(-)

diff --git a/handshakes-dl.py b/handshakes-dl.py
index 0e37bd5..bd4c32a 100644
--- a/handshakes-dl.py
+++ b/handshakes-dl.py
@@ -13,11 +13,9 @@
 TEMPLATE = """
 {% extends "base.html" %}
 {% set active_page = "handshakes" %}
-
 {% block title %}
     {{ title }}
 {% endblock %}
-
 {% block styles %}
     {{ super() }}
     <style>
@@ -34,7 +32,6 @@
     var shakeList = document.getElementById('list');
     var filter = document.getElementById('filter');
     var filterVal = filter.value.toUpperCase();
-
     filter.onkeyup = function() {
         document.body.style.cursor = 'progress';
         var table, tr, tds, td, i, txtValue;
@@ -50,24 +47,30 @@
         }
         document.body.style.cursor = 'default';
     }
-
 {% endblock %}
-
 {% block content %}
     <input type="text" id="filter" placeholder="Search for ..." title="Type in a filter">
     <ul id="list" data-role="listview" style="list-style-type:disc;">
         {% for handshake in handshakes %}
-            <li class="file">
-                <a href="/plugins/handshakes-dl/{{handshake}}">{{handshake}}</a>
-            </li>
+            {% for ext in handshake.ext %}
+                <li class="file">
+                    <a href="/plugins/handshakes-dl/{{handshake.name}}{{ext}}">{{handshake.name}}{{ext}}</a>
+                </li>
+            {% endfor %}
         {% endfor %}
     </ul>
 {% endblock %}
 """
 
+class handshakes:  
+    def __init__(self, name, path, ext):  
+        self.name = name  
+        self.path = path  
+        self.ext = ext 
+
 class HandshakesDL(plugins.Plugin):
     __author__ = 'me@sayakb.com'
-    __version__ = '0.2.1'
+    __version__ = '0.2.2'
     __license__ = 'GPL3'
     __description__ = 'Download handshake captures from web-ui.'
 
@@ -86,16 +89,31 @@ def on_webhook(self, path, request):
             return "Plugin not ready"
 
         if path == "/" or not path:
-            handshakes = glob.glob(os.path.join(self.config['bettercap']['handshakes'], "*.pcap"))
-            handshakes = [os.path.basename(path)[:-5] for path in handshakes]
+            pcapfiles = glob.glob(os.path.join(self.config['bettercap']['handshakes'], "*.pcap"))
+            
+            data = []
+            for path in pcapfiles:
+                name = os.path.basename(path)[:-5]
+                fullpathNoExt = path[:-5]
+                possibleExt = ['.2500', '.16800']
+                foundExt = ['.pcap']
+                for ext in possibleExt: 
+                    if os.path.isfile(fullpathNoExt +  ext):
+                        foundExt.append(ext)
+                data.append(handshakes(name, fullpathNoExt, foundExt)) 
+                
+            for obj in data: 
+                print( obj.path, obj.ext) 
+            
+            # pcapfiles = [os.path.basename(path)[:-5] for path in pcapfiles]
             return render_template_string(TEMPLATE,
                                     title="Handshakes | " + pwnagotchi.name(),
-                                    handshakes=handshakes)
+                                    handshakes=data)
 
         else:
             dir = self.config['bettercap']['handshakes']
             try:
-                logging.info(f"[HandshakesDL] serving {dir}/{path}.pcap")
-                return send_from_directory(directory=dir, filename=path+'.pcap', as_attachment=True)
+                logging.info(f"[HandshakesDL] serving {dir}/{path}")
+                return send_from_directory(directory=dir, filename=path, as_attachment=True)
             except FileNotFoundError:
                 abort(404)

From b7edbe46de0d0139ff44a6448f77dcda59034833 Mon Sep 17 00:00:00 2001
From: jLynx <admin@jlynx.net>
Date: Thu, 19 Nov 2020 13:59:06 +1300
Subject: [PATCH 2/3] Formatting

---
 handshakes-dl.py | 6 ------
 1 file changed, 6 deletions(-)

diff --git a/handshakes-dl.py b/handshakes-dl.py
index bd4c32a..b47af1b 100644
--- a/handshakes-dl.py
+++ b/handshakes-dl.py
@@ -101,15 +101,9 @@ def on_webhook(self, path, request):
                     if os.path.isfile(fullpathNoExt +  ext):
                         foundExt.append(ext)
                 data.append(handshakes(name, fullpathNoExt, foundExt)) 
-                
-            for obj in data: 
-                print( obj.path, obj.ext) 
-            
-            # pcapfiles = [os.path.basename(path)[:-5] for path in pcapfiles]
             return render_template_string(TEMPLATE,
                                     title="Handshakes | " + pwnagotchi.name(),
                                     handshakes=data)
-
         else:
             dir = self.config['bettercap']['handshakes']
             try:

From c2f9cb438179a9ebc087a6998c3b76671041b6f0 Mon Sep 17 00:00:00 2001
From: jLynx <admin@jlynx.net>
Date: Thu, 19 Nov 2020 17:41:55 +1300
Subject: [PATCH 3/3] Update handshakes-dl.py

Updated to include 22000 files
---
 handshakes-dl.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/handshakes-dl.py b/handshakes-dl.py
index b47af1b..d3ba07b 100644
--- a/handshakes-dl.py
+++ b/handshakes-dl.py
@@ -95,7 +95,7 @@ def on_webhook(self, path, request):
             for path in pcapfiles:
                 name = os.path.basename(path)[:-5]
                 fullpathNoExt = path[:-5]
-                possibleExt = ['.2500', '.16800']
+                possibleExt = ['.2500', '.16800', '.22000']
                 foundExt = ['.pcap']
                 for ext in possibleExt: 
                     if os.path.isfile(fullpathNoExt +  ext):