Extend get_vts with attribute version_only and return the version

If the attribute version_only is set, does not send any vt in the response.

CHANGELOG.md

@@ -26,6 +26,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/).
 - Add new scan status INTERRUPTED. 
   [#288](https://github.com/greenbone/ospd/pull/288)
   [#289](https://github.com/greenbone/ospd/pull/289)
+- Extend get_vts with attribute version_only and return the version [#291](https://github.com/greenbone/ospd/pull/291)
 
 ### Changes
 - Modify __init__() method and use new syntax for super(). [#186](https://github.com/greenbone/ospd/pull/186)

ospd/command/command.py

@@ -319,6 +319,7 @@ def handle_xml(self, xml: Element) -> Iterator[bytes]:
         vt_id = xml.get('vt_id')
         vt_filter = xml.get('filter')
         _details = xml.get('details')
+        version_only = xml.get('version_only')
 
         vt_details = False if _details == '0' else True
 
@@ -342,6 +343,9 @@ def handle_xml(self, xml: Element) -> Iterator[bytes]:
         yield xml_helper.create_response('get_vts')
 
         begin_vts_tag = xml_helper.create_element('vts')
+        begin_vts_tag = xml_helper.add_attr(
+            begin_vts_tag, "vts_version", self._daemon.get_vts_version()
+        )
         val = len(self._daemon.vts)
         begin_vts_tag = xml_helper.add_attr(begin_vts_tag, "total", val)
         if filtered_vts:
@@ -354,9 +358,9 @@ def handle_xml(self, xml: Element) -> Iterator[bytes]:
             )
 
         yield begin_vts_tag
-
-        for vt in self._daemon.get_vt_iterator(vts_selection, vt_details):
-            yield xml_helper.add_element(self._daemon.get_vt_xml(vt))
+        if not version_only:
+            for vt in self._daemon.get_vt_iterator(vts_selection, vt_details):
+                yield xml_helper.add_element(self._daemon.get_vt_xml(vt))
 
         yield xml_helper.create_element('vts', end=True)
         yield xml_helper.create_response('get_vts', end=True)

tests/test_scan_and_result.py

@@ -153,6 +153,39 @@ def test_get_vts_single_vt(self):
         vt = vts.find('vt')
         self.assertEqual(vt.get('id'), '1.2.3.4')
 
+    def test_get_vts_version(self):
+        fs = FakeStream()
+        self.daemon.add_vt('1.2.3.4', 'A vulnerability test')
+        self.daemon.set_vts_version('today')
+        self.daemon.handle_command('<get_vts />', fs)
+        response = fs.get_response()
+
+        self.assertEqual(response.get('status'), '200')
+
+        vts_version = response.find('vts').attrib['vts_version']
+        self.assertEqual(vts_version, self.daemon.get_vts_version())
+
+        vts = response.find('vts')
+        self.assertIsNotNone(vts.find('vt'))
+
+        vt = vts.find('vt')
+        self.assertEqual(vt.get('id'), '1.2.3.4')
+
+    def test_get_vts_version_only(self):
+        fs = FakeStream()
+        self.daemon.add_vt('1.2.3.4', 'A vulnerability test')
+        self.daemon.set_vts_version('today')
+        self.daemon.handle_command('<get_vts version_only="1"/>', fs)
+        response = fs.get_response()
+
+        self.assertEqual(response.get('status'), '200')
+
+        vts_version = response.find('vts').attrib['vts_version']
+        self.assertEqual(vts_version, self.daemon.get_vts_version())
+
+        vts = response.find('vts')
+        self.assertIsNone(vts.find('vt'))
+
     def test_get_vts_still_not_init(self):
         fs = FakeStream()
         self.daemon.initialized = False