From 854d1f9150367979a697c9da58ca568e033a6c4d Mon Sep 17 00:00:00 2001
From: Robbie Averill <robbie@averill.co.nz>
Date: Wed, 6 Dec 2017 13:37:13 +1300
Subject: [PATCH 1/4] FIX Ensure document set IDs are cast to integers

---
 code/cms/DMSDocumentAddController.php | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/code/cms/DMSDocumentAddController.php b/code/cms/DMSDocumentAddController.php
index dcba191..185f607 100644
--- a/code/cms/DMSDocumentAddController.php
+++ b/code/cms/DMSDocumentAddController.php
@@ -190,14 +190,14 @@ class DMSDocumentAddController extends LeftAndMain
                 return Controller::join_links(
                     $modelAdmin->Link('DMSDocumentSet'),
                     'EditForm/field/DMSDocumentSet/item',
-                    $this->getRequest()->getVar('dsid'),
+                    (int) $this->getRequest()->getVar('dsid'),
                     'edit'
                 );
             }
             return $modelAdmin->Link();
         }
 
-        return $this->getPageEditLink($this->currentPageID(), $this->getRequest()->getVar('dsid'));
+        return $this->getPageEditLink($this->currentPageID(), (int) $this->getRequest()->getVar('dsid'));
     }
 
     /**
@@ -212,7 +212,7 @@ class DMSDocumentAddController extends LeftAndMain
         return Controller::join_links(
             CMSPageEditController::singleton()->getEditForm($pageId)->FormAction(),
             'field/Document Sets/item',
-            $documentSetId
+            (int) $documentSetId
         );
     }
 

From 0e84799f59ebf8d14f6b165f48ea04500fab646b Mon Sep 17 00:00:00 2001
From: Robbie Averill <robbie@averill.co.nz>
Date: Wed, 6 Dec 2017 13:40:46 +1300
Subject: [PATCH 2/4] FIX Ensure actions panel keys and values have possible
 HTML escaped

---
 code/model/DMSDocument.php | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/code/model/DMSDocument.php b/code/model/DMSDocument.php
index a4ba0ca..8cd4bb5 100644
--- a/code/model/DMSDocument.php
+++ b/code/model/DMSDocument.php
@@ -1320,6 +1320,9 @@ class DMSDocument extends DataObject implements DMSDocumentInterface
             . '<ul>';
 
         foreach ($this->actionTasks as $panelKey => $title) {
+            $panelKey = Convert::raw2xml($panelKey);
+            $title = Convert::raw2xml($title);
+
             $html .= '<li class="ss-ui-button dmsdocument-action" data-panel="' . $panelKey . '">'
                 . _t('DMSDocument.ACTION_' . strtoupper($panelKey), $title)
                 . '</li>';

From 8efedf3158e157655f5f4beb9207fb3775e007f0 Mon Sep 17 00:00:00 2001
From: Robbie Averill <robbie@silverstripe.com>
Date: Thu, 7 Dec 2017 12:40:11 +1300
Subject: [PATCH 3/4] FIX Escape file path before loading file from filesystem

---
 code/model/DMSDocument_Controller.php | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/code/model/DMSDocument_Controller.php b/code/model/DMSDocument_Controller.php
index 6aa4912..3ddf7a6 100644
--- a/code/model/DMSDocument_Controller.php
+++ b/code/model/DMSDocument_Controller.php
@@ -82,6 +82,7 @@ class DMSDocument_Controller extends Controller
                         $finfo = finfo_open(FILEINFO_MIME_TYPE);
                         $mime = finfo_file($finfo, $path);
                     } elseif (is_executable($fileBin)) {
+                        $path = escapeshellarg($path);
                         // try to use the system tool
                         $mime = `$fileBin -i -b $path`;
                         $mime = explode(';', $mime);
@@ -89,7 +90,7 @@ class DMSDocument_Controller extends Controller
                     } else {
                         // make do with what we have
                         $ext = $doc->getExtension();
-                        if ($ext =='pdf') {
+                        if ($ext == 'pdf') {
                             $mime = 'application/pdf';
                         } elseif ($ext == 'html' || $ext =='htm') {
                             $mime = 'text/html';

From 046047e49fa321507dcf94ce5c230c49fb89835b Mon Sep 17 00:00:00 2001
From: Robbie Averill <robbie@silverstripe.com>
Date: Thu, 7 Dec 2017 16:41:52 +1300
Subject: [PATCH 4/4] FIX Cast shortcode document ID as int

---
 code/tools/ShortCodeRelationFinder.php | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/code/tools/ShortCodeRelationFinder.php b/code/tools/ShortCodeRelationFinder.php
index 9cddc50..45155d0 100644
--- a/code/tools/ShortCodeRelationFinder.php
+++ b/code/tools/ShortCodeRelationFinder.php
@@ -39,10 +39,12 @@ class ShortCodeRelationFinder
     }
 
     /**
+     * @param int $number
      * @return DataList
      */
     public function getList($number)
     {
+        $number = (int) $number;
         $list = DataList::create('SiteTree');
         $where = array();
         $fields = $this->getShortCodeFields('SiteTree');