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
|
# Copyright 2014 Google Inc.
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software Foundation,
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
"""An implementation of a sshkey map for nsscache.
SshkeyMap: An implementation of NSS sshkey maps based on the Map
class.
SshkeyMapEntry: A sshkey map entry based on the MapEntry class.
"""
__author__ = "mimianddaniel@gmail.com"
from nss_cache.maps import maps
class SshkeyMap(maps.Map):
"""This class represents an NSS sshkey map.
Map data is stored as a list of MapEntry objects, see the abstract
class Map.
"""
def Add(self, entry):
"""Add a new object, verify it is a SshkeyMapEntry instance.
Args:
entry: A SshkeyMapEntry instance.
Returns:
True if added successfully, False otherwise.
Raises:
TypeError: The argument is of the wrong type.
"""
if not isinstance(entry, SshkeyMapEntry):
raise TypeError
return super(SshkeyMap, self).Add(entry)
class SshkeyMapEntry(maps.MapEntry):
"""This class represents NSS sshkey map entries."""
# Using slots saves us over 2x memory on large maps.
__slots__ = ("name", "sshkey")
_KEY = "name"
_ATTRS = ("name", "sshkey")
def __init__(self, data=None):
"""Construct a SshkeyMapEntry, setting reasonable defaults."""
self.name = None
self.sshkey = None
super(SshkeyMapEntry, self).__init__(data)
# Seed data with defaults if still empty
if self.sshkey is None:
self.sshkey = ""
|
