diff options
Diffstat (limited to 'src/wx/doremi_certificate_dialog.cc')
| -rw-r--r-- | src/wx/doremi_certificate_dialog.cc | 126 |
1 files changed, 126 insertions, 0 deletions
diff --git a/src/wx/doremi_certificate_dialog.cc b/src/wx/doremi_certificate_dialog.cc new file mode 100644 index 000000000..51f2a3579 --- /dev/null +++ b/src/wx/doremi_certificate_dialog.cc @@ -0,0 +1,126 @@ +/* + Copyright (C) 2014 Carl Hetherington <cth@carlh.net> + + This program is free software; you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation; either version 2 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program; if not, write to the Free Software + Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. + +*/ + +#include <curl/curl.h> +#include <zip.h> +#include "lib/compose.hpp" +#include "lib/util.h" +#include "doremi_certificate_dialog.h" +#include "wx_util.h" + +using std::string; +using boost::function; + +DoremiCertificateDialog::DoremiCertificateDialog (wxWindow* parent, function<void (boost::filesystem::path)> load) + : DownloadCertificateDialog (parent, load) +{ + wxFlexGridSizer* table = new wxFlexGridSizer (2, DCPOMATIC_SIZER_X_GAP, DCPOMATIC_SIZER_Y_GAP); + + add_label_to_sizer (table, this, _("Server serial number"), true); + _serial = new wxTextCtrl (this, wxID_ANY); + table->Add (_serial, 1, wxEXPAND); + + _overall_sizer->Add (table, 1, wxALL, DCPOMATIC_DIALOG_BORDER); + + add_common_widgets (); +} + + + +static size_t +ftp_data (void* buffer, size_t size, size_t nmemb, void* stream) +{ + FILE* f = reinterpret_cast<FILE*> (stream); + return fwrite (buffer, size, nmemb, f); +} + +void +DoremiCertificateDialog::download () +{ + string const serial = wx_to_std (_serial->GetValue ()); + if (serial.length() != 6) { + _message->SetLabel (_("Doremi serial numbers must have 6 digits")); + return; + } + + CURL* curl = curl_easy_init (); + if (!curl) { + _message->SetLabel (N_("Could not set up libcurl")); + return; + } + + string const url = String::compose ( + "ftp://service:t3chn1c1an@ftp.doremilabs.com/Certificates/%1xxx/dcp2000-%2.dcicerts.zip", + serial.substr(0, 3), serial + ); + + curl_easy_setopt (curl, CURLOPT_URL, url.c_str ()); + + ScopedTemporary temp_zip; + FILE* f = temp_zip.open ("wb"); + curl_easy_setopt (curl, CURLOPT_WRITEFUNCTION, ftp_data); + curl_easy_setopt (curl, CURLOPT_WRITEDATA, f); + + _message->SetLabel (_("Downloading certificate from Doremi")); + run_gui_loop (); + + CURLcode const cr = curl_easy_perform (curl); + + _gauge->SetValue (50); + run_gui_loop (); + + temp_zip.close (); + curl_easy_cleanup (curl); + if (cr != CURLE_OK) { + _message->SetLabel (wxString::Format (_("Certificate download failed (%d)"), cr)); + return; + } + + _message->SetLabel (_("Unpacking")); + run_gui_loop (); + + struct zip* zip = zip_open (temp_zip.c_str(), 0, 0); + if (!zip) { + _message->SetLabel (N_("Could not open certificate ZIP file")); + return; + } + + string const name_in_zip = String::compose ("dcp2000-%1.cert.sha256.pem", serial); + struct zip_file* zip_file = zip_fopen (zip, name_in_zip.c_str(), 0); + if (!zip_file) { + _message->SetLabel (N_("Could not find certificate in ZIP file")); + return; + } + + ScopedTemporary temp_cert; + f = temp_cert.open ("wb"); + char buffer[4096]; + while (1) { + int const N = zip_fread (zip_file, buffer, sizeof (buffer)); + fwrite (buffer, 1, N, f); + if (N < int (sizeof (buffer))) { + break; + } + } + temp_cert.close (); + + _gauge->SetValue (100); + _message->SetLabel (_("OK")); + _load (temp_cert.file ()); +} |