From 6e0623335af1badc03cc2dde24a167a2e549d2ad Mon Sep 17 00:00:00 2001
From: Eduardo Pinho <eduardopinho@bmd-software.com>
Date: Tue, 5 Sep 2023 19:00:58 +0200
Subject: [PATCH] Make Find SCP resilient to missing query-retrievel level
 (#665)

- if query-retrieve level is missing or could not be matched,
  send error response
- use getString to retrieve DICOM text value
- clean up code a bit, removing commented out code and unwanted prints
---
 .../server/queryretrieve/FindRSP.java         | 50 +++----------------
 1 file changed, 8 insertions(+), 42 deletions(-)

diff --git a/dicoogle/src/main/java/pt/ua/dicoogle/server/queryretrieve/FindRSP.java b/dicoogle/src/main/java/pt/ua/dicoogle/server/queryretrieve/FindRSP.java
index 95b24d4ec..05771ebf0 100755
--- a/dicoogle/src/main/java/pt/ua/dicoogle/server/queryretrieve/FindRSP.java
+++ b/dicoogle/src/main/java/pt/ua/dicoogle/server/queryretrieve/FindRSP.java
@@ -30,7 +30,6 @@
 import java.util.ArrayList;
 import org.slf4j.LoggerFactory;
 
-import org.dcm4che2.data.DicomElement;
 import org.dcm4che2.data.DicomObject;
 import org.dcm4che2.data.Tag;
 import org.dcm4che2.data.VR;
@@ -68,17 +67,6 @@ public FindRSP(DicomObject keys, DicomObject rsp, String callingAET) {
 
         this.callingAET = callingAET;
 
-
-        /** Debug - show keys, rsp, index */
-        if (keys != null) {
-            // DebugManager.getSettings().debug("keys object: ");
-            // DebugManager.getSettings().debug(keys.toString());
-        }
-        if (rsp != null) {
-            // DebugManager.getSettings().debug("Rsp object");
-            // DebugManager.getSettings().debug(rsp.toString());
-        }
-
         /** 
          * Get object to search
          */
@@ -130,28 +118,8 @@ public FindRSP(DicomObject keys, DicomObject rsp, String callingAET) {
         // System.out.println("OLD Query: "+query);
         query = applyQueryFilter(query);
 
-        System.out.println("NEW Query: " + query);
-
-        /** 
-         * Search results
-         * TODO: 
-         * - Get Search String Query?
-         * - Is A Network Search?
-         * - How works extrafields?
-         */
-
-        SearchDicomResult.QUERYLEVEL level = null;
-        /*
-        if (CFindBuilder.isPatientRoot(rsp))
-        {
-            level = SearchDicomResult.QUERYLEVEL.PATIENT;
-        }
-        else if (CFindBuilder.isStudyRoot(rsp))
-        {
-            level = SearchDicomResult.QUERYLEVEL.STUDY;
-        }*/
-        DicomElement elem = keys.get(Integer.parseInt("00080052", 16));
-        String levelStr = new String(elem.getBytes());
+        final SearchDicomResult.QUERYLEVEL level;
+        String levelStr = keys.getString(Tag.QueryRetrieveLevel, VR.CS, "");
 
         if (levelStr.contains("PATIENT")) {
             level = SearchDicomResult.QUERYLEVEL.PATIENT;
@@ -161,17 +129,15 @@ else if (CFindBuilder.isStudyRoot(rsp))
             level = SearchDicomResult.QUERYLEVEL.SERIE;
         } else if (levelStr.contains("IMAGE")) {
             level = SearchDicomResult.QUERYLEVEL.IMAGE;
+        } else {
+            // illegal query-retrieve level, reject request
+            this.rsp.putInt(Tag.Status, VR.US, 0xA900);
+            this.rsp.putInt(Tag.OffendingElement, VR.AT, Tag.QueryRetrieveLevel);
+            this.rsp.putString(Tag.ErrorComment, VR.LO, "Query/Retrieve Level invalid or missing");
+            return;
         }
-        System.out.println("Charset: " + this.keys.get(Tag.SpecificCharacterSet));
         search = new SearchDicomResult(query, true, extrafields, level);
 
-
-
-        if (search == null) {
-            // DebugManager.getSettings().debug(">> Search is null, so" +
-            // " somethig is wrong ");
-        }
-
         // always return Specific Character Set
         if (!keys.contains(Tag.SpecificCharacterSet)) {
             this.keys.putNull(Tag.SpecificCharacterSet, VR.CS);