From fb13a2a304582c6abe61f4b3c6551121e34cb67d Mon Sep 17 00:00:00 2001
From: nokonoko <neku@pomf.se>
Date: Sat, 29 Jan 2022 20:03:35 +0100
Subject: [PATCH] fix extension fix

---
 dist.json                            |  2 +-
 package.json                         |  2 +-
 static/php/includes/Upload.class.php | 23 +++++++++++++++++++----
 3 files changed, 21 insertions(+), 6 deletions(-)

diff --git a/dist.json b/dist.json
index c9f1e8a..e0c4b20 100644
--- a/dist.json
+++ b/dist.json
@@ -3,7 +3,7 @@
     "allowErrors": false
   },
   "dest": "dist",
-  "pkgVersion": "1.5.0",
+  "pkgVersion": "1.5.1",
   "banners": [
     "banners/malware_scans.swig",
     "banners/donations.swig"
diff --git a/package.json b/package.json
index 6aa703b..c3ce950 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "uguu",
-  "version": "1.5.0",
+  "version": "1.5.1",
   "description": "Kawaii file host",
   "homepage": "https://uguu.se/",
   "repository": {
diff --git a/static/php/includes/Upload.class.php b/static/php/includes/Upload.class.php
index 31bb2d1..4318eea 100644
--- a/static/php/includes/Upload.class.php
+++ b/static/php/includes/Upload.class.php
@@ -119,19 +119,34 @@ class Upload
         ];
     }
 
+    public static function getIP()
+    {
+        if (!empty($_SERVER['HTTP_CLIENT_IP'])) {
+            self::$IP = $_SERVER['HTTP_CLIENT_IP'];
+        }
+        if (!empty($_SERVER['HTTP_X_FORWARDED_FOR'])) {
+            self::$IP = $_SERVER['HTTP_X_FORWARDED_FOR'];
+        }
+        if (!isset(self::$IP)) {
+            self::$IP = $_SERVER['REMOTE_ADDR'];
+        }
+    }
+
     public static function fileInfo()
     {
         if (isset($_FILES['files'])) {
             $finfo = finfo_open(FILEINFO_MIME_TYPE);
             self::$FILE_MIME = finfo_file($finfo, self::$TEMP_FILE);
-            $extension = explode('.', self::$FILE_NAME, 2);
-            self::$FILE_EXTENSION = $extension['1'];
             finfo_close($finfo);
 
+            $extension = explode('.', self::$FILE_NAME);
+            self::$FILE_EXTENSION = $extension[count($extension)-2].'.'.$extension[count($extension)-1];
+
+
             if (Settings::$LOG_IP) {
-                self::$IP = $_SERVER['REMOTE_ADDR'];
+                self::getIP();
             } else {
-                self::$IP = '0';
+                self::$IP = null;
             }
         }
     }