Revision 26bade45
b/settings.d/00-apps.conf | ||
---|---|---|
17 | 17 |
'synnefo.logic', |
18 | 18 |
'synnefo.invitations', |
19 | 19 |
'synnefo.helpdesk', |
20 |
'synnefo.userdata', |
|
20 |
'synnefo.ui.userdata',
|
|
21 | 21 |
|
22 | 22 |
'south' |
23 | 23 |
) |
b/ui/urls.py | ||
---|---|---|
42 | 42 |
url(r'^feedback$', 'synnefo.ui.views.feedback_submit', name='feedback'), |
43 | 43 |
url(r'^static/(.*)$', 'django.views.static.serve', |
44 | 44 |
{'document_root': os.path.join(os.path.dirname(__file__), 'static')}), |
45 |
url(r'userdata/', include('synnefo.userdata.urls'))
|
|
45 |
url(r'ui/userdata/', include('synnefo.ui.userdata.urls'))
|
|
46 | 46 |
) |
47 | 47 |
|
48 | 48 |
if settings.DEBUG or settings.TEST: |
b/ui/userdata/migrations/0001_initial.py | ||
---|---|---|
1 |
# encoding: utf-8 |
|
2 |
import datetime |
|
3 |
from south.db import db |
|
4 |
from south.v2 import SchemaMigration |
|
5 |
from django.db import models |
|
6 |
|
|
7 |
class Migration(SchemaMigration): |
|
8 |
|
|
9 |
def forwards(self, orm): |
|
10 |
|
|
11 |
# Adding model 'PublicKeyPair' |
|
12 |
db.create_table('userdata_publickeypair', ( |
|
13 |
('id', self.gf('django.db.models.fields.AutoField')(primary_key=True)), |
|
14 |
('user', self.gf('django.db.models.fields.related.ForeignKey')(to=orm['db.SynnefoUser'])), |
|
15 |
('name', self.gf('django.db.models.fields.CharField')(max_length=255)), |
|
16 |
('content', self.gf('django.db.models.fields.TextField')()), |
|
17 |
)) |
|
18 |
db.send_create_signal('userdata', ['PublicKeyPair']) |
|
19 |
|
|
20 |
|
|
21 |
def backwards(self, orm): |
|
22 |
|
|
23 |
# Deleting model 'PublicKeyPair' |
|
24 |
db.delete_table('userdata_publickeypair') |
|
25 |
|
|
26 |
|
|
27 |
models = { |
|
28 |
'db.synnefouser': { |
|
29 |
'Meta': {'object_name': 'SynnefoUser'}, |
|
30 |
'auth_token': ('django.db.models.fields.CharField', [], {'max_length': '32', 'null': 'True'}), |
|
31 |
'auth_token_created': ('django.db.models.fields.DateTimeField', [], {'auto_now_add': 'True', 'null': 'True', 'blank': 'True'}), |
|
32 |
'auth_token_expires': ('django.db.models.fields.DateTimeField', [], {'auto_now_add': 'True', 'null': 'True', 'blank': 'True'}), |
|
33 |
'created': ('django.db.models.fields.DateTimeField', [], {'auto_now_add': 'True', 'blank': 'True'}), |
|
34 |
'credit': ('django.db.models.fields.IntegerField', [], {}), |
|
35 |
'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}), |
|
36 |
'max_invitations': ('django.db.models.fields.IntegerField', [], {'null': 'True'}), |
|
37 |
'name': ('django.db.models.fields.CharField', [], {'default': "''", 'max_length': '255'}), |
|
38 |
'realname': ('django.db.models.fields.CharField', [], {'default': "''", 'max_length': '255'}), |
|
39 |
'state': ('django.db.models.fields.CharField', [], {'default': "'ACTIVE'", 'max_length': '30'}), |
|
40 |
'tmp_auth_token': ('django.db.models.fields.CharField', [], {'max_length': '32', 'null': 'True'}), |
|
41 |
'tmp_auth_token_expires': ('django.db.models.fields.DateTimeField', [], {'auto_now_add': 'True', 'null': 'True', 'blank': 'True'}), |
|
42 |
'type': ('django.db.models.fields.CharField', [], {'max_length': '30'}), |
|
43 |
'uniq': ('django.db.models.fields.CharField', [], {'max_length': '255', 'null': 'True'}), |
|
44 |
'updated': ('django.db.models.fields.DateTimeField', [], {'auto_now': 'True', 'blank': 'True'}) |
|
45 |
}, |
|
46 |
'userdata.publickeypair': { |
|
47 |
'Meta': {'object_name': 'PublicKeyPair'}, |
|
48 |
'content': ('django.db.models.fields.TextField', [], {}), |
|
49 |
'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}), |
|
50 |
'name': ('django.db.models.fields.CharField', [], {'max_length': '255'}), |
|
51 |
'user': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['db.SynnefoUser']"}) |
|
52 |
} |
|
53 |
} |
|
54 |
|
|
55 |
complete_apps = ['userdata'] |
b/ui/userdata/models.py | ||
---|---|---|
1 |
from django.db import models |
|
2 |
from synnefo.db import models as synnefo_models |
|
3 |
|
|
4 |
User = synnefo_models.SynnefoUser |
|
5 |
|
|
6 |
class ProfileModel(models.Model): |
|
7 |
""" |
|
8 |
Abstract model, provides a basic interface for models that store |
|
9 |
user specific information |
|
10 |
""" |
|
11 |
|
|
12 |
user = models.ForeignKey(User) |
|
13 |
|
|
14 |
class Meta: |
|
15 |
abstract = True |
|
16 |
app_label = 'userdata' |
|
17 |
|
|
18 |
|
|
19 |
class PublicKeyPair(ProfileModel): |
|
20 |
""" |
|
21 |
Public key model |
|
22 |
""" |
|
23 |
name = models.CharField(max_length=255, null=False, blank=False) |
|
24 |
content = models.TextField() |
|
25 |
|
|
26 |
class Meta: |
|
27 |
app_label = 'userdata' |
b/ui/userdata/rest.py | ||
---|---|---|
1 |
from django import http |
|
2 |
from django.template import RequestContext, loader |
|
3 |
from django.utils import simplejson as json |
|
4 |
from django.core import serializers |
|
5 |
from django.core.urlresolvers import reverse |
|
6 |
|
|
7 |
# base view class |
|
8 |
# https://github.com/bfirsh/django-class-based-views/blob/master/class_based_views/base.py |
|
9 |
class View(object): |
|
10 |
""" |
|
11 |
Intentionally simple parent class for all views. Only implements |
|
12 |
dispatch-by-method and simple sanity checking. |
|
13 |
""" |
|
14 |
|
|
15 |
method_names = ['GET', 'POST', 'DELETE', 'HEAD', 'OPTIONS', 'TRACE'] |
|
16 |
|
|
17 |
def __init__(self, *args, **kwargs): |
|
18 |
""" |
|
19 |
Constructor. Called in the URLconf; can contain helpful extra |
|
20 |
keyword arguments, and other things. |
|
21 |
""" |
|
22 |
# Go through keyword arguments, and either save their values to our |
|
23 |
# instance, or raise an error. |
|
24 |
for key, value in kwargs.items(): |
|
25 |
if key in self.method_names: |
|
26 |
raise TypeError(u"You tried to pass in the %s method name as a " |
|
27 |
u"keyword argument to %s(). Don't do that." |
|
28 |
% (key, self.__class__.__name__)) |
|
29 |
if hasattr(self, key): |
|
30 |
setattr(self, key, value) |
|
31 |
else: |
|
32 |
raise TypeError(u"%s() received an invalid keyword %r" % ( |
|
33 |
self.__class__.__name__, |
|
34 |
key, |
|
35 |
)) |
|
36 |
|
|
37 |
@classmethod |
|
38 |
def as_view(cls, *initargs, **initkwargs): |
|
39 |
""" |
|
40 |
Main entry point for a request-response process. |
|
41 |
""" |
|
42 |
def view(request, *args, **kwargs): |
|
43 |
self = cls(*initargs, **initkwargs) |
|
44 |
return self.dispatch(request, *args, **kwargs) |
|
45 |
return view |
|
46 |
|
|
47 |
def dispatch(self, request, *args, **kwargs): |
|
48 |
# Try to dispatch to the right method for that; if it doesn't exist, |
|
49 |
# raise a big error. |
|
50 |
if hasattr(self, request.method.upper()): |
|
51 |
self.request = request |
|
52 |
self.args = args |
|
53 |
self.kwargs = kwargs |
|
54 |
data = request.raw_post_data |
|
55 |
|
|
56 |
if request.method.upper() in ['POST', 'PUT']: |
|
57 |
# Expect json data |
|
58 |
if request.META.get('CONTENT_TYPE').startswith('application/json'): |
|
59 |
try: |
|
60 |
data = json.loads(data) |
|
61 |
except ValueError: |
|
62 |
raise http.HttpResponseServerError('Invalid JSON data.') |
|
63 |
else: |
|
64 |
raise http.HttpResponseServerError('Unsupported Content-Type.') |
|
65 |
|
|
66 |
return getattr(self, request.method.upper())(request, data, *args, **kwargs) |
|
67 |
else: |
|
68 |
allowed_methods = [m for m in self.method_names if hasattr(self, m)] |
|
69 |
return http.HttpResponseNotAllowed(allowed_methods) |
|
70 |
|
|
71 |
class JSONRestView(View): |
|
72 |
""" |
|
73 |
Class that provides helpers to produce a json response |
|
74 |
""" |
|
75 |
|
|
76 |
url_name = None |
|
77 |
def __init__(self, url_name, *args, **kwargs): |
|
78 |
self.url_name = url_name |
|
79 |
return super(JSONRestView, self).__init__(*args, **kwargs) |
|
80 |
|
|
81 |
def update_instance(self, i, data, exclude_fields=[]): |
|
82 |
update_keys = data.keys() |
|
83 |
for field in i._meta.get_all_field_names(): |
|
84 |
if field in update_keys and (field not in exclude_fields): |
|
85 |
i.__setattr__(field, data[field]) |
|
86 |
|
|
87 |
return i |
|
88 |
|
|
89 |
def instance_to_dict(self, i, exclude_fields=[]): |
|
90 |
""" |
|
91 |
Convert model instance to python dict |
|
92 |
""" |
|
93 |
d = {} |
|
94 |
d['uri'] = reverse(self.url_name, kwargs={'id': i.pk}) |
|
95 |
|
|
96 |
for field in i._meta.get_all_field_names(): |
|
97 |
if field in exclude_fields: |
|
98 |
continue |
|
99 |
|
|
100 |
d[field] = i.__getattribute__(field) |
|
101 |
return d |
|
102 |
|
|
103 |
def qs_to_dict_iter(self, qs, exclude_fields=[]): |
|
104 |
""" |
|
105 |
Convert queryset to an iterator of model instances dicts |
|
106 |
""" |
|
107 |
for i in qs: |
|
108 |
yield self.instance_to_dict(i, exclude_fields) |
|
109 |
|
|
110 |
def json_response(self, data): |
|
111 |
return http.HttpResponse(json.dumps(data), mimetype="application/json") |
|
112 |
|
|
113 |
class ResourceView(JSONRestView): |
|
114 |
method_names = ['GET', 'POST', 'PUT', 'DELETE'] |
|
115 |
|
|
116 |
model = None |
|
117 |
exclude_fields = [] |
|
118 |
|
|
119 |
def queryset(self): |
|
120 |
return self.model.objects.all() |
|
121 |
|
|
122 |
def instance(self): |
|
123 |
""" |
|
124 |
Retrieve selected instance based on url parameter |
|
125 |
|
|
126 |
id parameter should be set in urlpatterns expression |
|
127 |
""" |
|
128 |
try: |
|
129 |
return self.queryset().get(pk=self.kwargs.get("id")) |
|
130 |
except self.model.DoesNotExist: |
|
131 |
raise http.Http404 |
|
132 |
|
|
133 |
def GET(self, request, data, *args, **kwargs): |
|
134 |
return self.json_response(self.instance_to_dict(self.instance(), |
|
135 |
self.exclude_fields)) |
|
136 |
|
|
137 |
def PUT(self, request, data, *args, **kwargs): |
|
138 |
instance = self.instance() |
|
139 |
self.update_instance(instance, data, self.exclude_fields) |
|
140 |
instance.save() |
|
141 |
return self.GET(request, data, *args, **kwargs) |
|
142 |
|
|
143 |
def DELETE(self, request, data, *args, **kwargs): |
|
144 |
self.instance().delete() |
|
145 |
return self.json_response("") |
|
146 |
|
|
147 |
|
|
148 |
class CollectionView(JSONRestView): |
|
149 |
method_names = ['GET', 'POST'] |
|
150 |
|
|
151 |
model = None |
|
152 |
exclude_fields = [] |
|
153 |
|
|
154 |
def queryset(self): |
|
155 |
return self.model.objects.all() |
|
156 |
|
|
157 |
def GET(self, request, data, *args, **kwargs): |
|
158 |
return self.json_response(list(self.qs_to_dict_iter(self.queryset(), |
|
159 |
self.exclude_fields))) |
|
160 |
|
|
161 |
def POST(self, request, data, *args, **kwargs): |
|
162 |
instance = self.model() |
|
163 |
self.update_instance(instance, data, self.exclude_fields) |
|
164 |
instance.save() |
|
165 |
return self.json_response(self.instance_to_dict(instance, |
|
166 |
self.exclude_fields)) |
|
167 |
|
|
168 |
class UserResourceView(ResourceView): |
|
169 |
""" |
|
170 |
Filter resource queryset for request user entries |
|
171 |
""" |
|
172 |
def queryset(self): |
|
173 |
return super(UserResourceView, |
|
174 |
self).queryset().filter(user=self.request.user) |
|
175 |
|
|
176 |
class UserCollectionView(CollectionView): |
|
177 |
""" |
|
178 |
Filter collection queryset for request user entries |
|
179 |
""" |
|
180 |
def queryset(self): |
|
181 |
return super(UserCollectionView, self).queryset().filter(user=self.request.user) |
|
182 |
|
|
183 |
def POST(self, request, data, *args, **kwargs): |
|
184 |
instance = self.model() |
|
185 |
self.update_instance(instance, data, self.exclude_fields) |
|
186 |
instance.user = request.user |
|
187 |
instance.save() |
|
188 |
return self.json_response(self.instance_to_dict(instance, |
|
189 |
self.exclude_fields)) |
|
190 |
|
b/ui/userdata/templates/userdata/public_keys_view.html | ||
---|---|---|
1 |
{% load i18n %} |
|
2 |
<div class="public-keys-view clearfix"> |
|
3 |
<div class="loading-models">{% trans "Loading..." %}</div> |
|
4 |
<div class="models-view"> |
|
5 |
<div class="previous-view-link"><div class="change-view-action">Back to |
|
6 |
machine create wizard</div></div> |
|
7 |
|
|
8 |
<div class="list-wrapper model-list"> |
|
9 |
<h3 class="list-title hidden">{% trans "SSH public keys list" %}</h3> |
|
10 |
<div class="top-actions clearfix"> |
|
11 |
<div class="collection-action add add-new">{% trans "create/import new" %}</div> |
|
12 |
<div class="collection-action generate add-generate">{% trans "generate new" %}</div> |
|
13 |
</div> |
|
14 |
<div class="model-description"> |
|
15 |
<p>You can use SSH keys to establish a secure connection |
|
16 |
between your computer and the virtual machines. </p> |
|
17 |
</div> |
|
18 |
<div class="list-messages"> |
|
19 |
</div> |
|
20 |
<div class="private-cont"> |
|
21 |
<div class="private-download clearfix"> |
|
22 |
<div class="close-private">{% trans "close" %}</div> |
|
23 |
<div class="private-msg download"> |
|
24 |
</div> |
|
25 |
<div class="private-msg copy"> |
|
26 |
{% trans "Create a file with the following contents" %} |
|
27 |
</div> |
|
28 |
<form target="_blank" id="private-key-form" method="post"> |
|
29 |
<input type="hidden" name="data" /> |
|
30 |
<input type="hidden" name="name" /> |
|
31 |
|
|
32 |
<span class="form-text">{% trans "Your new public key has been added" %}</span> |
|
33 |
<input type="submit" class="down-button form-text" |
|
34 |
id="download-private-key" value="click here" /> |
|
35 |
<span class="form-text"> |
|
36 |
{% trans " to download private key." %} |
|
37 |
</span> |
|
38 |
</form> |
|
39 |
<div class="key-contents clearfix"><textarea></textarea></div> |
|
40 |
</div> |
|
41 |
</div> |
|
42 |
<div class="hidden public-key-item clearfix" id="model-item-tpl"> |
|
43 |
<div class="param key-type"></div> |
|
44 |
<div class="param name"></div> |
|
45 |
<div class="param publicid hidden"> |
|
46 |
<div class="param-content"><textarea></textarea></div> |
|
47 |
</div> |
|
48 |
</div> |
|
49 |
<ul class="keys-list"> |
|
50 |
<li class="header"> |
|
51 |
<div class="title">{% trans "Name" %}</div> |
|
52 |
<div class="title">{% trans "Public key ID" %}</div> |
|
53 |
</li> |
|
54 |
</ul> |
|
55 |
<ul class="items-list"> |
|
56 |
<li class="items-empty-msg hidden msg"> |
|
57 |
<div class="title">{% trans "No public keys exist" %} <span class="quick-add">add one</span> now</div> |
|
58 |
</li> |
|
59 |
</ul> |
|
60 |
</div> |
|
61 |
<div class="form-wrapper model-form clearfix"> |
|
62 |
<h3 class="new-title hidden">{% trans "Create new SSH public key" %}</h3> |
|
63 |
<h3 class="edit-title hidden">{% trans "Edit SSH public key" %}</h3> |
|
64 |
<form class="model-form"> |
|
65 |
<div class="form-messages"> |
|
66 |
</div> |
|
67 |
<div class="model-form"> |
|
68 |
<div class="form-field"> |
|
69 |
<label for="">{% trans "Key name" %}</label> |
|
70 |
<div class="errors"></div> |
|
71 |
<input type="text" class="input-name inline"/> |
|
72 |
</div> |
|
73 |
<div class="form-field clearfix"> |
|
74 |
<label for="">{% trans "Key content" %}</label> |
|
75 |
<div class="errors"></div> |
|
76 |
<textarea class="input-content"></textarea> |
|
77 |
</div> |
|
78 |
<div class="form-field inline clearfix fromfile-field"> |
|
79 |
<label for="">{% trans "Import from file" %}</label> |
|
80 |
<div class="fromfile"> |
|
81 |
<input type="file" class="content-input-file" |
|
82 |
title="Import from file" /> |
|
83 |
</div> |
|
84 |
</div> |
|
85 |
<div class="form-field"> |
|
86 |
</div> |
|
87 |
</div> |
|
88 |
</form> |
|
89 |
<div class="form-actions clearfix"> |
|
90 |
<div class="form-action submit">{% trans "Submit" %}</div> |
|
91 |
<div class="form-action cancel">{% trans "Cancel" %}</div> |
|
92 |
</div> |
|
93 |
</div> |
|
94 |
</div> |
|
95 |
</div> |
|
96 |
</div> |
b/ui/userdata/tests.py | ||
---|---|---|
1 |
""" |
|
2 |
This file demonstrates two different styles of tests (one doctest and one |
|
3 |
unittest). These will both pass when you run "manage.py test". |
|
4 |
|
|
5 |
Replace these with more appropriate tests for your application. |
|
6 |
""" |
|
7 |
|
|
8 |
from django.test import TestCase |
|
9 |
from django.conf import settings |
|
10 |
from django.test.client import Client |
|
11 |
from django.core.urlresolvers import clear_url_caches |
|
12 |
from django.utils import simplejson as json |
|
13 |
|
|
14 |
from synnefo.ui.userdata.models import User |
|
15 |
from synnefo.ui.userdata.models import * |
|
16 |
|
|
17 |
class AaiClient(Client): |
|
18 |
|
|
19 |
def request(self, **request): |
|
20 |
request['HTTP_X_AUTH_TOKEN'] = '46e427d657b20defe352804f0eb6f8a2' |
|
21 |
return super(AaiClient, self).request(**request) |
|
22 |
|
|
23 |
class TestRestViews(TestCase): |
|
24 |
|
|
25 |
fixtures = ['users'] |
|
26 |
|
|
27 |
def setUp(self): |
|
28 |
settings.ROOT_URLCONF = 'synnefo.ui.userdata.urls' |
|
29 |
clear_url_caches() |
|
30 |
self.client = AaiClient() |
|
31 |
self.user = User.objects.get(pk=1) |
|
32 |
|
|
33 |
def test_keys_collection_get(self): |
|
34 |
resp = self.client.get("/keys") |
|
35 |
self.assertEqual(resp.content, "[]") |
|
36 |
|
|
37 |
PublicKeyPair.objects.create(user=self.user, name="key pair 1", |
|
38 |
content="content1") |
|
39 |
|
|
40 |
resp = self.client.get("/keys") |
|
41 |
self.assertEqual(resp.content, """[{"content": "content1", "uri": "/keys/1", "name": "key pair 1", "id": 1}]""") |
|
42 |
|
|
43 |
PublicKeyPair.objects.create(user=self.user, name="key pair 2", |
|
44 |
content="content2") |
|
45 |
|
|
46 |
resp = self.client.get("/keys") |
|
47 |
self.assertEqual(resp.content, """[{"content": "content1", "uri": "/keys/1", "name": "key pair 1", "id": 1}, {"content": "content2", "uri": "/keys/2", "name": "key pair 2", "id": 2}]""") |
|
48 |
|
|
49 |
def test_keys_resourse_get(self): |
|
50 |
resp = self.client.get("/keys/1") |
|
51 |
self.assertEqual(resp.status_code, 404) |
|
52 |
|
|
53 |
# create a public key |
|
54 |
PublicKeyPair.objects.create(user=self.user, name="key pair 1", |
|
55 |
content="content1") |
|
56 |
resp = self.client.get("/keys/1") |
|
57 |
self.assertEqual(resp.content, """{"content": "content1", "uri": "/keys/1", "name": "key pair 1", "id": 1}""") |
|
58 |
|
|
59 |
# update |
|
60 |
resp = self.client.put("/keys/1", json.dumps({'name':'key pair 1 new name'}), |
|
61 |
content_type='application/json') |
|
62 |
pk = PublicKeyPair.objects.get(pk=1) |
|
63 |
self.assertEqual(pk.name, "key pair 1 new name") |
|
64 |
|
|
65 |
# delete |
|
66 |
resp = self.client.delete("/keys/1") |
|
67 |
self.assertEqual(PublicKeyPair.objects.count(), 0) |
|
68 |
|
|
69 |
resp = self.client.get("/keys/1") |
|
70 |
self.assertEqual(resp.status_code, 404) |
|
71 |
|
|
72 |
resp = self.client.get("/keys") |
|
73 |
self.assertEqual(resp.content, "[]") |
|
74 |
|
|
75 |
# test rest create |
|
76 |
resp = self.client.post("/keys", json.dumps({'name':'key pair 2', |
|
77 |
'content':"""key 2 content"""}), |
|
78 |
content_type='application/json') |
|
79 |
self.assertEqual(PublicKeyPair.objects.count(), 1) |
|
80 |
pk = PublicKeyPair.objects.get(pk=1) |
|
81 |
self.assertEqual(pk.name, "key pair 2") |
|
82 |
self.assertEqual(pk.content, "key 2 content") |
|
83 |
|
|
84 |
def test_generate_views(self): |
|
85 |
import base64 |
|
86 |
|
|
87 |
# just test that |
|
88 |
resp = self.client.get("/keys/generate") |
|
89 |
self.assertNotEqual(resp, "") |
|
90 |
|
|
91 |
data = json.loads(resp.content) |
|
92 |
self.assertEqual(data.has_key('private'), True) |
|
93 |
self.assertEqual(data.has_key('private'), True) |
|
94 |
|
|
95 |
# public key is base64 encoded |
|
96 |
base64.b64decode(data['public'].replace("ssh-rsa ","")) |
|
97 |
|
|
98 |
# remove header/footer |
|
99 |
private = "".join(data['private'].split("\n")[1:-1]) |
|
100 |
|
|
101 |
# private key is base64 encoded |
|
102 |
base64.b64decode(private) |
|
103 |
|
b/ui/userdata/urls.py | ||
---|---|---|
1 |
from django.conf.urls.defaults import * |
|
2 |
from synnefo.ui.userdata import views |
|
3 |
|
|
4 |
urlpatterns = patterns('', |
|
5 |
url(r'^keys$', views.PublicKeyPairCollectionView.as_view('keys_resource'), |
|
6 |
name='keys_collection'), |
|
7 |
url(r'^keys/(?P<id>\d+)', |
|
8 |
views.PublicKeyPairResourceView.as_view('keys_resource'), |
|
9 |
name="keys_resource"), |
|
10 |
url(r'keys/generate', views.generate_key_pair, name="generate_public_key"), |
|
11 |
url(r'keys/download', views.download_private_key, name="download_public_key") |
|
12 |
) |
b/ui/userdata/views.py | ||
---|---|---|
1 |
from django import http |
|
2 |
from django.template import RequestContext, loader |
|
3 |
from django.utils import simplejson as json |
|
4 |
from django.conf import settings |
|
5 |
|
|
6 |
from synnefo.ui.userdata import rest |
|
7 |
from synnefo.ui.userdata.models import PublicKeyPair |
|
8 |
|
|
9 |
import M2Crypto as M2C |
|
10 |
import base64 |
|
11 |
|
|
12 |
class PublicKeyPairResourceView(rest.UserResourceView): |
|
13 |
model = PublicKeyPair |
|
14 |
exclude_fields = ["user"] |
|
15 |
|
|
16 |
class PublicKeyPairCollectionView(rest.UserCollectionView): |
|
17 |
model = PublicKeyPair |
|
18 |
exclude_fields = ["user"] |
|
19 |
|
|
20 |
SSH_KEY_LENGTH = getattr(settings, 'USERDATA_SSH_KEY_LENGTH', 2048) |
|
21 |
SSH_KEY_EXPONENT = getattr(settings, 'USERDATA_SSH_KEY_EXPONENT', 65537) |
|
22 |
def generate_key_pair(request): |
|
23 |
""" |
|
24 |
Response to generate private/public RSA key pair |
|
25 |
""" |
|
26 |
# generate RSA key |
|
27 |
key = M2C.RSA.gen_key(SSH_KEY_LENGTH, SSH_KEY_EXPONENT, lambda x: ""); |
|
28 |
|
|
29 |
# get PEM string |
|
30 |
pem_buffer = M2C.BIO.MemoryBuffer() |
|
31 |
M2C.m2.rsa_write_key_no_cipher(key.rsa, pem_buffer._ptr(), lambda : "") |
|
32 |
pem = pem_buffer.getvalue() |
|
33 |
|
|
34 |
# generate public content |
|
35 |
public = "ssh-rsa %s" % base64.b64encode('\x00\x00\x00\x07ssh-rsa%s%s' % (key.pub()[0], key.pub()[1])) |
|
36 |
|
|
37 |
data = {'private': pem, 'public': public} |
|
38 |
return http.HttpResponse(json.dumps(data), mimetype="application/json") |
|
39 |
|
|
40 |
def download_private_key(request): |
|
41 |
""" |
|
42 |
Return key contents |
|
43 |
""" |
|
44 |
data = request.POST.get("data") |
|
45 |
name = request.POST.get("name", "key") |
|
46 |
|
|
47 |
response = http.HttpResponse(mimetype='application/x-pem-key') |
|
48 |
response['Content-Disposition'] = 'attachment; filename=%s.pem' % name |
|
49 |
response.write(data) |
|
50 |
return response |
/dev/null | ||
---|---|---|
1 |
# encoding: utf-8 |
|
2 |
import datetime |
|
3 |
from south.db import db |
|
4 |
from south.v2 import SchemaMigration |
|
5 |
from django.db import models |
|
6 |
|
|
7 |
class Migration(SchemaMigration): |
|
8 |
|
|
9 |
def forwards(self, orm): |
|
10 |
|
|
11 |
# Adding model 'PublicKeyPair' |
|
12 |
db.create_table('userdata_publickeypair', ( |
|
13 |
('id', self.gf('django.db.models.fields.AutoField')(primary_key=True)), |
|
14 |
('user', self.gf('django.db.models.fields.related.ForeignKey')(to=orm['db.SynnefoUser'])), |
|
15 |
('name', self.gf('django.db.models.fields.CharField')(max_length=255)), |
|
16 |
('content', self.gf('django.db.models.fields.TextField')()), |
|
17 |
)) |
|
18 |
db.send_create_signal('userdata', ['PublicKeyPair']) |
|
19 |
|
|
20 |
|
|
21 |
def backwards(self, orm): |
|
22 |
|
|
23 |
# Deleting model 'PublicKeyPair' |
|
24 |
db.delete_table('userdata_publickeypair') |
|
25 |
|
|
26 |
|
|
27 |
models = { |
|
28 |
'db.synnefouser': { |
|
29 |
'Meta': {'object_name': 'SynnefoUser'}, |
|
30 |
'auth_token': ('django.db.models.fields.CharField', [], {'max_length': '32', 'null': 'True'}), |
|
31 |
'auth_token_created': ('django.db.models.fields.DateTimeField', [], {'auto_now_add': 'True', 'null': 'True', 'blank': 'True'}), |
|
32 |
'auth_token_expires': ('django.db.models.fields.DateTimeField', [], {'auto_now_add': 'True', 'null': 'True', 'blank': 'True'}), |
|
33 |
'created': ('django.db.models.fields.DateTimeField', [], {'auto_now_add': 'True', 'blank': 'True'}), |
|
34 |
'credit': ('django.db.models.fields.IntegerField', [], {}), |
|
35 |
'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}), |
|
36 |
'max_invitations': ('django.db.models.fields.IntegerField', [], {'null': 'True'}), |
|
37 |
'name': ('django.db.models.fields.CharField', [], {'default': "''", 'max_length': '255'}), |
|
38 |
'realname': ('django.db.models.fields.CharField', [], {'default': "''", 'max_length': '255'}), |
|
39 |
'state': ('django.db.models.fields.CharField', [], {'default': "'ACTIVE'", 'max_length': '30'}), |
|
40 |
'tmp_auth_token': ('django.db.models.fields.CharField', [], {'max_length': '32', 'null': 'True'}), |
|
41 |
'tmp_auth_token_expires': ('django.db.models.fields.DateTimeField', [], {'auto_now_add': 'True', 'null': 'True', 'blank': 'True'}), |
|
42 |
'type': ('django.db.models.fields.CharField', [], {'max_length': '30'}), |
|
43 |
'uniq': ('django.db.models.fields.CharField', [], {'max_length': '255', 'null': 'True'}), |
|
44 |
'updated': ('django.db.models.fields.DateTimeField', [], {'auto_now': 'True', 'blank': 'True'}) |
|
45 |
}, |
|
46 |
'userdata.publickeypair': { |
|
47 |
'Meta': {'object_name': 'PublicKeyPair'}, |
|
48 |
'content': ('django.db.models.fields.TextField', [], {}), |
|
49 |
'id': ('django.db.models.fields.AutoField', [], {'primary_key': 'True'}), |
|
50 |
'name': ('django.db.models.fields.CharField', [], {'max_length': '255'}), |
|
51 |
'user': ('django.db.models.fields.related.ForeignKey', [], {'to': "orm['db.SynnefoUser']"}) |
|
52 |
} |
|
53 |
} |
|
54 |
|
|
55 |
complete_apps = ['userdata'] |
/dev/null | ||
---|---|---|
1 |
from django.db import models |
|
2 |
from synnefo.db import models as synnefo_models |
|
3 |
|
|
4 |
User = synnefo_models.SynnefoUser |
|
5 |
|
|
6 |
class ProfileModel(models.Model): |
|
7 |
""" |
|
8 |
Abstract model, provides a basic interface for models that store |
|
9 |
user specific information |
|
10 |
""" |
|
11 |
|
|
12 |
user = models.ForeignKey(User) |
|
13 |
|
|
14 |
class Meta: |
|
15 |
abstract = True |
|
16 |
|
|
17 |
|
|
18 |
class PublicKeyPair(ProfileModel): |
|
19 |
""" |
|
20 |
Public key model |
|
21 |
""" |
|
22 |
name = models.CharField(max_length=255, null=False, blank=False) |
|
23 |
content = models.TextField() |
|
24 |
|
/dev/null | ||
---|---|---|
1 |
from django import http |
|
2 |
from django.template import RequestContext, loader |
|
3 |
from django.utils import simplejson as json |
|
4 |
from django.core import serializers |
|
5 |
from django.core.urlresolvers import reverse |
|
6 |
|
|
7 |
# base view class |
|
8 |
# https://github.com/bfirsh/django-class-based-views/blob/master/class_based_views/base.py |
|
9 |
class View(object): |
|
10 |
""" |
|
11 |
Intentionally simple parent class for all views. Only implements |
|
12 |
dispatch-by-method and simple sanity checking. |
|
13 |
""" |
|
14 |
|
|
15 |
method_names = ['GET', 'POST', 'DELETE', 'HEAD', 'OPTIONS', 'TRACE'] |
|
16 |
|
|
17 |
def __init__(self, *args, **kwargs): |
|
18 |
""" |
|
19 |
Constructor. Called in the URLconf; can contain helpful extra |
|
20 |
keyword arguments, and other things. |
|
21 |
""" |
|
22 |
# Go through keyword arguments, and either save their values to our |
|
23 |
# instance, or raise an error. |
|
24 |
for key, value in kwargs.items(): |
|
25 |
if key in self.method_names: |
|
26 |
raise TypeError(u"You tried to pass in the %s method name as a " |
|
27 |
u"keyword argument to %s(). Don't do that." |
|
28 |
% (key, self.__class__.__name__)) |
|
29 |
if hasattr(self, key): |
|
30 |
setattr(self, key, value) |
|
31 |
else: |
|
32 |
raise TypeError(u"%s() received an invalid keyword %r" % ( |
|
33 |
self.__class__.__name__, |
|
34 |
key, |
|
35 |
)) |
|
36 |
|
|
37 |
@classmethod |
|
38 |
def as_view(cls, *initargs, **initkwargs): |
|
39 |
""" |
|
40 |
Main entry point for a request-response process. |
|
41 |
""" |
|
42 |
def view(request, *args, **kwargs): |
|
43 |
self = cls(*initargs, **initkwargs) |
|
44 |
return self.dispatch(request, *args, **kwargs) |
|
45 |
return view |
|
46 |
|
|
47 |
def dispatch(self, request, *args, **kwargs): |
|
48 |
# Try to dispatch to the right method for that; if it doesn't exist, |
|
49 |
# raise a big error. |
|
50 |
if hasattr(self, request.method.upper()): |
|
51 |
self.request = request |
|
52 |
self.args = args |
|
53 |
self.kwargs = kwargs |
|
54 |
data = request.raw_post_data |
|
55 |
|
|
56 |
if request.method.upper() in ['POST', 'PUT']: |
|
57 |
# Expect json data |
|
58 |
if request.META.get('CONTENT_TYPE').startswith('application/json'): |
|
59 |
try: |
|
60 |
data = json.loads(data) |
|
61 |
except ValueError: |
|
62 |
raise http.HttpResponseServerError('Invalid JSON data.') |
|
63 |
else: |
|
64 |
raise http.HttpResponseServerError('Unsupported Content-Type.') |
|
65 |
|
|
66 |
return getattr(self, request.method.upper())(request, data, *args, **kwargs) |
|
67 |
else: |
|
68 |
allowed_methods = [m for m in self.method_names if hasattr(self, m)] |
|
69 |
return http.HttpResponseNotAllowed(allowed_methods) |
|
70 |
|
|
71 |
class JSONRestView(View): |
|
72 |
""" |
|
73 |
Class that provides helpers to produce a json response |
|
74 |
""" |
|
75 |
|
|
76 |
url_name = None |
|
77 |
def __init__(self, url_name, *args, **kwargs): |
|
78 |
self.url_name = url_name |
|
79 |
return super(JSONRestView, self).__init__(*args, **kwargs) |
|
80 |
|
|
81 |
def update_instance(self, i, data, exclude_fields=[]): |
|
82 |
update_keys = data.keys() |
|
83 |
for field in i._meta.get_all_field_names(): |
|
84 |
if field in update_keys and (field not in exclude_fields): |
|
85 |
i.__setattr__(field, data[field]) |
|
86 |
|
|
87 |
return i |
|
88 |
|
|
89 |
def instance_to_dict(self, i, exclude_fields=[]): |
|
90 |
""" |
|
91 |
Convert model instance to python dict |
|
92 |
""" |
|
93 |
d = {} |
|
94 |
d['uri'] = reverse(self.url_name, kwargs={'id': i.pk}) |
|
95 |
|
|
96 |
for field in i._meta.get_all_field_names(): |
|
97 |
if field in exclude_fields: |
|
98 |
continue |
|
99 |
|
|
100 |
d[field] = i.__getattribute__(field) |
|
101 |
return d |
|
102 |
|
|
103 |
def qs_to_dict_iter(self, qs, exclude_fields=[]): |
|
104 |
""" |
|
105 |
Convert queryset to an iterator of model instances dicts |
|
106 |
""" |
|
107 |
for i in qs: |
|
108 |
yield self.instance_to_dict(i, exclude_fields) |
|
109 |
|
|
110 |
def json_response(self, data): |
|
111 |
return http.HttpResponse(json.dumps(data), mimetype="application/json") |
|
112 |
|
|
113 |
class ResourceView(JSONRestView): |
|
114 |
method_names = ['GET', 'POST', 'PUT', 'DELETE'] |
|
115 |
|
|
116 |
model = None |
|
117 |
exclude_fields = [] |
|
118 |
|
|
119 |
def queryset(self): |
|
120 |
return self.model.objects.all() |
|
121 |
|
|
122 |
def instance(self): |
|
123 |
""" |
|
124 |
Retrieve selected instance based on url parameter |
|
125 |
|
|
126 |
id parameter should be set in urlpatterns expression |
|
127 |
""" |
|
128 |
try: |
|
129 |
return self.queryset().get(pk=self.kwargs.get("id")) |
|
130 |
except self.model.DoesNotExist: |
|
131 |
raise http.Http404 |
|
132 |
|
|
133 |
def GET(self, request, data, *args, **kwargs): |
|
134 |
return self.json_response(self.instance_to_dict(self.instance(), |
|
135 |
self.exclude_fields)) |
|
136 |
|
|
137 |
def PUT(self, request, data, *args, **kwargs): |
|
138 |
instance = self.instance() |
|
139 |
self.update_instance(instance, data, self.exclude_fields) |
|
140 |
instance.save() |
|
141 |
return self.GET(request, data, *args, **kwargs) |
|
142 |
|
|
143 |
def DELETE(self, request, data, *args, **kwargs): |
|
144 |
self.instance().delete() |
|
145 |
return self.json_response("") |
|
146 |
|
|
147 |
|
|
148 |
class CollectionView(JSONRestView): |
|
149 |
method_names = ['GET', 'POST'] |
|
150 |
|
|
151 |
model = None |
|
152 |
exclude_fields = [] |
|
153 |
|
|
154 |
def queryset(self): |
|
155 |
return self.model.objects.all() |
|
156 |
|
|
157 |
def GET(self, request, data, *args, **kwargs): |
|
158 |
return self.json_response(list(self.qs_to_dict_iter(self.queryset(), |
|
159 |
self.exclude_fields))) |
|
160 |
|
|
161 |
def POST(self, request, data, *args, **kwargs): |
|
162 |
instance = self.model() |
|
163 |
self.update_instance(instance, data, self.exclude_fields) |
|
164 |
instance.save() |
|
165 |
return self.json_response(self.instance_to_dict(instance, |
|
166 |
self.exclude_fields)) |
|
167 |
|
|
168 |
class UserResourceView(ResourceView): |
|
169 |
""" |
|
170 |
Filter resource queryset for request user entries |
|
171 |
""" |
|
172 |
def queryset(self): |
|
173 |
return super(UserResourceView, |
|
174 |
self).queryset().filter(user=self.request.user) |
|
175 |
|
|
176 |
class UserCollectionView(CollectionView): |
|
177 |
""" |
|
178 |
Filter collection queryset for request user entries |
|
179 |
""" |
|
180 |
def queryset(self): |
|
181 |
return super(UserCollectionView, self).queryset().filter(user=self.request.user) |
|
182 |
|
|
183 |
def POST(self, request, data, *args, **kwargs): |
|
184 |
instance = self.model() |
|
185 |
self.update_instance(instance, data, self.exclude_fields) |
|
186 |
instance.user = request.user |
|
187 |
instance.save() |
|
188 |
return self.json_response(self.instance_to_dict(instance, |
|
189 |
self.exclude_fields)) |
|
190 |
|
/dev/null | ||
---|---|---|
1 |
{% load i18n %} |
|
2 |
<div class="public-keys-view clearfix"> |
|
3 |
<div class="loading-models">{% trans "Loading..." %}</div> |
|
4 |
<div class="models-view"> |
|
5 |
<div class="previous-view-link"><div class="change-view-action">Back to |
|
6 |
machine create wizard</div></div> |
|
7 |
|
|
8 |
<div class="list-wrapper model-list"> |
|
9 |
<h3 class="list-title hidden">{% trans "SSH public keys list" %}</h3> |
|
10 |
<div class="top-actions clearfix"> |
|
11 |
<div class="collection-action add add-new">{% trans "create/import new" %}</div> |
|
12 |
<div class="collection-action generate add-generate">{% trans "generate new" %}</div> |
|
13 |
</div> |
|
14 |
<div class="model-description"> |
|
15 |
<p>You can use SSH keys to establish a secure connection |
|
16 |
between your computer and the virtual machines. </p> |
|
17 |
</div> |
|
18 |
<div class="list-messages"> |
|
19 |
</div> |
|
20 |
<div class="private-cont"> |
|
21 |
<div class="private-download clearfix"> |
|
22 |
<div class="close-private">{% trans "close" %}</div> |
|
23 |
<div class="private-msg download"> |
|
24 |
</div> |
|
25 |
<div class="private-msg copy"> |
|
26 |
{% trans "Create a file with the following contents" %} |
|
27 |
</div> |
|
28 |
<form target="_blank" id="private-key-form" method="post"> |
|
29 |
<input type="hidden" name="data" /> |
|
30 |
<input type="hidden" name="name" /> |
|
31 |
|
|
32 |
<span class="form-text">{% trans "Your new public key has been added" %}</span> |
|
33 |
<input type="submit" class="down-button form-text" |
|
34 |
id="download-private-key" value="click here" /> |
|
35 |
<span class="form-text"> |
|
36 |
{% trans " to download private key." %} |
|
37 |
</span> |
|
38 |
</form> |
|
39 |
<div class="key-contents clearfix"><textarea></textarea></div> |
|
40 |
</div> |
|
41 |
</div> |
|
42 |
<div class="hidden public-key-item clearfix" id="model-item-tpl"> |
|
43 |
<div class="param key-type"></div> |
|
44 |
<div class="param name"></div> |
|
45 |
<div class="param publicid hidden"> |
|
46 |
<div class="param-content"><textarea></textarea></div> |
|
47 |
</div> |
|
48 |
</div> |
|
49 |
<ul class="keys-list"> |
|
50 |
<li class="header"> |
|
51 |
<div class="title">{% trans "Name" %}</div> |
|
52 |
<div class="title">{% trans "Public key ID" %}</div> |
|
53 |
</li> |
|
54 |
</ul> |
|
55 |
<ul class="items-list"> |
|
56 |
<li class="items-empty-msg hidden msg"> |
|
57 |
<div class="title">{% trans "No public keys exist" %} <span class="quick-add">add one</span> now</div> |
|
58 |
</li> |
|
59 |
</ul> |
|
60 |
</div> |
|
61 |
<div class="form-wrapper model-form clearfix"> |
|
62 |
<h3 class="new-title hidden">{% trans "Create new SSH public key" %}</h3> |
|
63 |
<h3 class="edit-title hidden">{% trans "Edit SSH public key" %}</h3> |
|
64 |
<form class="model-form"> |
|
65 |
<div class="form-messages"> |
|
66 |
</div> |
|
67 |
<div class="model-form"> |
|
68 |
<div class="form-field"> |
|
69 |
<label for="">{% trans "Key name" %}</label> |
|
70 |
<div class="errors"></div> |
|
71 |
<input type="text" class="input-name inline"/> |
|
72 |
</div> |
|
73 |
<div class="form-field clearfix"> |
|
74 |
<label for="">{% trans "Key content" %}</label> |
|
75 |
<div class="errors"></div> |
|
76 |
<textarea class="input-content"></textarea> |
|
77 |
</div> |
|
78 |
<div class="form-field inline clearfix fromfile-field"> |
|
79 |
<label for="">{% trans "Import from file" %}</label> |
|
80 |
<div class="fromfile"> |
|
81 |
<input type="file" class="content-input-file" |
|
82 |
title="Import from file" /> |
|
83 |
</div> |
|
84 |
</div> |
|
85 |
<div class="form-field"> |
|
86 |
</div> |
|
87 |
</div> |
|
88 |
</form> |
|
89 |
<div class="form-actions clearfix"> |
|
90 |
<div class="form-action submit">{% trans "Submit" %}</div> |
|
91 |
<div class="form-action cancel">{% trans "Cancel" %}</div> |
|
92 |
</div> |
|
93 |
</div> |
|
94 |
</div> |
|
95 |
</div> |
|
96 |
</div> |
/dev/null | ||
---|---|---|
1 |
""" |
|
2 |
This file demonstrates two different styles of tests (one doctest and one |
|
3 |
unittest). These will both pass when you run "manage.py test". |
|
4 |
|
|
5 |
Replace these with more appropriate tests for your application. |
|
6 |
""" |
|
7 |
|
|
8 |
from django.test import TestCase |
|
9 |
from django.conf import settings |
|
10 |
from django.test.client import Client |
|
11 |
from django.core.urlresolvers import clear_url_caches |
|
12 |
from django.utils import simplejson as json |
|
13 |
|
|
14 |
from synnefo.userdata.models import User |
|
15 |
from synnefo.userdata.models import * |
|
16 |
|
|
17 |
class AaiClient(Client): |
|
18 |
|
|
19 |
def request(self, **request): |
|
20 |
request['HTTP_X_AUTH_TOKEN'] = '46e427d657b20defe352804f0eb6f8a2' |
|
21 |
return super(AaiClient, self).request(**request) |
|
22 |
|
|
23 |
class TestRestViews(TestCase): |
|
24 |
|
|
25 |
fixtures = ['users'] |
|
26 |
|
|
27 |
def setUp(self): |
|
28 |
settings.ROOT_URLCONF = 'synnefo.userdata.urls' |
|
29 |
clear_url_caches() |
|
30 |
self.client = AaiClient() |
|
31 |
self.user = User.objects.get(pk=1) |
|
32 |
|
|
33 |
def test_keys_collection_get(self): |
|
34 |
resp = self.client.get("/keys") |
|
35 |
self.assertEqual(resp.content, "[]") |
|
36 |
|
|
37 |
PublicKeyPair.objects.create(user=self.user, name="key pair 1", |
|
38 |
content="content1") |
|
39 |
|
|
40 |
resp = self.client.get("/keys") |
|
41 |
self.assertEqual(resp.content, """[{"content": "content1", "uri": "/keys/1", "name": "key pair 1", "id": 1}]""") |
|
42 |
|
|
43 |
PublicKeyPair.objects.create(user=self.user, name="key pair 2", |
|
44 |
content="content2") |
|
45 |
|
|
46 |
resp = self.client.get("/keys") |
|
47 |
self.assertEqual(resp.content, """[{"content": "content1", "uri": "/keys/1", "name": "key pair 1", "id": 1}, {"content": "content2", "uri": "/keys/2", "name": "key pair 2", "id": 2}]""") |
|
48 |
|
|
49 |
def test_keys_resourse_get(self): |
|
50 |
resp = self.client.get("/keys/1") |
|
51 |
self.assertEqual(resp.status_code, 404) |
|
52 |
|
|
53 |
# create a public key |
|
54 |
PublicKeyPair.objects.create(user=self.user, name="key pair 1", |
|
55 |
content="content1") |
|
56 |
resp = self.client.get("/keys/1") |
|
57 |
self.assertEqual(resp.content, """{"content": "content1", "uri": "/keys/1", "name": "key pair 1", "id": 1}""") |
|
58 |
|
|
59 |
# update |
|
60 |
resp = self.client.put("/keys/1", json.dumps({'name':'key pair 1 new name'}), |
|
61 |
content_type='application/json') |
|
62 |
pk = PublicKeyPair.objects.get(pk=1) |
|
63 |
self.assertEqual(pk.name, "key pair 1 new name") |
|
64 |
|
|
65 |
# delete |
|
66 |
resp = self.client.delete("/keys/1") |
|
67 |
self.assertEqual(PublicKeyPair.objects.count(), 0) |
|
68 |
|
|
69 |
resp = self.client.get("/keys/1") |
|
70 |
self.assertEqual(resp.status_code, 404) |
|
71 |
|
|
72 |
resp = self.client.get("/keys") |
|
73 |
self.assertEqual(resp.content, "[]") |
|
74 |
|
|
75 |
# test rest create |
|
76 |
resp = self.client.post("/keys", json.dumps({'name':'key pair 2', |
|
77 |
'content':"""key 2 content"""}), |
|
78 |
content_type='application/json') |
|
79 |
self.assertEqual(PublicKeyPair.objects.count(), 1) |
|
80 |
pk = PublicKeyPair.objects.get(pk=1) |
|
81 |
self.assertEqual(pk.name, "key pair 2") |
|
82 |
self.assertEqual(pk.content, "key 2 content") |
|
83 |
|
|
84 |
def test_generate_views(self): |
|
85 |
import base64 |
|
86 |
|
|
87 |
# just test that |
|
88 |
resp = self.client.get("/keys/generate") |
|
89 |
self.assertNotEqual(resp, "") |
|
90 |
|
|
91 |
data = json.loads(resp.content) |
|
92 |
self.assertEqual(data.has_key('private'), True) |
|
93 |
self.assertEqual(data.has_key('private'), True) |
|
94 |
|
|
95 |
# public key is base64 encoded |
|
96 |
base64.b64decode(data['public'].replace("ssh-rsa ","")) |
|
97 |
|
|
98 |
# remove header/footer |
|
99 |
private = "".join(data['private'].split("\n")[1:-1]) |
|
100 |
|
|
101 |
# private key is base64 encoded |
|
102 |
base64.b64decode(private) |
|
103 |
|
/dev/null | ||
---|---|---|
1 |
from django.conf.urls.defaults import * |
|
2 |
from synnefo.userdata import views |
|
3 |
|
|
4 |
urlpatterns = patterns('', |
|
5 |
url(r'^keys$', views.PublicKeyPairCollectionView.as_view('keys_resource'), |
|
6 |
name='keys_collection'), |
|
7 |
url(r'^keys/(?P<id>\d+)', |
|
8 |
views.PublicKeyPairResourceView.as_view('keys_resource'), |
|
9 |
name="keys_resource"), |
|
10 |
url(r'keys/generate', views.generate_key_pair, name="generate_public_key"), |
|
11 |
url(r'keys/download', views.download_private_key, name="download_public_key") |
|
12 |
) |
/dev/null | ||
---|---|---|
1 |
from django import http |
|
2 |
from django.template import RequestContext, loader |
|
3 |
from django.utils import simplejson as json |
|
4 |
from django.conf import settings |
|
5 |
|
|
6 |
from synnefo.userdata import rest |
|
7 |
from synnefo.userdata.models import PublicKeyPair |
|
8 |
|
|
9 |
import M2Crypto as M2C |
|
10 |
import base64 |
|
11 |
|
|
12 |
class PublicKeyPairResourceView(rest.UserResourceView): |
|
13 |
model = PublicKeyPair |
|
14 |
exclude_fields = ["user"] |
|
15 |
|
|
16 |
class PublicKeyPairCollectionView(rest.UserCollectionView): |
|
17 |
model = PublicKeyPair |
|
18 |
exclude_fields = ["user"] |
|
19 |
|
|
20 |
SSH_KEY_LENGTH = getattr(settings, 'USERDATA_SSH_KEY_LENGTH', 2048) |
|
21 |
SSH_KEY_EXPONENT = getattr(settings, 'USERDATA_SSH_KEY_EXPONENT', 65537) |
|
22 |
def generate_key_pair(request): |
|
23 |
""" |
|
24 |
Response to generate private/public RSA key pair |
|
25 |
""" |
|
26 |
# generate RSA key |
|
27 |
key = M2C.RSA.gen_key(SSH_KEY_LENGTH, SSH_KEY_EXPONENT, lambda x: ""); |
|
28 |
|
|
29 |
# get PEM string |
|
30 |
pem_buffer = M2C.BIO.MemoryBuffer() |
|
31 |
M2C.m2.rsa_write_key_no_cipher(key.rsa, pem_buffer._ptr(), lambda : "") |
|
32 |
pem = pem_buffer.getvalue() |
|
33 |
|
|
34 |
# generate public content |
|
35 |
public = "ssh-rsa %s" % base64.b64encode('\x00\x00\x00\x07ssh-rsa%s%s' % (key.pub()[0], key.pub()[1])) |
|
36 |
|
|
37 |
data = {'private': pem, 'public': public} |
|
38 |
return http.HttpResponse(json.dumps(data), mimetype="application/json") |
|
39 |
|
|
40 |
def download_private_key(request): |
|
41 |
""" |
|
42 |
Return key contents |
|
43 |
""" |
|
44 |
data = request.POST.get("data") |
|
45 |
name = request.POST.get("name", "key") |
|
46 |
|
|
47 |
response = http.HttpResponse(mimetype='application/x-pem-key') |
|
48 |
response['Content-Disposition'] = 'attachment; filename=%s.pem' % name |
|
49 |
response.write(data) |
|
50 |
return response |
Also available in: Unified diff