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