1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130
|
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#ifndef _NS_CERTTREE_H_
#define _NS_CERTTREE_H_
#include "nsCOMPtr.h"
#include "nsICertTree.h"
#include "nsITreeSelection.h"
#include "nsIMutableArray.h"
#include "nsNSSComponent.h"
#include "nsTArray.h"
#include "PLDHashTable.h"
/* Disable the "base class XXX should be explicitly initialized
in the copy constructor" warning. */
#if defined(__clang__)
# pragma clang diagnostic push
# pragma clang diagnostic ignored "-Wextra"
#elif defined(__GNUC__)
# pragma GCC diagnostic push
# pragma GCC diagnostic ignored "-Wextra"
#endif // __clang__ || __GNUC__
#include "mozilla/dom/XULTreeElement.h"
#if defined(__clang__)
# pragma clang diagnostic pop
#elif defined(__GNUC__)
# pragma GCC diagnostic pop
#endif // __clang__ || __GNUC__
typedef struct treeArrayElStr treeArrayEl;
struct CompareCacheHashEntry {
enum { max_criterions = 3 };
CompareCacheHashEntry();
void* key; // no ownership
bool mCritInit[max_criterions];
nsString mCrit[max_criterions];
};
struct CompareCacheHashEntryPtr : PLDHashEntryHdr {
CompareCacheHashEntryPtr();
~CompareCacheHashEntryPtr();
CompareCacheHashEntry* entry;
};
class nsCertTreeDispInfo : public nsICertTreeItem {
protected:
virtual ~nsCertTreeDispInfo();
public:
explicit nsCertTreeDispInfo(nsIX509Cert* aCert) : mCert(aCert) {}
NS_DECL_ISUPPORTS
NS_DECL_NSICERTTREEITEM
nsCOMPtr<nsIX509Cert> mCert;
};
class nsCertTree : public nsICertTree {
public:
NS_DECL_ISUPPORTS
NS_DECL_NSICERTTREE
NS_DECL_NSITREEVIEW
nsCertTree();
enum sortCriterion {
sort_IssuerOrg,
sort_Org,
sort_Token,
sort_CommonName,
sort_IssuedDateDescending,
sort_Email,
sort_None
};
protected:
virtual ~nsCertTree();
void ClearCompareHash();
void RemoveCacheEntry(void* key);
typedef int (*nsCertCompareFunc)(void*, nsIX509Cert* a, nsIX509Cert* b);
static CompareCacheHashEntry* getCacheEntry(void* cache, void* aCert);
static void CmpInitCriterion(nsIX509Cert* cert, CompareCacheHashEntry* entry,
sortCriterion crit, int32_t level);
static int32_t CmpByCrit(nsIX509Cert* a, CompareCacheHashEntry* ace,
nsIX509Cert* b, CompareCacheHashEntry* bce,
sortCriterion crit, int32_t level);
static int32_t CmpBy(void* cache, nsIX509Cert* a, nsIX509Cert* b,
sortCriterion c0, sortCriterion c1, sortCriterion c2);
static int32_t CmpCACert(void* cache, nsIX509Cert* a, nsIX509Cert* b);
static int32_t CmpUserCert(void* cache, nsIX509Cert* a, nsIX509Cert* b);
static int32_t CmpEmailCert(void* cache, nsIX509Cert* a, nsIX509Cert* b);
nsCertCompareFunc GetCompareFuncFromCertType(uint32_t aType);
int32_t CountOrganizations();
private:
static const uint32_t kInitialCacheLength = 64;
nsTArray<RefPtr<nsCertTreeDispInfo>> mDispInfo;
RefPtr<mozilla::dom::XULTreeElement> mTree;
nsCOMPtr<nsITreeSelection> mSelection;
treeArrayEl* mTreeArray;
int32_t mNumOrgs;
int32_t mNumRows;
PLDHashTable mCompareCache;
treeArrayEl* GetThreadDescAtIndex(int32_t _index);
already_AddRefed<nsIX509Cert> GetCertAtIndex(
int32_t _index, int32_t* outAbsoluteCertOffset = nullptr);
already_AddRefed<nsCertTreeDispInfo> GetDispInfoAtIndex(
int32_t index, int32_t* outAbsoluteCertOffset = nullptr);
void FreeCertArray();
nsresult UpdateUIContents();
nsresult GetCertsByTypeFromCertList(
const nsTArray<RefPtr<nsIX509Cert>>& aCertList, uint32_t aWantedType,
nsCertCompareFunc aCertCmpFn, void* aCertCmpFnArg);
nsCOMPtr<nsIMutableArray> mCellText;
};
#endif /* _NS_CERTTREE_H_ */
|