summary refs log tree commit diff
path: root/crypto/src/cms/RecipientInformationStore.cs
blob: e317bc381668ec487f065fd571a36eab719c2a29 (plain) (blame)
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
using System;
using System.Collections.Generic;

using Org.BouncyCastle.Utilities;

namespace Org.BouncyCastle.Cms
{
	public class RecipientInformationStore
		: IEnumerable<RecipientInformation>
	{
		private readonly IList<RecipientInformation> m_all;
		private readonly IDictionary<RecipientID, IList<RecipientInformation>> m_table =
			new Dictionary<RecipientID, IList<RecipientInformation>>();

		public RecipientInformationStore(IEnumerable<RecipientInformation> recipientInfos)
		{
			foreach (RecipientInformation recipientInformation in recipientInfos)
			{
				RecipientID rid = recipientInformation.RecipientID;

				if (!m_table.TryGetValue(rid, out var list))
                {
					m_table[rid] = list = new List<RecipientInformation>(1);
				}

				list.Add(recipientInformation);
			}

            m_all = new List<RecipientInformation>(recipientInfos);
		}

		public RecipientInformation this[RecipientID selector]
		{
			get { return GetFirstRecipient(selector); }
		}

		/**
		* Return the first RecipientInformation object that matches the
		* passed in selector. Null if there are no matches.
		*
		* @param selector to identify a recipient
		* @return a single RecipientInformation object. Null if none matches.
		*/
		public RecipientInformation GetFirstRecipient(RecipientID selector)
		{
			if (!m_table.TryGetValue(selector, out var list))
				return null;

			return list[0];
		}

		/**
		* Return the number of recipients in the collection.
		*
		* @return number of recipients identified.
		*/
		public int Count
		{
			get { return m_all.Count; }
		}

		/**
		* Return all recipients in the collection
		*
		* @return a collection of recipients.
		*/
		public IList<RecipientInformation> GetRecipients()
		{
			return new List<RecipientInformation>(m_all);
		}

		/**
		* Return possible empty collection with recipients matching the passed in RecipientID
		*
		* @param selector a recipient id to select against.
		* @return a collection of RecipientInformation objects.
		*/
		public IList<RecipientInformation> GetRecipients(RecipientID selector)
		{
			if (!m_table.TryGetValue(selector, out var list))
				return new List<RecipientInformation>(0);

			return new List<RecipientInformation>(list);
		}

        System.Collections.IEnumerator System.Collections.IEnumerable.GetEnumerator()
        {
            return GetEnumerator();
        }

        public virtual IEnumerator<RecipientInformation> GetEnumerator()
        {
            return GetRecipients().GetEnumerator();
        }
    }
}