ipc/keystore/KeyStore.h

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set sw=2 ts=8 et ft=cpp: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #ifndef mozilla_ipc_KeyStore_h
   8 #define mozilla_ipc_KeyStore_h 1
   9
  10 #include "mozilla/ipc/UnixSocket.h"
  11 #include <sys/socket.h>
  12 #include <sys/un.h>
  13
  14 #include "cert.h"
  15
  16 namespace mozilla {
  17 namespace ipc {
  18
  19 enum ResponseCode {
  20   SUCCESS           =  1,
  21   LOCKED            =  2,
  22   UNINITIALIZED     =  3,
  23   SYSTEM_ERROR      =  4,
  24   PROTOCOL_ERROR    =  5,
  25   PERMISSION_DENIED =  6,
  26   KEY_NOT_FOUND     =  7,
  27   VALUE_CORRUPTED   =  8,
  28   UNDEFINED_ACTION  =  9,
  29   WRONG_PASSWORD_0  = 10,
  30   WRONG_PASSWORD_1  = 11,
  31   WRONG_PASSWORD_2  = 12,
  32   WRONG_PASSWORD_3  = 13, // MAX_RETRY = 4
  33   NO_RESPONSE
  34 };
  35
  36 void FormatCaData(const uint8_t *aCaData, int aCaDataLength,
  37                   const char *aName, const uint8_t **aFormatData,
  38                   int *aFormatDataLength);
  39
  40 ResponseCode getCertificate(const char *aCertName, const uint8_t **aCertData,
  41                             int *aCertDataLength);
  42
  43 bool checkPermission(uid_t uid);
  44
  45 static const int MAX_PARAM = 2;
  46 static const int KEY_SIZE = ((NAME_MAX - 15) / 2);
  47 static const int VALUE_SIZE = 32768;
  48 static const int PASSWORD_SIZE = VALUE_SIZE;
  49
  50 static const char *CA_BEGIN = "-----BEGIN ",
  51                   *CA_END   = "-----END ",
  52                   *CA_TAILER = "-----\n";
  53 static const int CA_LINE_SIZE = 64;
  54
  55 struct ProtocolCommand {
  56   int8_t  command;
  57   int     paramNum;
  58 };
  59
  60 static const struct ProtocolCommand commands[] = {
  61   {'g', 1}, // Get CA, command "g CERT_NAME"
  62   { 0,  0}
  63 };
  64
  65 struct ProtocolParam{
  66   uint    length;
  67   int8_t  data[VALUE_SIZE];
  68 };
  69
  70 typedef enum {
  71   STATE_IDLE,
  72   STATE_READ_PARAM_LEN,
  73   STATE_READ_PARAM_DATA,
  74   STATE_PROCESSING
  75 } ProtocolHandlerState;
  76
  77 class KeyStoreConnector : public mozilla::ipc::UnixSocketConnector
  78 {
  79 public:
  80   KeyStoreConnector()
  81   {}
  82
  83   virtual ~KeyStoreConnector()
  84   {}
  85
  86   virtual int Create();
  87   virtual bool CreateAddr(bool aIsServer,
  88                           socklen_t& aAddrSize,
  89                           sockaddr_any& aAddr,
  90                           const char* aAddress);
  91   virtual bool SetUp(int aFd);
  92   virtual bool SetUpListenSocket(int aFd);
  93   virtual void GetSocketAddr(const sockaddr_any& aAddr,
  94                              nsAString& aAddrStr);
  95 };
  96
  97 class KeyStore : public mozilla::ipc::UnixSocketConsumer
  98 {
  99 public:
 100   KeyStore();
 101
 102   void Shutdown();
 103
 104 private:
 105   virtual ~KeyStore();
 106
 107   virtual void ReceiveSocketData(nsAutoPtr<UnixSocketRawData>& aMessage);
 108
 109   virtual void OnConnectSuccess();
 110   virtual void OnConnectError();
 111   virtual void OnDisconnect();
 112
 113   struct {
 114     ProtocolHandlerState          state;
 115     uint8_t                       command;
 116     struct ProtocolParam          param[MAX_PARAM];
 117     int                           paramCount;
 118     const struct ProtocolCommand  *commandPattern;
 119   } mHandlerInfo;
 120   void ResetHandlerInfo();
 121   void Listen();
 122
 123   bool CheckSize(UnixSocketRawData *aMessage, size_t aExpectSize);
 124   ResponseCode ReadCommand(UnixSocketRawData *aMessage);
 125   ResponseCode ReadLength(UnixSocketRawData *aMessage);
 126   ResponseCode ReadData(UnixSocketRawData *aMessage);
 127   void SendResponse(ResponseCode response);
 128   void SendData(const uint8_t *data, int length);
 129
 130   bool mShutdown;
 131 };
 132
 133 } // namespace ipc
 134 } // namespace mozilla
 135
 136 #endif // mozilla_ipc_KeyStore_h