mcs/tools/security/mozroots.cs

   1 //
   2 // mozroots.cs: Import the Mozilla's trusted root certificates into Mono
   3 //
   4 // Authors:
   5 //      Sebastien Pouliot <sebastien@ximian.com>
   6 //
   7 // Copyright (C) 2005 Novell, Inc (http://www.novell.com)
   8 //
   9 // Permission is hereby granted, free of charge, to any person obtaining
  10 // a copy of this software and associated documentation files (the
  11 // "Software"), to deal in the Software without restriction, including
  12 // without limitation the rights to use, copy, modify, merge, publish,
  13 // distribute, sublicense, and/or sell copies of the Software, and to
  14 // permit persons to whom the Software is furnished to do so, subject to
  15 // the following conditions:
  16 //
  17 // The above copyright notice and this permission notice shall be
  18 // included in all copies or substantial portions of the Software.
  19 //
  20 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  21 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  22 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  23 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  24 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  25 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  26 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  27 //
  28
  29 using System;
  30 using System.Collections;
  31 using System.IO;
  32 using System.Net;
  33 using System.Reflection;
  34 using System.Security.Cryptography;
  35 using System.Text;
  36
  37 using Mono.Security.Authenticode;
  38 using Mono.Security.X509;
  39
  40 [assembly: AssemblyTitle ("Mozilla Roots Importer")]
  41 [assembly: AssemblyDescription ("Download and import trusted root certificates from Mozilla's LXR.")]
  42
  43 namespace Mono.Tools {
  44
  45         class MozRoots {
  46
  47                 private const string defaultUrl = "http://lxr.mozilla.org/seamonkey/source/security/nss/lib/ckfw/builtins/certdata.txt";
  48
  49                 static string url;
  50                 static string inputFile;
  51                 static string pkcs7filename;
  52                 static bool import;
  53                 static bool machine;
  54                 static bool confirmAddition;
  55                 static bool confirmRemoval;
  56                 static bool quiet;
  57
  58                 static byte[] DecodeOctalString (string s)
  59                 {
  60                         string[] pieces = s.Split ('\\');
  61                         byte[] data = new byte[pieces.Length - 1];
  62                         for (int i = 1; i < pieces.Length; i++) {
  63                                 data[i - 1] = (byte) ((pieces[i][0] - '0' << 6) + (pieces[i][1] - '0' << 3) + (pieces[i][2] - '0'));
  64                         }
  65                         return data;
  66                 }
  67
  68                 static X509Certificate DecodeCertificate (string s)
  69                 {
  70                         byte[] rawdata = DecodeOctalString (s);
  71                         return new X509Certificate (rawdata);
  72                 }
  73
  74                 static Stream GetFile ()
  75                 {
  76                         try {
  77                                 if (inputFile != null) {
  78                                         return File.OpenRead (inputFile);
  79                                 } else {
  80                                         WriteLine ("Downloading from '{0}'...", url);
  81                                         HttpWebRequest req = (HttpWebRequest) WebRequest.Create (url);
  82                                         return req.GetResponse ().GetResponseStream ();
  83                                 }
  84                         }
  85                         catch {
  86                                 return null;
  87                         }
  88                 }
  89
  90                 static X509CertificateCollection DecodeCollection ()
  91                 {
  92                         X509CertificateCollection roots = new X509CertificateCollection ();
  93                         StringBuilder sb = new StringBuilder ();
  94                         bool processing = false;
  95
  96                         Stream s = GetFile ();
  97                         if (s == null) {
  98                                 WriteLine ("Couldn't retrieve the file using the supplied informations.", null);
  99                                 return null;
 100                         }
 101
 102                         StreamReader sr = new StreamReader (s);
 103                         while (true) {
 104                                 string line = sr.ReadLine ();
 105                                 if (line == null)
 106                                         break;
 107                                 int start = line.IndexOf ("</a> ");
 108                                 if (start < 0)
 109                                         continue;
 110
 111                                 if (processing) {
 112                                         if (line.IndexOf ("END") > start) {
 113                                                 processing = false;
 114                                                 X509Certificate root = DecodeCertificate (sb.ToString ());
 115                                                 roots.Add (root);
 116
 117                                                 sb = new StringBuilder ();
 118                                                 continue;
 119                                         }
 120                                         sb.Append (line.Substring (start + 5));
 121                                 } else {
 122                                         processing = (line.IndexOf ("CKA_VALUE MULTILINE_OCTAL") > start);
 123                                 }
 124                         }
 125                         return roots;
 126                 }
 127
 128                 static int Process ()
 129                 {
 130                         X509CertificateCollection roots = DecodeCollection ();
 131                         if (roots == null) {
 132                                 return 1;
 133                         } else if (roots.Count == 0) {
 134                                 WriteLine ("No certificates were found.");
 135                                 return 0;
 136                         }
 137
 138                         if (pkcs7filename != null) {
 139                                 SoftwarePublisherCertificate pkcs7 = new SoftwarePublisherCertificate ();
 140                                 pkcs7.Certificates.AddRange (roots);
 141
 142                                 WriteLine ("Saving root certificates into '{0}' file...", pkcs7filename);
 143                                 using (FileStream fs = File.OpenWrite (pkcs7filename)) {
 144                                         byte[] data = pkcs7.GetBytes ();
 145                                         fs.Write (data, 0, data.Length);
 146                                         fs.Close ();
 147                                 }
 148                         }
 149
 150                         if (import) {
 151                                 WriteLine ("Importing certificates into {0} store...",
 152                                         machine ? "machine" : "user");
 153
 154                                 X509Stores stores = (machine ? X509StoreManager.LocalMachine : X509StoreManager.CurrentUser);
 155                                 X509CertificateCollection trusted = stores.TrustedRoot.Certificates;
 156                                 int additions = 0;
 157                                 foreach (X509Certificate root in roots) {
 158                                         if (!trusted.Contains (root)) {
 159                                                 if (!confirmAddition || AskConfirmation ("add", root)) {
 160                                                         trusted.Add (root);
 161                                                         if (confirmAddition)
 162                                                                 WriteLine ("Certificate added.{0}", Environment.NewLine);
 163                                                         additions++;
 164                                                 }
 165                                         }
 166                                 }
 167                                 if (additions > 0)
 168                                         WriteLine ("{0} new root certificates were added to your trust store.", additions);
 169
 170                                 X509CertificateCollection removed = new X509CertificateCollection ();
 171                                 foreach (X509Certificate trust in trusted) {
 172                                         if (!roots.Contains (trust)) {
 173                                                 removed.Add (trust);
 174                                         }
 175                                 }
 176                                 if (removed.Count > 0) {
 177                                         if (confirmRemoval) {
 178                                                 WriteLine ("{0} previously trusted certificates were not part of the update.");
 179                                         } else {
 180                                                 WriteLine ("{0} previously trusted certificates were removed.");
 181                                         }
 182
 183                                         foreach (X509Certificate old in removed) {
 184                                                 if (!confirmRemoval || AskConfirmation ("remove", old)) {
 185                                                         stores.TrustedRoot.Remove (old);
 186                                                         if (confirmRemoval)
 187                                                                 WriteLine ("Certificate removed.{0}", Environment.NewLine);
 188                                                 }
 189                                         }
 190                                 }
 191                         }
 192                         return 0;
 193                 }
 194
 195                 static string Thumbprint (string algorithm, X509Certificate certificate)
 196                 {
 197                         HashAlgorithm hash = HashAlgorithm.Create (algorithm);
 198                         byte[] digest = hash.ComputeHash (certificate.RawData);
 199                         return BitConverter.ToString (digest);
 200                 }
 201
 202                 static bool AskConfirmation (string action, X509Certificate certificate)
 203                 {
 204                         // the quiet flag is ignored for confirmations
 205                         Console.WriteLine ("Issuer: {0}", certificate.IssuerName);
 206                         Console.WriteLine ("Serial number: {0}", BitConverter.ToString (certificate.SerialNumber));
 207                         Console.WriteLine ("Valid from {0} to {1}", certificate.ValidFrom, certificate.ValidUntil);
 208                         Console.WriteLine ("Thumbprint SHA-1: {0}", Thumbprint ("SHA1", certificate));
 209                         Console.WriteLine ("Thumbprint MD5:   {0}", Thumbprint ("MD5", certificate));
 210                         while (true) {
 211                                 Console.Write ("Are you sure you want to {0} this certificate ? ", action);
 212                                 string s = Console.ReadLine ().ToLower ();
 213                                 if (s == "yes")
 214                                         return true;
 215                                 else if (s == "no")
 216                                         return false;
 217                         }
 218                 }
 219
 220                 static bool ParseOptions (string[] args)
 221                 {
 222                         if (args.Length < 1)
 223                                 return false;
 224
 225                         // set defaults
 226                         url = defaultUrl;
 227                         confirmAddition = true;
 228                         confirmRemoval = true;
 229
 230                         for (int i = 0; i < args.Length; i++) {
 231                                 switch (args[i]) {
 232                                 case "--url":
 233                                         if (i >= args.Length - 1)
 234                                                 return false;
 235                                         url = args[++i];
 236                                         break;
 237                                 case "--file":
 238                                         if (i >= args.Length - 1)
 239                                                 return false;
 240                                         inputFile = args[++i];
 241                                         break;
 242                                 case "--pkcs7":
 243                                         if (i >= args.Length - 1)
 244                                                 return false;
 245                                         pkcs7filename = args[++i];
 246                                         break;
 247                                 case "--import":
 248                                         import = true;
 249                                         break;
 250                                 case "--machine":
 251                                         machine = true;
 252                                         break;
 253                                 case "--sync":
 254                                         confirmAddition = false;
 255                                         confirmRemoval = false;
 256                                         break;
 257                                 case "--ask":
 258                                         confirmAddition = true;
 259                                         confirmRemoval = true;
 260                                         break;
 261                                 case "--ask-add":
 262                                         confirmRemoval = true;
 263                                         break;
 264                                 case "--ask-remove":
 265                                         confirmRemoval = true;
 266                                         break;
 267                                 case "--quiet":
 268                                         quiet = true;
 269                                         break;
 270                                 default:
 271                                         WriteLine ("Unknown option '{0}'.");
 272                                         return false;
 273                                 }
 274                         }
 275                         return true;
 276                 }
 277
 278                 static void Header ()
 279                 {
 280                         Console.WriteLine (new AssemblyInfo ().ToString ());
 281                 }
 282
 283                 static void Help ()
 284                 {
 285                         Console.WriteLine ("Usage: mozroots [--import [--machine] [--sync | --ask | --ask-add | --ask-remove]]");
 286                         Console.WriteLine ("Where the basic options are:");
 287                         Console.WriteLine (" --import\tImport the certificates into the trust store.");
 288                         Console.WriteLine (" --sync\t\tSynchronize (add/remove) the trust store with the certificates.");
 289                         Console.WriteLine (" --ask\t\tAlways confirm before adding or removing trusted certificates.");
 290                         Console.WriteLine (" --ask-add\tAlways confirm before adding a new trusted certificate.");
 291                         Console.WriteLine (" --ask-remove\tAlways confirm before removing an existing trusted certificate.");
 292                         Console.WriteLine ("{0}and the advanced options are", Environment.NewLine);
 293                         Console.WriteLine (" --url url\tSpecify an alternative URL for downloading the trusted");
 294                         Console.WriteLine ("\t\tcertificates (LXR source format).");
 295                         Console.WriteLine (" --file name\tDo not download but use the specified file.");
 296                         Console.WriteLine (" --pkcs7 name\tExport the certificates into a PKCS#7 file.");
 297                         Console.WriteLine (" --machine\tImport the certificate in the machine trust store.");
 298                         Console.WriteLine ("\t\tThe default is to import into the user store.");
 299                         Console.WriteLine (" --quiet\tLimit console output to errors and confirmations messages.");
 300                 }
 301
 302                 static void WriteLine (string format, params object[] args)
 303                 {
 304                         if (!quiet)
 305                                 Console.WriteLine (format, args);
 306                 }
 307
 308                 static int Main (string[] args)
 309                 {
 310                         try {
 311                                 if (!ParseOptions (args)) {
 312                                         Header ();
 313                                         Help ();
 314                                         return 1;
 315                                 }
 316                                 if (!quiet) {
 317                                         Header ();
 318                                 }
 319                                 return Process ();
 320                         }
 321                         catch (Exception e) {
 322                                 // ignore quiet on exception
 323                                 Console.WriteLine ("Error: {0}", e);
 324                                 return 1;
 325                         }
 326                 }
 327         }
 328 }