summaryrefslogtreecommitdiff
path: root/src/mscorlib/src/System/Runtime/InteropServices/WindowsRuntime/DictionaryToMapAdapter.cs
blob: bb54d49b60dfb49ce15ab8972e25cdc7ef06ae5d (plain)
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
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

//

using System;
using System.Security;
using System.Reflection;
using System.Collections;
using System.Collections.Generic;
using System.Collections.ObjectModel;
using System.Diagnostics;
using System.Diagnostics.Contracts;
using System.Runtime.InteropServices;
using System.Runtime.CompilerServices;

namespace System.Runtime.InteropServices.WindowsRuntime
{
    // This is a set of stub methods implementing the support for the IMap`2 interface on managed
    // objects that implement IDictionary`2. Used by the interop mashaling infrastructure.
    //
    // The methods on this class must be written VERY carefully to avoid introducing security holes.
    // That's because they are invoked with special "this"! The "this" object
    // for all of these methods are not DictionaryToMapAdapter objects. Rather, they are of type
    // IDictionary<K, V>. No actual DictionaryToMapAdapter object is ever instantiated. Thus, you will
    // see a lot of expressions that cast "this" to "IDictionary<K, V>". 
    internal sealed class DictionaryToMapAdapter
    {
        private DictionaryToMapAdapter()
        {
            Debug.Assert(false, "This class is never instantiated");
        }

        // V Lookup(K key)
        internal V Lookup<K, V>(K key)
        {
            IDictionary<K, V> _this = JitHelpers.UnsafeCast<IDictionary<K, V>>(this);
            V value;
            bool keyFound = _this.TryGetValue(key, out value);

            if (!keyFound)
            {
                Exception e = new KeyNotFoundException(SR.Arg_KeyNotFound);
                e.SetErrorCode(__HResults.E_BOUNDS);
                throw e;
            }

            return value;
        }

        // uint Size { get }
        internal uint Size<K, V>()
        {
            IDictionary<K, V> _this = JitHelpers.UnsafeCast<IDictionary<K, V>>(this);
            return (uint)_this.Count;
        }

        // bool HasKey(K key)
        internal bool HasKey<K, V>(K key)
        {
            IDictionary<K, V> _this = JitHelpers.UnsafeCast<IDictionary<K, V>>(this);
            return _this.ContainsKey(key);
        }

        // IMapView<K, V> GetView()
        internal IReadOnlyDictionary<K, V> GetView<K, V>()
        {
            IDictionary<K, V> _this = JitHelpers.UnsafeCast<IDictionary<K, V>>(this);
            Debug.Assert(_this != null);

            // Note: This dictionary is not really read-only - you could QI for a modifiable
            // dictionary.  We gain some perf by doing this.  We believe this is acceptable.
            IReadOnlyDictionary<K, V> roDictionary = _this as IReadOnlyDictionary<K, V>;
            if (roDictionary == null)
            {
                roDictionary = new ReadOnlyDictionary<K, V>(_this);
            }
            return roDictionary;
        }

        // bool Insert(K key, V value)
        internal bool Insert<K, V>(K key, V value)
        {
            IDictionary<K, V> _this = JitHelpers.UnsafeCast<IDictionary<K, V>>(this);
            bool replacing = _this.ContainsKey(key);
            _this[key] = value;
            return replacing;
        }

        // void Remove(K key)
        internal void Remove<K, V>(K key)
        {
            IDictionary<K, V> _this = JitHelpers.UnsafeCast<IDictionary<K, V>>(this);
            bool removed = _this.Remove(key);

            if (!removed)
            {
                Exception e = new KeyNotFoundException(SR.Arg_KeyNotFound);
                e.SetErrorCode(__HResults.E_BOUNDS);
                throw e;
            }
        }

        // void Clear()
        internal void Clear<K, V>()
        {
            IDictionary<K, V> _this = JitHelpers.UnsafeCast<IDictionary<K, V>>(this);
            _this.Clear();
        }
    }
}