Common

# Conflicts:
#	models.py
#	templates/containers/card.logs.html
#	templates/containers/list.logs.html

__init__.py

@@ -82,7 +82,13 @@ def load(app):
     @page_blueprint.route("/admin/containers")
     @admins_only
     def admin_list_containers():
-        result = AdminContainers.get()
+        page_type = request.args.get("type", "instances")
+        if page_type == "logs":
+            result = AdminContainers.logs()
+            session['page_type'] = 'logs'
+        else:
+            result = AdminContainers.get()
+            session['page_type'] = 'containers'
         view_mode = request.args.get('mode', session.get('view_mode', 'list'))
         session['view_mode'] = view_mode
         return render_template("whale_containers.html",

api.py

@@ -31,7 +31,7 @@ class AdminContainers(Resource):
     @admins_only
     def get():
         page = abs(request.args.get("page", 1, type=int))
-        results_per_page = abs(request.args.get("per_page", 20, type=int))
+        results_per_page = abs(request.args.get("per_page", 10, type=int))
         page_start = results_per_page * (page - 1)
         page_end = results_per_page * (page - 1) + results_per_page
 
@@ -46,6 +46,25 @@ class AdminContainers(Resource):
             'page_start': page_start,
         }}
 
+    @staticmethod
+    @admins_only
+    def logs():
+        page = abs(request.args.get("page", 1, type=int))
+        results_per_page = abs(request.args.get("per_page", 10, type=int))
+        page_start = results_per_page * (page - 1)
+        page_end = results_per_page * (page - 1) + results_per_page
+
+        count = DBContainer.get_all_container_logs_count()
+        containers = DBContainer.get_all_container_logs_page(
+            page_start, page_end)
+
+        return {'success': True, 'data': {
+            'containers': containers,
+            'total': count,
+            'pages': int(count / results_per_page) + (count % results_per_page > 0),
+            'page_start': page_start,
+        }}
+
     @staticmethod
     @admins_only
     def patch():

models.py

@@ -103,3 +103,31 @@ class WhaleContainer(db.Model):
     def __repr__(self):
         return "<WhaleContainer ID:{0} {1} {2} {3} {4}>".format(self.id, self.user_id, self.challenge_id,
                                                                 self.start_time, self.renew_count)
+
+class WhaleContainerLog(db.Model):
+    id = db.Column(db.Integer,primary_key=True, autoincrement=True)
+    user_id = db.Column(None, db.ForeignKey("users.id"))
+    challenge_id = db.Column(None, db.ForeignKey("challenges.id"))
+    start_time = db.Column(db.DateTime, nullable=False, default=datetime.utcnow)
+    uuid = db.Column(db.String(256))
+    port = db.Column(db.Integer, nullable=True, default=0)
+    flag = db.Column(db.String(128), nullable=False)
+
+    # Relationships
+    user = db.relationship(
+        "Users", foreign_keys="WhaleContainerLog.user_id", lazy="select")
+    challenge = db.relationship(
+        "DynamicDockerChallenge", foreign_keys="WhaleContainerLog.challenge_id", lazy="select"
+    )
+
+    def __init__(self, container):
+        self.user_id = container.user_id
+        self.challenge_id = container.challenge_id
+        self.start_time = container.start_time
+        self.uuid = container.uuid
+        self.port = container.port
+        self.flag = container.flag
+
+    def __repr__(self):
+        return "<WhaleContainer ID:{0} {1} {2} {3} {4} {5}>".format(self.id, self.user_id, self.challenge_id,
+                                                                self.start_time, self.uuid, self.port)

templates/containers/card.logs.html

@@ -0,0 +1,44 @@
+<style>
+    .info-card.card {
+        height: 11rem;
+    }
+
+    .card-text {
+        text-overflow: ellipsis;
+        white-space: nowrap;
+        overflow: hidden;
+    }
+
+    .card-text:hover {
+        white-space: pre-line;
+        overflow: visible;
+    }
+</style>
+
+<div class="row">
+    {% for container in containers %}
+        <div class="col-sm-6 pb-3">
+            <div class="info-card card">
+                <div class="card-body">
+                    <h5 class="d-inline-block card-title">
+                        <a style="width: 5rem;"
+                           href="{{ url_for('admin.challenges_detail', challenge_id=container.challenge.id) }}"
+                        >{{ container.challenge.name | truncate(15) }}
+                        </a>
+                    </h5>
+                    <h6 class="d-inline-block card-subtitle float-right">
+                        <a style="width: 5rem;"
+                           class="btn btn-outline-secondary rounded"
+                           href="{{ url_for('admin.users_detail', user_id=container.user.id) }}"
+                        >{{ container.user.name | truncate(5) }}
+                        </a>
+                    </h6>
+                    <p class="card-text">Container: {{ container.uuid }}</p>
+                    <p class="card-text">{{ container.flag }}</p>
+                    Time Started: {{ container.start_time }}
+                    <span class="badge badge-secondary float-right">Log Record</span>
+                </div>
+            </div>
+        </div>
+    {% endfor %}
+</div>

templates/containers/list.logs.html

@@ -0,0 +1,63 @@
+<div class="row">
+    <div class="col-md-12">
+        <table class="table table-striped border">
+            <thead>
+            <tr>
+                <th class="border-right" data-checkbox>
+                    <div class="form-check text-center">&nbsp;
+                        <input type="checkbox" class="form-check-input" data-checkbox-all>
+                    </div>
+                </th>
+                <th class="sort-col text-center"><b>ID</b></td>
+                <th class="text-center"><b>User</b></td>
+                <th class="sort-col text-center"><b>Challenge</b></td>
+                <th class="text-center"><b>Container</b></td>
+                <th class="text-center"><b>Port</b></td>
+                <th class="text-center"><b>Flag</b></td>
+                <th class="sort-col text-center"><b>Startup Time</b></td>
+            </tr>
+            </thead>
+            <tbody>
+            {% for container in containers %}
+                <tr>
+                    <td class="border-right" data-checkbox>
+                        <div class="form-check text-center">&nbsp;
+                            <input type="checkbox" class="form-check-input" data-user-id="{{ container.user.id if container.user else '' }}">
+                        </div>
+                    </td>
+                    <td class="text-center">
+                        {{ container.id }}
+                    </td>
+                    <td class="text-center">
+                            <a href="{{ url_for('admin.users_detail', user_id=container.user.id) }}">
+                                {{ container.user.name | truncate(12) }}
+                            </a>
+                    </td>
+                    <td class="text-center">
+                        <a href="{{ url_for('admin.challenges_detail', challenge_id=container.challenge.id) }}">
+                            {{ container.challenge.name }}
+                        </a>
+                    </td>
+                    <td class="text-center" style="word-break: break-all; white-space: normal; max-width: 200px;">
+                        <span class="d-inline-block">{{ container.uuid }}</span>
+                        <button class="btn btn-link p-0 click-copy ml-1" data-copy="{{ container.uuid }}" title="复制UUID">
+                            <i class="fas fa-clipboard"></i>
+                        </button>
+                    </td>
+                    <td class="text-center">
+                        {{ container.port if container.port else '-' }}
+                    </td>
+                    <td class="text-center">
+                        <button class="btn btn-link p-0 click-copy" data-copy="{{ container.flag }}">
+                            <i class="fas fa-clipboard"></i>
+                        </button>
+                    </td>
+                    <td class="text-center">
+                        <span data-time="{{ container.start_time | isoformat }}"></span>
+                    </td>
+                </tr>
+            {% endfor %}
+            </tbody>
+        </table>
+    </div>
+</div>

templates/whale_config.html

@@ -14,7 +14,7 @@
         <a class="nav-link" data-toggle="pill" href="#challenges">Challenges</a>
     </li>
     <li class="nav-item">
-        <a class="nav-link" href="/plugins/ctfd-whale/admin/containers">🔗 Instances</a>
+        <a class="nav-link" href="/plugins/ctfd-whale/admin/containers">🔗 Containers</a>
     </li>
 {% endblock %}
 

templates/whale_containers.html

@@ -5,7 +5,12 @@
         <a class="nav-link" href="/plugins/ctfd-whale/admin/settings">🔗 Settings</a>
     </li>
     <li class="nav-item">
-        <a class="nav-link active" href="#">Instances</a>
+        <a class="nav-link{{ ' active' if session.get('page_type', 'containers') == 'containers' else '' }}" 
+        href="/plugins/ctfd-whale/admin/containers?type=instances">Instances</a>
+    </li>
+    <li class="nav-item">
+        <a class="nav-link{{ ' active' if session.get('page_type') == 'logs' else '' }}" 
+        href="/plugins/ctfd-whale/admin/containers?type=logs">Logs</a>
     </li>
 
     <li class="nav-item nav-link">
@@ -25,7 +30,7 @@
         <ul class="pagination">
             <li class="page-item{{ ' disabled' if curr_page <= 1 else '' }}">
                 <a class="page-link" aria-label="Previous"
-                   href="/plugins/ctfd-whale/admin/containers?page={{ curr_page - 1 }}"
+                   href="/plugins/ctfd-whale/admin/containers?page={{ curr_page - 1 }}&type={{ session.get('page_type', 'containers') }}"
                 >
                     <span aria-hidden="true">&laquo;</span>
                     <span class="sr-only">Previous</span>
@@ -36,13 +41,13 @@
             {% for page in range(range_l, range_r + 1) %}
                 <li class="page-item{{ ' active' if curr_page == page }}">
                     <a class="page-link"
-                       href="/plugins/ctfd-whale/admin/containers?page={{ page }}"
+                       href="/plugins/ctfd-whale/admin/containers?page={{ page }}&type={{ session.get('page_type', 'containers') }}"
                     >{{ page }}</a>
                 </li>
             {% endfor %}
             <li class="page-item{{ ' disabled' if curr_page >= pages else '' }}">
                 <a class="page-link" aria-label="Next"
-                   href="/plugins/ctfd-whale/admin/containers?page={{ curr_page + 1 }}"
+                   href="/plugins/ctfd-whale/admin/containers?page={{ curr_page + 1 }}&type={{ session.get('page_type', 'containers') }}"
                 >
                     <span aria-hidden="true">&raquo;</span>
                     <span class="sr-only">Next</span>
@@ -53,17 +58,22 @@
     
     <li class="nav-item nav-link">
         {% if session['view_mode'] == 'card' %}
-            <a href="?mode=list">Switch to list mode</a>
+            <a href="?mode=list&type={{ session.get('page_type', 'containers') }}">Switch to list mode</a>
         {% else %}
-            <a href="?mode=card">Switch to card mode</a>
+            <a href="?mode=card&type={{ session.get('page_type', 'containers') }}">Switch to card mode</a>
         {% endif %}
     </li>
 {% endblock %}
 
 {% block panel %}
-    {% include "containers/" + session["view_mode"] + ".containers.html" %}
+    {% if session.get("page_type", "containers") == "containers" %}
+        {% include "containers/" + session["view_mode"] + ".containers.html" %}
+    {% elif session["page_type"] == "logs" %}
+        {% include "containers/" + session["view_mode"] + ".logs.html" %}
+    {% endif %}
 {% endblock %}
 
+
 {% block scripts %}
     <script defer src="{{ url_for('plugins.ctfd-whale.assets', path='containers.js') }}"></script>
 {% endblock %}

utils/control.py

@@ -22,6 +22,7 @@ class ControlUtil:
             DockerUtils.remove_container(container)
             DBContainer.remove_container_record(user_id)
             return False, msg
+        DBContainer.create_container_log(container)
         return True, 'Container created'
 
     @staticmethod

utils/db.py

@@ -2,7 +2,7 @@ import datetime
 
 from CTFd.models import db
 from CTFd.utils import get_config
-from ..models import WhaleContainer, WhaleRedirectTemplate
+from ..models import WhaleContainer, WhaleContainerLog, WhaleRedirectTemplate
 
 
 class DBContainer:
@@ -14,6 +14,14 @@ class DBContainer:
 
         return container
 
+    @staticmethod
+    def create_container_log(container):
+        container = WhaleContainerLog(container)
+        db.session.add(container)
+        db.session.commit()
+
+        return container
+
     @staticmethod
     def get_current_containers(user_id):
         q = db.session.query(WhaleContainer)
@@ -55,6 +63,15 @@ class DBContainer:
         )
         return q.all()
 
+    @staticmethod
+    def get_all_container_logs_page(page_start, page_end):
+        q = db.session.query(WhaleContainerLog)
+        q = q.order_by(WhaleContainerLog.id.desc())
+        q = q.slice(page_start, page_end)
+
+        return q.all()
+
+
     @staticmethod
     def get_all_container():
         q = db.session.query(WhaleContainer)
@@ -69,9 +86,16 @@ class DBContainer:
             WhaleContainer.start_time >=
             datetime.datetime.now() - datetime.timedelta(seconds=timeout)
         )
+        q = q.order_by(WhaleContainer.id.desc())
         q = q.slice(page_start, page_end)
         return q.all()
 
+    @staticmethod
+    def get_all_container_logs_count():
+        q = db.session.query(WhaleContainerLog)
+
+        return q.count()
+
     @staticmethod
     def get_all_alive_container_count():
         timeout = int(get_config("whale:docker_timeout", "3600"))

utils/docker.py

@@ -43,7 +43,7 @@ class DockerUtils:
                 'if you are using unix:///var/run/docker.sock, check if the socket is correctly mapped'
             )
         credentials = get_config("whale:docker_credentials")
-        if credentials and credentials.count(':') == 1:
+        if credentials and credentials.count(':') == 3:
             try:
                 DockerUtils.client.login(*credentials.split(':'))
             except Exception: