Mercurial > hg > orthanc-tests

comparison Tests/CheckDicomTls.py @ 375:08fac54c8555 Orthanc-1.9.0

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
CheckDicomTls.py
author Sebastien Jodogne <s.jodogne@gmail.com>
date Thu, 28 Jan 2021 15:52:41 +0100
parents
children f454fe86061b
comparison
equal deleted inserted replaced
374:78bef07c4118 375:08fac54c8555
1 #!/usr/bin/python
2
3 # Orthanc - A Lightweight, RESTful DICOM Store
4 # Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
5 # Department, University Hospital of Liege, Belgium
6 # Copyright (C) 2017-2021 Osimis S.A., Belgium
7 #
8 # This program is free software: you can redistribute it and/or
9 # modify it under the terms of the GNU General Public License as
10 # published by the Free Software Foundation, either version 3 of the
11 # License, or (at your option) any later version.
12 #
13 # This program is distributed in the hope that it will be useful, but
14 # WITHOUT ANY WARRANTY; without even the implied warranty of
15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 # General Public License for more details.
17 #
18 # You should have received a copy of the GNU General Public License
19 # along with this program. If not, see <http://www.gnu.org/licenses/>.
20
21
22
23 import argparse
24 import os
25 import pprint
26 import re
27 import sys
28 import subprocess
29 import unittest
30
31 from Toolbox import *
32
33
34 ##
35 ## Parse the command-line arguments
36 ##
37
38 parser = argparse.ArgumentParser(description = 'Run the integration tests for DICOM TLS in Orthanc.')
39
40 parser.add_argument('--server',
41 default = 'localhost',
42 help = 'Address of the Orthanc server to test')
43 parser.add_argument('--aet',
44 default = 'ORTHANC',
45 help = 'AET of the Orthanc instance to test')
46 parser.add_argument('--dicom',
47 type = int,
48 default = 4242,
49 help = 'DICOM port of the Orthanc instance to test')
50 parser.add_argument('--rest',
51 type = int,
52 default = 8042,
53 help = 'Port to the REST API')
54 parser.add_argument('--username',
55 default = 'alice',
56 help = 'Username to the REST API')
57 parser.add_argument('--password',
58 default = 'orthanctest',
59 help = 'Password to the REST API')
60 parser.add_argument('--force', help = 'Do not warn the user',
61 action = 'store_true')
62 parser.add_argument('--config', help = 'Create the configuration files for this test in the current folder',
63 action = 'store_true')
64 parser.add_argument('options', metavar = 'N', nargs = '*',
65 help='Arguments to Python unittest')
66
67 args = parser.parse_args()
68
69
70 ##
71 ## Configure the testing context
72 ##
73
74
75 if args.config:
76 def CreateCertificate(name):
77 subprocess.check_call([ 'openssl', 'req', '-x509', '-nodes', '-days', '365', '-newkey', 'rsa:2048',
78 '-keyout', '%s.key' % name,
79 '-out', '%s.crt' % name,
80 '-subj', '/C=BE/CN=localhost' ])
81
82 print('Writing configuration to folder: %s' % args.config)
83 CreateCertificate('dicom-tls-a')
84 CreateCertificate('dicom-tls-b')
85 CreateCertificate('dicom-tls-c') # Not trusted by Orthanc
86
87 with open('dicom-tls-trusted.crt', 'w') as f:
88 for i in [ 'dicom-tls-a.crt', 'dicom-tls-b.crt' ]:
89 with open(i, 'r') as g:
90 f.write(g.read())
91
92 with open('dicom-tls.json', 'w') as f:
93 f.write(json.dumps({
94 'DicomTlsEnabled' : True,
95 'DicomTlsCertificate' : 'dicom-tls-a.crt',
96 'DicomTlsPrivateKey' : 'dicom-tls-a.key',
97 'DicomTlsTrustedCertificates' : 'dicom-tls-trusted.crt',
98 'ExecuteLuaEnabled' : True,
99 'RemoteAccessAllowed' : True,
100 'RegisteredUsers' : {
101 'alice' : 'orthanctest'
102 },
103 }))
104
105 exit(0)
106
107
108 if not args.force:
109 print("""
110 WARNING: This test will remove all the content of your
111 Orthanc instance running on %s!
112
113 Are you sure ["yes" to go on]?""" % args.server)
114
115 if sys.stdin.readline().strip() != 'yes':
116 print('Aborting...')
117 exit(0)
118
119
120 ORTHANC = DefineOrthanc(server = args.server,
121 username = args.username,
122 password = args.password,
123 restPort = args.rest,
124 aet = args.aet,
125 dicomPort = args.dicom)
126
127
128 ##
129 ## The tests
130 ##
131
132
133 FNULL = open(os.devnull, 'w') # Emulates "subprocess.DEVNULL" on Python 2.7
134
135
136 class Orthanc(unittest.TestCase):
137 def setUp(self):
138 if (sys.version_info >= (3, 0)):
139 # Remove annoying warnings about unclosed socket in Python 3
140 import warnings
141 warnings.simplefilter('ignore', ResourceWarning)
142
143 DropOrthanc(ORTHANC)
144
145
146 def test_incoming(self):
147 # No certificate
148 self.assertRaises(Exception, lambda: subprocess.check_call([
149 FindExecutable('echoscu'),
150 ORTHANC['Server'],
151 str(ORTHANC['DicomPort']),
152 '-aec', 'ORTHANC',
153 ], stderr = FNULL))
154
155 subprocess.check_call([
156 FindExecutable('echoscu'),
157 ORTHANC['Server'],
158 str(ORTHANC['DicomPort']),
159 '-aec', 'ORTHANC',
160 '+tls', 'dicom-tls-b.key', 'dicom-tls-b.crt',
161 '+cf', 'dicom-tls-a.crt',
162 ], stderr = FNULL)
163
164 self.assertRaises(Exception, lambda: subprocess.check_call([
165 FindExecutable('echoscu'),
166 ORTHANC['Server'],
167 str(ORTHANC['DicomPort']),
168 '-aec', 'ORTHANC',
169 '+tls', 'dicom-tls-c.key', 'dicom-tls-c.crt', # Not trusted by Orthanc
170 '+cf', 'dicom-tls-a.crt',
171 ], stderr = FNULL))
172
173 self.assertRaises(Exception, lambda: subprocess.check_call([
174 FindExecutable('echoscu'),
175 ORTHANC['Server'],
176 str(ORTHANC['DicomPort']),
177 '-aec', 'ORTHANC',
178 '+tls', 'dicom-tls-b.key', 'dicom-tls-b.crt',
179 '+cf', 'dicom-tls-b.crt', # Not the certificate of Orthanc
180 ], stderr = FNULL))
181
182
183 def test_outgoing_to_self(self):
184 u = UploadInstance(ORTHANC, 'DummyCT.dcm') ['ID']
185
186 # Error, as DICOM TLS is not enabled
187 DoPut(ORTHANC, '/modalities/self', {
188 'AET' : 'ORTHANC',
189 'Host' : ORTHANC['Server'],
190 'Port' : ORTHANC['DicomPort'],
191 })
192
193 self.assertRaises(Exception, lambda: DoPost(ORTHANC, '/modalities/self/store', u))
194
195 # Retry using DICOM TLS
196 DoPut(ORTHANC, '/modalities/self', {
197 'AET' : 'ORTHANC',
198 'Host' : ORTHANC['Server'],
199 'Port' : ORTHANC['DicomPort'],
200 'UseDicomTls' : True,
201 })
202
203 self.assertEqual(1, DoPost(ORTHANC, '/modalities/self/store', u) ['InstancesCount'])
204
205
206 try:
207 print('\nStarting the tests...')
208 unittest.main(argv = [ sys.argv[0] ] + args.options)
209
210 finally:
211 print('\nDone')