src/moonshot-local-flat-file-store.vala

   1 /*
   2  * Copyright (c) 2011-2016, JANET(UK)
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  *
   9  * 1. Redistributions of source code must retain the above copyright
  10  *    notice, this list of conditions and the following disclaimer.
  11  *
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  *
  16  * 3. Neither the name of JANET(UK) nor the names of its contributors
  17  *    may be used to endorse or promote products derived from this software
  18  *    without specific prior written permission.
  19  *
  20  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  21  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  23  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
  24  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  25  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  26  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  28  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  29  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  30  * SUCH DAMAGE.
  31  */
  32 using Gee;
  33
  34 public class LocalFlatFileStore : Object, IIdentityCardStore {
  35     static MoonshotLogger logger = get_logger("LocalFlatFileStore");
  36
  37     private LinkedList<IdCard> id_card_list;
  38     private const string FILE_NAME = "identities.txt";
  39
  40     public void add_card(IdCard card) {
  41         id_card_list.add(card);
  42         store_id_cards();
  43     }
  44
  45     public IdCard? update_card(IdCard card) {
  46         id_card_list.remove(card);
  47         id_card_list.add(card);
  48         store_id_cards();
  49         foreach(IdCard idcard in id_card_list) {
  50             if (idcard.display_name == card.display_name) {
  51                 return idcard;
  52             }
  53         }
  54         logger.error(@"update_card: card '$(card.display_name)' was not found after re-loading!");
  55         return null;
  56     }
  57
  58     public bool remove_card(IdCard card) {
  59         if (id_card_list.remove(card)) {
  60             store_id_cards();
  61             return true;
  62         }
  63         return false;
  64     }
  65
  66     public LinkedList<IdCard> get_card_list() {
  67         return id_card_list;
  68     }
  69
  70     public IIdentityCardStore.StoreType get_store_type() {
  71         return IIdentityCardStore.StoreType.FLAT_FILE;
  72     }
  73
  74     private void load_id_cards() {
  75         id_card_list.clear();
  76         var key_file = new KeyFile();
  77         var path = get_data_dir();
  78         var filename = Path.build_filename(path, FILE_NAME);
  79         logger.trace("load_id_cards: attempting to load from " + filename);
  80
  81         try {
  82             key_file.load_from_file(filename, KeyFileFlags.NONE);
  83         }
  84         catch (Error e) {
  85             stdout.printf("Error while attempting to load from %s: %s\n", filename, e.message);
  86             return;
  87         }
  88
  89         var identities_uris = key_file.get_groups();
  90         foreach (string identity in identities_uris) {
  91             try {
  92                 IdCard id_card = new IdCard();
  93
  94                 id_card.issuer = key_file.get_string(identity, "Issuer");
  95                 id_card.username = key_file.get_string(identity, "Username");
  96                 id_card.password = key_file.get_string(identity, "Password");
  97                 id_card.update_services(key_file.get_string_list(identity, "Services"));
  98                 id_card.display_name = key_file.get_string(identity, "DisplayName");
  99                 if (key_file.has_key(identity, "StorePassword")) {
 100                     id_card.store_password = (key_file.get_string(identity, "StorePassword") == "yes");
 101                 } else {
 102                     id_card.store_password = (id_card.password != null) && (id_card.password != "");
 103                 }
 104
 105                 if (key_file.has_key(identity, "Rules-Patterns") &&
 106                     key_file.has_key(identity, "Rules-AlwaysConfirm")) {
 107                     string [] rules_patterns =    key_file.get_string_list(identity, "Rules-Patterns");
 108                     string [] rules_always_conf = key_file.get_string_list(identity, "Rules-AlwaysConfirm");
 109
 110                     if (rules_patterns.length == rules_always_conf.length) {
 111                         Rule[] rules = new Rule[rules_patterns.length];
 112                         for (int i = 0; i < rules_patterns.length; i++) {
 113                             rules[i] = {rules_patterns[i], rules_always_conf[i]};
 114                         }
 115                         id_card.rules = rules;
 116                     }
 117                 }
 118
 119                 // Trust anchor
 120                 string ca_cert = key_file.get_string(identity, "CA-Cert").strip();
 121                 string server_cert = key_file.get_string(identity, "ServerCert");
 122                 string subject = key_file.get_string(identity, "Subject");
 123                 string subject_alt = key_file.get_string(identity, "SubjectAlt");
 124                 var ta = new TrustAnchor(ca_cert, server_cert, subject, subject_alt);
 125                 string ta_datetime_added = get_string_setting(identity, "TA_DateTime_Added", "", key_file);
 126                 if (ta_datetime_added != "") {
 127                     ta.set_datetime_added(ta_datetime_added);
 128                 }
 129                 id_card.set_trust_anchor_from_store(ta);
 130                 id_card_list.add(id_card);
 131             }
 132             catch (Error e) {
 133                 logger.error("load_id_cards: Error while loading keyfile %s: %s\n".printf(filename, e.message));
 134                 stdout.printf("Error while attempting to load from %s: %s\n", filename, e.message);
 135             }
 136         }
 137     }
 138
 139     private string get_data_dir() {
 140         string path;
 141         path = Path.build_filename(Environment.get_user_data_dir(),
 142                                    Config.PACKAGE_TARNAME);
 143
 144         if (!FileUtils.test(path, FileTest.EXISTS)) {
 145             DirUtils.create_with_parents(path, 0700);
 146         }
 147         return path;
 148     }
 149
 150     internal void store_id_cards() {
 151         var key_file = new KeyFile();
 152         foreach (IdCard id_card in this.id_card_list) {
 153             logger.trace(@"store_id_cards: Storing '$(id_card.display_name)'");
 154
 155             /* workaround for Centos vala array property bug: use temp arrays */
 156             var rules = id_card.rules;
 157             string[] rules_patterns = new string[rules.length];
 158             string[] rules_always_conf = new string[rules.length];
 159
 160             for (int i = 0; i < rules.length; i++) {
 161                 rules_patterns[i] = rules[i].pattern;
 162                 rules_always_conf[i] = rules[i].always_confirm;
 163             }
 164
 165             key_file.set_string(id_card.display_name, "Issuer", id_card.issuer ?? "");
 166             key_file.set_string(id_card.display_name, "DisplayName", id_card.display_name ?? "");
 167             key_file.set_string(id_card.display_name, "Username", id_card.username ?? "");
 168             if (id_card.store_password && (id_card.password != null))
 169                 key_file.set_string(id_card.display_name, "Password", id_card.password);
 170             else
 171                 key_file.set_string(id_card.display_name, "Password", "");
 172
 173             // Using id_card.services.to_array() seems to cause a crash, possibly due to
 174             // an unowned reference to the array.
 175             string[] svcs = new string[id_card.services.size];
 176             for (int i = 0; i < id_card.services.size; i++) {
 177                 svcs[i] = id_card.services[i];
 178             }
 179
 180             key_file.set_string_list(id_card.display_name, "Services", svcs);
 181
 182             if (rules.length > 0) {
 183                 key_file.set_string_list(id_card.display_name, "Rules-Patterns", rules_patterns);
 184                 key_file.set_string_list(id_card.display_name, "Rules-AlwaysConfirm", rules_always_conf);
 185             }
 186             key_file.set_string(id_card.display_name, "StorePassword", id_card.store_password ? "yes" : "no");
 187
 188             // Trust anchor
 189             key_file.set_string(id_card.display_name, "CA-Cert", id_card.trust_anchor.ca_cert);
 190             key_file.set_string(id_card.display_name, "Subject", id_card.trust_anchor.subject);
 191             key_file.set_string(id_card.display_name, "SubjectAlt", id_card.trust_anchor.subject_alt);
 192             key_file.set_string(id_card.display_name, "ServerCert", id_card.trust_anchor.server_cert);
 193             if (id_card.trust_anchor.datetime_added != "") {
 194                 key_file.set_string(id_card.display_name, "TA_DateTime_Added", id_card.trust_anchor.datetime_added);
 195             }
 196             logger.trace(@"store_id_cards: Stored '$(id_card.display_name)'");
 197         }
 198
 199         var text = key_file.to_data(null);
 200
 201         try {
 202             var path = get_data_dir();
 203             var filename = Path.build_filename(path, FILE_NAME);
 204             logger.trace("store_id_cards: attempting to store to " + filename);
 205             var file  = File.new_for_path(filename);
 206             var stream = file.replace(null, false, FileCreateFlags.PRIVATE);
 207             #if GIO_VAPI_USES_ARRAYS
 208             stream.write(text.data);
 209             #else
 210             var bits = text.data;
 211             stream.write(&bits[0], bits.length);
 212             #endif
 213                 }
 214         catch (Error e) {
 215             logger.error("store_id_cards: Error while saving keyfile: %s\n".printf(e.message));
 216             stdout.printf("Error:  %s\n", e.message);
 217         }
 218
 219         load_id_cards();
 220     }
 221
 222     public LocalFlatFileStore() {
 223         id_card_list = new LinkedList<IdCard>();
 224         load_id_cards();
 225     }
 226 }
 227