BOOST_FOREACH.
[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 <iostream>
32
33 using std::string;
34 using std::cout;
35 using std::list;
36 using boost::function;
37 using boost::optional;
38 #if BOOST_VERSION >= 106100
39 using namespace boost::placeholders;
40 #endif
41 using dcp::raw_convert;
42
43 DolbyDoremiCertificatePanel::DolbyDoremiCertificatePanel (DownloadCertificateDialog* dialog)
44         : DownloadCertificatePanel (dialog)
45 {
46
47 }
48
49 static void
50 try_dcp2000 (list<string>& urls, list<string>& files, string prefix, string serial)
51 {
52         urls.push_back (String::compose ("%1%2xxx/dcp2000-%3.dcicerts.zip", prefix, serial.substr(0, 3), serial));
53         files.push_back (String::compose ("dcp2000-%1.cert.sha256.pem", serial));
54
55         urls.push_back (String::compose ("%1%2xxx/dcp2000-%3.dcicerts.zip", prefix, serial.substr(0, 3), serial));
56         files.push_back (String::compose ("dcp2000-%1.cert.sha256.pem", serial));
57
58         urls.push_back (String::compose ("%1%2xxx/dcp2000-%3.certs.zip", prefix, serial.substr(0, 3), serial));
59         files.push_back (String::compose ("dcp2000-%1.cert.sha256.pem", serial));
60 }
61
62 static void
63 try_imb (list<string>& urls, list<string>& files, string prefix, string serial)
64 {
65         urls.push_back (String::compose ("%1%2xxx/imb-%3.dcicerts.zip", prefix, serial.substr(0, 3), serial));
66         files.push_back (String::compose ("imb-%1.cert.sha256.pem", serial));
67 }
68
69 static void
70 try_ims (list<string>& urls, list<string>& files, string prefix, string serial)
71 {
72         urls.push_back (String::compose ("%1%2xxx/ims-%3.dcicerts.zip", prefix, serial.substr(0, 3), serial));
73         files.push_back (String::compose ("ims-%1.cert.sha256.pem", serial));
74 }
75
76 static void
77 try_cat862 (list<string>& urls, list<string>& files, string prefix, string serial)
78 {
79         int const serial_int = raw_convert<int> (serial);
80
81         string cat862;
82         if (serial_int <= 510999) {
83                 cat862 = "CAT862_510999_and_lower";
84         } else if (serial_int >= 617000) {
85                 cat862 = "CAT862_617000_and_higher";
86         } else {
87                 int const lower = serial_int - (serial_int % 1000);
88                 cat862 = String::compose ("CAT862_%1-%2", lower, lower + 999);
89         }
90
91         urls.push_back (String::compose ("%1%2/cert_Dolby256-CAT862-%3.zip", prefix, cat862, serial_int));
92         files.push_back (String::compose ("cert_Dolby256-CAT862-%1.pem.crt", serial_int));
93 }
94
95 static void
96 try_dsp100 (list<string>& urls, list<string>& files, string prefix, string serial)
97 {
98         int const serial_int = raw_convert<int> (serial);
99
100         string dsp100;
101         if (serial_int <= 999) {
102                 dsp100 = "DSP100_053_thru_999";
103         } else if (serial_int >= 3000) {
104                 dsp100 = "DSP100_3000_and_higher";
105         } else {
106                 int const lower = serial_int - (serial_int % 1000);
107                 dsp100 = String::compose ("DSP100_%1_thru_%2", lower, lower + 999);
108         }
109
110         urls.push_back (String::compose ("%1%2/cert_Dolby256-DSP100-%3.zip", prefix, dsp100, serial_int));
111         files.push_back (String::compose ("cert_Dolby256-DSP100-%1.pem.crt", serial_int));
112 }
113
114 static void
115 try_cat745 (list<string>& urls, list<string>& files, string prefix, string serial)
116 {
117         int const serial_int = raw_convert<int> (serial.substr (1));
118
119         string cat745;
120         if (serial_int <= 999) {
121                 cat745 = "CAT745_1_thru_999";
122         } else if (serial_int >= 6000) {
123                 cat745 = "CAT745_6000_and_higher";
124         } else {
125                 int const lower = serial_int - (serial_int % 1000);
126                 cat745 = String::compose ("CAT745_%1_thru_%2", lower, lower + 999);
127         }
128
129         urls.push_back (String::compose ("%1%2/cert_Dolby-CAT745-%3.zip", prefix, cat745, serial_int));
130         files.push_back (String::compose ("cert_Dolby-CAT745-%1.pem.crt", serial_int));
131 }
132
133 static void
134 try_cp850 (list<string>& urls, list<string>& files, string prefix, string serial)
135 {
136         int const serial_int = raw_convert<int> (serial.substr (1));
137
138         int const lower = serial_int - (serial_int % 1000);
139         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));
140         files.push_back (String::compose ("cert_RMB_SPB_MDE_FMA.Dolby-CP850-F%1.pem.crt", serial_int));
141 }
142
143 void
144 DolbyDoremiCertificatePanel::do_download ()
145 {
146         string const serial = wx_to_std (_serial->GetValue());
147
148         /* Try dcp2000, imb and ims prefixes (see mantis #375) */
149
150         string const prefix = "ftp://ftp.cinema.dolby.com/Certificates/";
151         list<string> urls;
152         list<string> files;
153
154         bool starts_with_digit = false;
155         optional<char> starting_char;
156
157         if (!serial.empty()) {
158                 if (isdigit (serial[0])) {
159                         starts_with_digit = true;
160                 } else {
161                         starting_char = serial[0];
162                 }
163         }
164
165         if (starts_with_digit) {
166                 try_dcp2000 (urls, files, prefix, serial);
167                 try_imb (urls, files, prefix, serial);
168                 try_ims (urls, files, prefix, serial);
169                 try_cat862 (urls, files, prefix, serial);
170                 try_dsp100 (urls, files, prefix, serial);
171         } else if (starting_char == 'H') {
172                 try_cat745 (urls, files, prefix, serial);
173         } else if (starting_char == 'F') {
174                 try_cp850 (urls, files, prefix, serial);
175         }
176
177         list<string> errors;
178         bool ok = false;
179         list<string>::const_iterator i = urls.begin ();
180         list<string>::const_iterator j = files.begin ();
181         while (!ok && i != urls.end ()) {
182                 optional<string> error = get_from_zip_url (*i++, *j++, true, true, boost::bind (&DownloadCertificatePanel::load_certificate, this, _1));
183                 if (error) {
184                         errors.push_back (error.get ());
185                 } else {
186                         ok = true;
187                 }
188         }
189
190         if (ok) {
191                 _dialog->message()->SetLabel (_("Certificate downloaded"));
192                 _dialog->setup_sensitivity ();
193         } else {
194                 _dialog->message()->SetLabel (wxT (""));
195
196                 string s;
197                 for (auto e: errors) {
198                         s += e + "\n";
199                 }
200
201                 error_dialog (this, std_to_wx (s));
202         }
203 }
204
205 wxString
206 DolbyDoremiCertificatePanel::name () const
207 {
208         return _("Dolby / Doremi");
209 }