fix: hash of file name for secure

QXIP · Nov 19, 2023 · 8ca7c41 · 8ca7c41
1 parent 4f9f63e
commit 8ca7c41
Showing 1 changed file with 20 additions and 5 deletions.
diff --git a/src/app/services/web-shark-data.service.ts b/src/app/services/web-shark-data.service.ts
@@ -28,9 +28,24 @@ export class WebSharkDataService {
   public updates: Observable<any>;
 
   constructor(private http: HttpClient) {
-    if (location.hash) {
-      this.setCaptureFile(decodeURIComponent(location.hash.slice(1)))
-    }
+    const getParam = decodeURIComponent(location.hash.slice(1))
+    setTimeout(() => {
+      if (getParam) {
+        this.getFiles().then(f => {
+          // hash
+          const fileObject = f.files.find((i: any) => hash(i.name) === getParam);
+          if (!fileObject) {
+            location.hash = '';
+            return;
+          }
+
+          const fileName = fileObject.name;
+          StaticData.captureFile = fileName;
+          location.hash = '#' + encodeURIComponent(hash(fileName));
+          this.behavior.next({});
+        })
+      }
+    }, 200);
     this.updates = this.behavior.asObservable();
   }
 
@@ -57,7 +72,7 @@ export class WebSharkDataService {
   }
   public setCaptureFile(fileName: string) {
     StaticData.captureFile = fileName;
-    location.hash = '#' + encodeURIComponent(fileName);
+    location.hash = '#' + hash(fileName);
     this.behavior.next({});
   }
 
@@ -105,7 +120,7 @@ export class WebSharkDataService {
     if (limit === 0) {
       return this.httpGet('frames', {});
     }
-    return this.httpGet('frames', {limit});
+    return this.httpGet('frames', { limit });
   }
 
   getFrameData(frameId: number): Promise<any> {