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
|
/* 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 CompareCacheEntry {
enum { max_criterions = 3 };
CompareCacheEntry();
bool mCritInit[max_criterions];
nsString mCrit[max_criterions];
};
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 final : 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
};
private:
virtual ~nsCertTree();
void ClearCompareHash();
void RemoveCacheEntry(nsIX509Cert* key);
using CompareCache =
nsTHashMap<nsIX509Cert*, std::unique_ptr<CompareCacheEntry>>;
using nsCertCompareFunc = int (*)(CompareCache*, nsIX509Cert* a,
nsIX509Cert* b);
static CompareCacheEntry* getCacheEntry(CompareCache* cache,
nsIX509Cert* aCert);
static void CmpInitCriterion(nsIX509Cert* cert, CompareCacheEntry* entry,
sortCriterion crit, int32_t level);
static int32_t CmpByCrit(nsIX509Cert* a, CompareCacheEntry* ace,
nsIX509Cert* b, CompareCacheEntry* bce,
sortCriterion crit, int32_t level);
static int32_t CmpBy(CompareCache* cache, nsIX509Cert* a, nsIX509Cert* b,
sortCriterion c0, sortCriterion c1, sortCriterion c2);
static int32_t CmpCACert(CompareCache* cache, nsIX509Cert* a, nsIX509Cert* b);
static int32_t CmpUserCert(CompareCache* cache, nsIX509Cert* a,
nsIX509Cert* b);
static int32_t CmpEmailCert(CompareCache* cache, nsIX509Cert* a,
nsIX509Cert* b);
nsCertCompareFunc GetCompareFuncFromCertType(uint32_t aType);
int32_t CountOrganizations();
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;
CompareCache 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, CompareCache* aCertCmpFnArg);
nsCOMPtr<nsIMutableArray> mCellText;
};
#endif /* NS_CERTTREE_H_ */
|
