0ad3058c287babf1ba645732f7e63328ece6b415
[dcpomatic.git] / src / wx / dolby_doremi_certificate_panel.cc
1 /*
2     Copyright (C) 2014-2015 Carl Hetherington <cth@carlh.net>
3
4     This file is part of DCP-o-matic.
5
6     DCP-o-matic is free software; you can redistribute it and/or modify
7     it under the terms of the GNU General Public License as published by
8     the Free Software Foundation; either version 2 of the License, or
9     (at your option) any later version.
10
11     DCP-o-matic is distributed in the hope that it will be useful,
12     but WITHOUT ANY WARRANTY; without even the implied warranty of
13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14     GNU General Public License for more details.
15
16     You should have received a copy of the GNU General Public License
17     along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
18
19 */
20
21 #include "dolby_doremi_certificate_panel.h"
22 #include "download_certificate_dialog.h"
23 #include "wx_util.h"
24 #include "lib/compose.hpp"
25 #include "lib/util.h"
26 #include "lib/signal_manager.h"
27 #include "lib/internet.h"
28 #include <dcp/raw_convert.h>
29 #include <curl/curl.h>
30 #include <zip.h>
31 #include <boost/foreach.hpp>
32 #include <iostream>
33
34 using std::string;
35 using std::cout;
36 using std::list;
37 using boost::function;
38 using boost::optional;
39 using dcp::raw_convert;
40
41 DolbyDoremiCertificatePanel::DolbyDoremiCertificatePanel (DownloadCertificateDialog* dialog)
42         : DownloadCertificatePanel (dialog)
43 {
44
45 }
46
47 static void
48 try_dcp2000 (list<string>& urls, list<string>& files, string prefix, string serial)
49 {
50         urls.push_back (String::compose ("%1%2xxx/dcp2000-%3.dcicerts.zip", prefix, serial.substr(0, 3), serial));
51         files.push_back (String::compose ("dcp2000-%1.cert.sha256.pem", serial));
52
53         urls.push_back (String::compose ("%1%2xxx/dcp2000-%3.dcicerts.zip", prefix, serial.substr(0, 3), serial));
54         files.push_back (String::compose ("dcp2000-%1.cert.sha256.pem", serial));
55
56         urls.push_back (String::compose ("%1%2xxx/dcp2000-%3.certs.zip", prefix, serial.substr(0, 3), serial));
57         files.push_back (String::compose ("dcp2000-%1.cert.sha256.pem", serial));
58 }
59
60 static void
61 try_imb (list<string>& urls, list<string>& files, string prefix, string serial)
62 {
63         urls.push_back (String::compose ("%1%2xxx/imb-%3.dcicerts.zip", prefix, serial.substr(0, 3), serial));
64         files.push_back (String::compose ("imb-%1.cert.sha256.pem", serial));
65 }
66
67 static void
68 try_ims (list<string>& urls, list<string>& files, string prefix, string serial)
69 {
70         urls.push_back (String::compose ("%1%2xxx/ims-%3.dcicerts.zip", prefix, serial.substr(0, 3), serial));
71         files.push_back (String::compose ("ims-%1.cert.sha256.pem", serial));
72 }
73
74 static void
75 try_cat862 (list<string>& urls, list<string>& files, string prefix, string serial)
76 {
77         int const serial_int = raw_convert<int> (serial);
78
79         string cat862;
80         if (serial_int <= 510999) {
81                 cat862 = "CAT862_510999_and_lower";
82         } else if (serial_int >= 617000) {
83                 cat862 = "CAT862_617000_and_higher";
84         } else {
85                 int const lower = serial_int - (serial_int % 1000);
86                 cat862 = String::compose ("CAT862_%1-%2", lower, lower + 999);
87         }
88
89         urls.push_back (String::compose ("%1%2/cert_Dolby256-CAT862-%3.zip", prefix, cat862, serial_int));
90         files.push_back (String::compose ("cert_Dolby256-CAT862-%1.pem.crt", serial_int));
91 }
92
93 static void
94 try_dsp100 (list<string>& urls, list<string>& files, string prefix, string serial)
95 {
96         int const serial_int = raw_convert<int> (serial);
97
98         string dsp100;
99         if (serial_int <= 999) {
100                 dsp100 = "DSP100_053_thru_999";
101         } else if (serial_int >= 3000) {
102                 dsp100 = "DSP100_3000_and_higher";
103         } else {
104                 int const lower = serial_int - (serial_int % 1000);
105                 dsp100 = String::compose ("DSP100_%1_thru_%2", lower, lower + 999);
106         }
107
108         urls.push_back (String::compose ("%1%2/cert_Dolby256-DSP100-%3.zip", prefix, dsp100, serial_int));
109         files.push_back (String::compose ("cert_Dolby256-DSP100-%1.pem.crt", serial_int));
110 }
111
112 static void
113 try_cat745 (list<string>& urls, list<string>& files, string prefix, string serial)
114 {
115         int const serial_int = raw_convert<int> (serial.substr (1));
116
117         string cat745;
118         if (serial_int <= 999) {
119                 cat745 = "CAT745_1_thru_999";
120         } else if (serial_int >= 6000) {
121                 cat745 = "CAT745_6000_and_higher";
122         } else {
123                 int const lower = serial_int - (serial_int % 1000);
124                 cat745 = String::compose ("CAT745_%1_thru_%2", lower, lower + 999);
125         }
126
127         urls.push_back (String::compose ("%1%2/cert_Dolby-CAT745-%3.zip", prefix, cat745, serial_int));
128         files.push_back (String::compose ("cert_Dolby-CAT745-%1.pem.crt", serial_int));
129 }
130
131 static void
132 try_cp850 (list<string>& urls, list<string>& files, string prefix, string serial)
133 {
134         int const serial_int = raw_convert<int> (serial.substr (1));
135
136         int const lower = serial_int - (serial_int % 1000);
137         urls.push_back (String::compose ("%1CP850_CAT1600_F%2-F%3/cert_RMB_SPB_MDE_FMA.Dolby-CP850-F%4.zip", prefix, lower, lower + 999, serial_int));
138         files.push_back (String::compose ("cert_RMB_SPB_MDE_FMA.Dolby-CP850-F%1.pem.crt", serial_int));
139 }
140
141 void
142 DolbyDoremiCertificatePanel::do_download ()
143 {
144         string const serial = wx_to_std (_serial->GetValue());
145
146         /* Try dcp2000, imb and ims prefixes (see mantis #375) */
147
148         string const prefix = "ftp://ftp.cinema.dolby.com/Certificates/";
149         list<string> urls;
150         list<string> files;
151
152         bool starts_with_digit = false;
153         optional<char> starting_char;
154
155         if (!serial.empty()) {
156                 if (isdigit (serial[0])) {
157                         starts_with_digit = true;
158                 } else {
159                         starting_char = serial[0];
160                 }
161         }
162
163         if (starts_with_digit) {
164                 try_dcp2000 (urls, files, prefix, serial);
165                 try_imb (urls, files, prefix, serial);
166                 try_ims (urls, files, prefix, serial);
167                 try_cat862 (urls, files, prefix, serial);
168                 try_dsp100 (urls, files, prefix, serial);
169         } else if (starting_char == 'H') {
170                 try_cat745 (urls, files, prefix, serial);
171         } else if (starting_char == 'F') {
172                 try_cp850 (urls, files, prefix, serial);
173         }
174
175         list<string> errors;
176         bool ok = false;
177         list<string>::const_iterator i = urls.begin ();
178         list<string>::const_iterator j = files.begin ();
179         while (!ok && i != urls.end ()) {
180                 optional<string> error = get_from_zip_url (*i++, *j++, true, true, boost::bind (&DownloadCertificatePanel::load_certificate, this, _1));
181                 if (error) {
182                         errors.push_back (error.get ());
183                 } else {
184                         ok = true;
185                 }
186         }
187
188         if (ok) {
189                 _dialog->message()->SetLabel (_("Certificate downloaded"));
190                 _dialog->setup_sensitivity ();
191         } else {
192                 _dialog->message()->SetLabel (wxT (""));
193
194                 string s;
195                 BOOST_FOREACH (string e, errors) {
196                         s += e + "\n";
197                 }
198
199                 error_dialog (this, std_to_wx (s));
200         }
201 }
202
203 wxString
204 DolbyDoremiCertificatePanel::name () const
205 {
206         return _("Dolby / Doremi");
207 }