Statistics
| Branch: | Tag: | Revision:

root / pithos / api / functions.py @ f6c97079

History | View | Annotate | Download (39.1 kB)

1
# Copyright 2011 GRNET S.A. All rights reserved.
2
# 
3
# Redistribution and use in source and binary forms, with or
4
# without modification, are permitted provided that the following
5
# conditions are met:
6
# 
7
#   1. Redistributions of source code must retain the above
8
#      copyright notice, this list of conditions and the following
9
#      disclaimer.
10
# 
11
#   2. Redistributions in binary form must reproduce the above
12
#      copyright notice, this list of conditions and the following
13
#      disclaimer in the documentation and/or other materials
14
#      provided with the distribution.
15
# 
16
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
17
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
20
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
21
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
22
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
23
# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
24
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
26
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
27
# POSSIBILITY OF SUCH DAMAGE.
28
# 
29
# The views and conclusions contained in the software and
30
# documentation are those of the authors and should not be
31
# interpreted as representing official policies, either expressed
32
# or implied, of GRNET S.A.
33

    
34
import os
35
import logging
36
import hashlib
37

    
38
from django.conf import settings
39
from django.http import HttpResponse
40
from django.template.loader import render_to_string
41
from django.utils import simplejson as json
42
from django.utils.http import parse_etags
43
from xml.dom import minidom
44

    
45
from pithos.api.faults import (Fault, NotModified, BadRequest, Unauthorized, ItemNotFound, Conflict,
46
    LengthRequired, PreconditionFailed, RangeNotSatisfiable, UnprocessableEntity)
47
from pithos.api.util import (format_header_key, printable_header_dict, get_account_headers,
48
    put_account_headers, get_container_headers, put_container_headers, get_object_headers, put_object_headers,
49
    update_manifest_meta, update_sharing_meta, update_public_meta, validate_modification_preconditions,
50
    validate_matching_preconditions, split_container_object_string, copy_or_move_object,
51
    get_int_parameter, get_content_length, get_content_range, raw_input_socket,
52
    socket_read_iterator, object_data_response, put_object_block, hashmap_hash, api_method)
53
from pithos.backends import backend
54
from pithos.backends.base import NotAllowedError
55

    
56

    
57
logger = logging.getLogger(__name__)
58

    
59

    
60
def top_demux(request):
61
    if request.method == 'GET':
62
        if request.user:
63
            return account_list(request)
64
        return authenticate(request)
65
    else:
66
        return method_not_allowed(request)
67

    
68
def account_demux(request, v_account):
69
    if request.method == 'HEAD':
70
        return account_meta(request, v_account)
71
    elif request.method == 'POST':
72
        return account_update(request, v_account)
73
    elif request.method == 'GET':
74
        return container_list(request, v_account)
75
    else:
76
        return method_not_allowed(request)
77

    
78
def container_demux(request, v_account, v_container):
79
    if request.method == 'HEAD':
80
        return container_meta(request, v_account, v_container)
81
    elif request.method == 'PUT':
82
        return container_create(request, v_account, v_container)
83
    elif request.method == 'POST':
84
        return container_update(request, v_account, v_container)
85
    elif request.method == 'DELETE':
86
        return container_delete(request, v_account, v_container)
87
    elif request.method == 'GET':
88
        return object_list(request, v_account, v_container)
89
    else:
90
        return method_not_allowed(request)
91

    
92
def object_demux(request, v_account, v_container, v_object):
93
    if request.method == 'HEAD':
94
        return object_meta(request, v_account, v_container, v_object)
95
    elif request.method == 'GET':
96
        return object_read(request, v_account, v_container, v_object)
97
    elif request.method == 'PUT':
98
        return object_write(request, v_account, v_container, v_object)
99
    elif request.method == 'COPY':
100
        return object_copy(request, v_account, v_container, v_object)
101
    elif request.method == 'MOVE':
102
        return object_move(request, v_account, v_container, v_object)
103
    elif request.method == 'POST':
104
        if request.META.get('CONTENT_TYPE', '').startswith('multipart/form-data'):
105
            return object_write_form(request, v_account, v_container, v_object)
106
        return object_update(request, v_account, v_container, v_object)
107
    elif request.method == 'DELETE':
108
        return object_delete(request, v_account, v_container, v_object)
109
    else:
110
        return method_not_allowed(request)
111

    
112
@api_method('GET')
113
def authenticate(request):
114
    # Normal Response Codes: 204
115
    # Error Response Codes: serviceUnavailable (503),
116
    #                       unauthorized (401),
117
    #                       badRequest (400)
118
    
119
    x_auth_user = request.META.get('HTTP_X_AUTH_USER')
120
    x_auth_key = request.META.get('HTTP_X_AUTH_KEY')
121
    if not x_auth_user or not x_auth_key:
122
        raise BadRequest('Missing X-Auth-User or X-Auth-Key header')
123
    response = HttpResponse(status=204)
124
    inv_auth_tokens = dict((v, k) for k, v in settings.AUTH_TOKENS.items())
125
    response['X-Auth-Token'] = inv_auth_tokens.get(x_auth_user, '0000')
126
    response['X-Storage-Url'] = os.path.join(request.build_absolute_uri(),
127
                                            x_auth_user)
128
    return response
129

    
130
@api_method('GET', format_allowed=True)
131
def account_list(request):
132
    # Normal Response Codes: 200, 204
133
    # Error Response Codes: serviceUnavailable (503),
134
    #                       badRequest (400)
135
    
136
    response = HttpResponse()
137
    
138
    marker = request.GET.get('marker')
139
    limit = get_int_parameter(request.GET.get('limit'))
140
    if not limit:
141
        limit = 10000
142
    
143
    accounts = backend.list_accounts(request.user, marker, limit)
144
    
145
    if request.serialization == 'text':
146
        if len(accounts) == 0:
147
            # The cloudfiles python bindings expect 200 if json/xml.
148
            response.status_code = 204
149
            return response
150
        response.status_code = 200
151
        response.content = '\n'.join(accounts) + '\n'
152
        return response
153
    
154
    account_meta = []
155
    for x in accounts:
156
        try:
157
            meta = backend.get_account_meta(request.user, x)
158
            groups = backend.get_account_groups(request.user, x)
159
        except NotAllowedError:
160
            raise Unauthorized('Access denied')
161
        else:
162
            for k, v in groups.iteritems():
163
                meta['X-Container-Group-' + k] = ','.join(v)
164
            account_meta.append(printable_header_dict(meta))
165
    if request.serialization == 'xml':
166
        data = render_to_string('accounts.xml', {'accounts': account_meta})
167
    elif request.serialization  == 'json':
168
        data = json.dumps(account_meta)
169
    response.status_code = 200
170
    response.content = data
171
    return response
172

    
173
@api_method('HEAD')
174
def account_meta(request, v_account):
175
    # Normal Response Codes: 204
176
    # Error Response Codes: serviceUnavailable (503),
177
    #                       unauthorized (401),
178
    #                       badRequest (400)
179
    
180
    until = get_int_parameter(request.GET.get('until'))
181
    try:
182
        meta = backend.get_account_meta(request.user, v_account, until)
183
        groups = backend.get_account_groups(request.user, v_account)
184
    except NotAllowedError:
185
        raise Unauthorized('Access denied')
186
    
187
    response = HttpResponse(status=204)
188
    put_account_headers(response, meta, groups)
189
    return response
190

    
191
@api_method('POST')
192
def account_update(request, v_account):
193
    # Normal Response Codes: 202
194
    # Error Response Codes: serviceUnavailable (503),
195
    #                       unauthorized (401),
196
    #                       badRequest (400)
197
    
198
    meta, groups = get_account_headers(request)
199
    replace = True
200
    if 'update' in request.GET:
201
        replace = False    
202
    if groups:
203
        try:
204
            backend.update_account_groups(request.user, v_account, groups, replace)
205
        except NotAllowedError:
206
            raise Unauthorized('Access denied')
207
        except ValueError:
208
            raise BadRequest('Invalid groups header')
209
    try:
210
        backend.update_account_meta(request.user, v_account, meta, replace)
211
    except NotAllowedError:
212
        raise Unauthorized('Access denied')
213
    return HttpResponse(status=202)
214

    
215
@api_method('GET', format_allowed=True)
216
def container_list(request, v_account):
217
    # Normal Response Codes: 200, 204
218
    # Error Response Codes: serviceUnavailable (503),
219
    #                       itemNotFound (404),
220
    #                       unauthorized (401),
221
    #                       badRequest (400)
222
    
223
    until = get_int_parameter(request.GET.get('until'))
224
    try:
225
        meta = backend.get_account_meta(request.user, v_account, until)
226
        groups = backend.get_account_groups(request.user, v_account)
227
    except NotAllowedError:
228
        raise Unauthorized('Access denied')
229
    
230
    validate_modification_preconditions(request, meta)
231
    
232
    response = HttpResponse()
233
    put_account_headers(response, meta, groups)
234
    
235
    marker = request.GET.get('marker')
236
    limit = get_int_parameter(request.GET.get('limit'))
237
    if not limit:
238
        limit = 10000
239
    
240
    try:
241
        containers = backend.list_containers(request.user, v_account, marker, limit, until)
242
    except NotAllowedError:
243
        raise Unauthorized('Access denied')
244
    except NameError:
245
        containers = []
246
    
247
    if request.serialization == 'text':
248
        if len(containers) == 0:
249
            # The cloudfiles python bindings expect 200 if json/xml.
250
            response.status_code = 204
251
            return response
252
        response.status_code = 200
253
        response.content = '\n'.join(containers) + '\n'
254
        return response
255
    
256
    container_meta = []
257
    for x in containers:
258
        try:
259
            meta = backend.get_container_meta(request.user, v_account, x, until)
260
            policy = backend.get_container_policy(request.user, v_account, x)
261
        except NotAllowedError:
262
            raise Unauthorized('Access denied')
263
        except NameError:
264
            pass
265
        else:
266
            for k, v in policy.iteritems():
267
                meta['X-Container-Policy-' + k] = v
268
            container_meta.append(printable_header_dict(meta))
269
    if request.serialization == 'xml':
270
        data = render_to_string('containers.xml', {'account': v_account, 'containers': container_meta})
271
    elif request.serialization  == 'json':
272
        data = json.dumps(container_meta)
273
    response.status_code = 200
274
    response.content = data
275
    return response
276

    
277
@api_method('HEAD')
278
def container_meta(request, v_account, v_container):
279
    # Normal Response Codes: 204
280
    # Error Response Codes: serviceUnavailable (503),
281
    #                       itemNotFound (404),
282
    #                       unauthorized (401),
283
    #                       badRequest (400)
284
    
285
    until = get_int_parameter(request.GET.get('until'))
286
    try:
287
        meta = backend.get_container_meta(request.user, v_account, v_container, until)
288
        meta['object_meta'] = backend.list_object_meta(request.user, v_account, v_container, until)
289
        policy = backend.get_container_policy(request.user, v_account, v_container)
290
    except NotAllowedError:
291
        raise Unauthorized('Access denied')
292
    except NameError:
293
        raise ItemNotFound('Container does not exist')
294
    
295
    response = HttpResponse(status=204)
296
    put_container_headers(response, meta, policy)
297
    return response
298

    
299
@api_method('PUT')
300
def container_create(request, v_account, v_container):
301
    # Normal Response Codes: 201, 202
302
    # Error Response Codes: serviceUnavailable (503),
303
    #                       itemNotFound (404),
304
    #                       unauthorized (401),
305
    #                       badRequest (400)
306
    
307
    meta, policy = get_container_headers(request)
308
    
309
    try:
310
        backend.put_container(request.user, v_account, v_container, policy)
311
        ret = 201
312
    except NotAllowedError:
313
        raise Unauthorized('Access denied')
314
    except NameError:
315
        ret = 202
316
    
317
    if len(meta) > 0:
318
        try:
319
            backend.update_container_meta(request.user, v_account, v_container, meta, replace=True)
320
        except NotAllowedError:
321
            raise Unauthorized('Access denied')
322
        except NameError:
323
            raise ItemNotFound('Container does not exist')
324
    
325
    return HttpResponse(status=ret)
326

    
327
@api_method('POST')
328
def container_update(request, v_account, v_container):
329
    # Normal Response Codes: 202
330
    # Error Response Codes: serviceUnavailable (503),
331
    #                       itemNotFound (404),
332
    #                       unauthorized (401),
333
    #                       badRequest (400)
334
    
335
    meta, policy = get_container_headers(request)
336
    replace = True
337
    if 'update' in request.GET:
338
        replace = False
339
    if policy:
340
        try:
341
            backend.update_container_policy(request.user, v_account, v_container, policy, replace)
342
        except NotAllowedError:
343
            raise Unauthorized('Access denied')
344
        except NameError:
345
            raise ItemNotFound('Container does not exist')
346
        except ValueError:
347
            raise BadRequest('Invalid policy header')
348
    try:
349
        backend.update_container_meta(request.user, v_account, v_container, meta, replace)
350
    except NotAllowedError:
351
        raise Unauthorized('Access denied')
352
    except NameError:
353
        raise ItemNotFound('Container does not exist')
354
    return HttpResponse(status=202)
355

    
356
@api_method('DELETE')
357
def container_delete(request, v_account, v_container):
358
    # Normal Response Codes: 204
359
    # Error Response Codes: serviceUnavailable (503),
360
    #                       conflict (409),
361
    #                       itemNotFound (404),
362
    #                       unauthorized (401),
363
    #                       badRequest (400)
364
    
365
    until = get_int_parameter(request.GET.get('until'))
366
    try:
367
        backend.delete_container(request.user, v_account, v_container, until)
368
    except NotAllowedError:
369
        raise Unauthorized('Access denied')
370
    except NameError:
371
        raise ItemNotFound('Container does not exist')
372
    except IndexError:
373
        raise Conflict('Container is not empty')
374
    return HttpResponse(status=204)
375

    
376
@api_method('GET', format_allowed=True)
377
def object_list(request, v_account, v_container):
378
    # Normal Response Codes: 200, 204
379
    # Error Response Codes: serviceUnavailable (503),
380
    #                       itemNotFound (404),
381
    #                       unauthorized (401),
382
    #                       badRequest (400)
383
    
384
    until = get_int_parameter(request.GET.get('until'))
385
    try:
386
        meta = backend.get_container_meta(request.user, v_account, v_container, until)
387
        meta['object_meta'] = backend.list_object_meta(request.user, v_account, v_container, until)
388
        policy = backend.get_container_policy(request.user, v_account, v_container)
389
    except NotAllowedError:
390
        raise Unauthorized('Access denied')
391
    except NameError:
392
        raise ItemNotFound('Container does not exist')
393
    
394
    validate_modification_preconditions(request, meta)
395
    
396
    response = HttpResponse()
397
    put_container_headers(response, meta, policy)
398
    
399
    path = request.GET.get('path')
400
    prefix = request.GET.get('prefix')
401
    delimiter = request.GET.get('delimiter')
402
    
403
    # Path overrides prefix and delimiter.
404
    virtual = True
405
    if path:
406
        prefix = path
407
        delimiter = '/'
408
        virtual = False
409
    
410
    # Naming policy.
411
    if prefix and delimiter:
412
        prefix = prefix + delimiter
413
    if not prefix:
414
        prefix = ''
415
    prefix = prefix.lstrip('/')
416
    
417
    marker = request.GET.get('marker')
418
    limit = get_int_parameter(request.GET.get('limit'))
419
    if not limit:
420
        limit = 10000
421
    
422
    keys = request.GET.get('meta')
423
    if keys:
424
        keys = keys.split(',')
425
        keys = [format_header_key('X-Object-Meta-' + x.strip()) for x in keys if x.strip() != '']
426
    else:
427
        keys = []
428
    
429
    try:
430
        objects = backend.list_objects(request.user, v_account, v_container, prefix, delimiter, marker, limit, virtual, keys, until)
431
    except NotAllowedError:
432
        raise Unauthorized('Access denied')
433
    except NameError:
434
        raise ItemNotFound('Container does not exist')
435
    
436
    if request.serialization == 'text':
437
        if len(objects) == 0:
438
            # The cloudfiles python bindings expect 200 if json/xml.
439
            response.status_code = 204
440
            return response
441
        response.status_code = 200
442
        response.content = '\n'.join([x[0] for x in objects]) + '\n'
443
        return response
444
    
445
    object_meta = []
446
    for x in objects:
447
        if x[1] is None:
448
            # Virtual objects/directories.
449
            object_meta.append({'subdir': x[0]})
450
        else:
451
            try:
452
                meta = backend.get_object_meta(request.user, v_account, v_container, x[0], x[1])
453
                if until is None:
454
                    permissions = backend.get_object_permissions(request.user, v_account, v_container, x[0])
455
                    public = backend.get_object_public(request.user, v_account, v_container, x[0])
456
                else:
457
                    permissions = None
458
                    public = None
459
            except NotAllowedError:
460
                raise Unauthorized('Access denied')
461
            except NameError:
462
                pass
463
            else:
464
                update_sharing_meta(permissions, v_account, v_container, x[0], meta)
465
                update_public_meta(public, meta)
466
                object_meta.append(printable_header_dict(meta))
467
    if request.serialization == 'xml':
468
        data = render_to_string('objects.xml', {'container': v_container, 'objects': object_meta})
469
    elif request.serialization  == 'json':
470
        data = json.dumps(object_meta)
471
    response.status_code = 200
472
    response.content = data
473
    return response
474

    
475
@api_method('HEAD')
476
def object_meta(request, v_account, v_container, v_object):
477
    # Normal Response Codes: 204
478
    # Error Response Codes: serviceUnavailable (503),
479
    #                       itemNotFound (404),
480
    #                       unauthorized (401),
481
    #                       badRequest (400)
482
    
483
    version = request.GET.get('version')
484
    try:
485
        meta = backend.get_object_meta(request.user, v_account, v_container, v_object, version)
486
        if version is None:
487
            permissions = backend.get_object_permissions(request.user, v_account, v_container, v_object)
488
            public = backend.get_object_public(request.user, v_account, v_container, v_object)
489
        else:
490
            permissions = None
491
            public = None
492
    except NotAllowedError:
493
        raise Unauthorized('Access denied')
494
    except NameError:
495
        raise ItemNotFound('Object does not exist')
496
    except IndexError:
497
        raise ItemNotFound('Version does not exist')
498
    
499
    update_manifest_meta(request, v_account, meta)
500
    update_sharing_meta(permissions, v_account, v_container, v_object, meta)
501
    update_public_meta(public, meta)
502
    
503
    response = HttpResponse(status=200)
504
    put_object_headers(response, meta)
505
    return response
506

    
507
@api_method('GET', format_allowed=True)
508
def object_read(request, v_account, v_container, v_object):
509
    # Normal Response Codes: 200, 206
510
    # Error Response Codes: serviceUnavailable (503),
511
    #                       rangeNotSatisfiable (416),
512
    #                       preconditionFailed (412),
513
    #                       itemNotFound (404),
514
    #                       unauthorized (401),
515
    #                       badRequest (400),
516
    #                       notModified (304)
517
    
518
    version = request.GET.get('version')
519
    
520
    # Reply with the version list. Do this first, as the object may be deleted.
521
    if version == 'list':
522
        if request.serialization == 'text':
523
            raise BadRequest('No format specified for version list.')
524
        
525
        try:
526
            v = backend.list_versions(request.user, v_account, v_container, v_object)
527
        except NotAllowedError:
528
            raise Unauthorized('Access denied')
529
        d = {'versions': v}
530
        if request.serialization == 'xml':
531
            d['object'] = v_object
532
            data = render_to_string('versions.xml', d)
533
        elif request.serialization  == 'json':
534
            data = json.dumps(d)
535
        
536
        response = HttpResponse(data, status=200)
537
        response['Content-Length'] = len(data)
538
        return response
539
    
540
    try:
541
        meta = backend.get_object_meta(request.user, v_account, v_container, v_object, version)
542
        if version is None:
543
            permissions = backend.get_object_permissions(request.user, v_account, v_container, v_object)
544
            public = backend.get_object_public(request.user, v_account, v_container, v_object)
545
        else:
546
            permissions = None
547
            public = None
548
    except NotAllowedError:
549
        raise Unauthorized('Access denied')
550
    except NameError:
551
        raise ItemNotFound('Object does not exist')
552
    except IndexError:
553
        raise ItemNotFound('Version does not exist')
554
    
555
    update_manifest_meta(request, v_account, meta)
556
    update_sharing_meta(permissions, v_account, v_container, v_object, meta)
557
    update_public_meta(public, meta)
558
    
559
    # Evaluate conditions.
560
    validate_modification_preconditions(request, meta)
561
    try:
562
        validate_matching_preconditions(request, meta)
563
    except NotModified:
564
        response = HttpResponse(status=304)
565
        response['ETag'] = meta['hash']
566
        return response
567
    
568
    sizes = []
569
    hashmaps = []
570
    if 'X-Object-Manifest' in meta:
571
        try:
572
            src_container, src_name = split_container_object_string('/' + meta['X-Object-Manifest'])
573
            objects = backend.list_objects(request.user, v_account, src_container, prefix=src_name, virtual=False)
574
        except NotAllowedError:
575
            raise Unauthorized('Access denied')
576
        except ValueError:
577
            raise BadRequest('Invalid X-Object-Manifest header')
578
        except NameError:
579
            raise ItemNotFound('Container does not exist')
580
        
581
        try:
582
            for x in objects:
583
                s, h = backend.get_object_hashmap(request.user, v_account, src_container, x[0], x[1])
584
                sizes.append(s)
585
                hashmaps.append(h)
586
        except NotAllowedError:
587
            raise Unauthorized('Access denied')
588
        except NameError:
589
            raise ItemNotFound('Object does not exist')
590
        except IndexError:
591
            raise ItemNotFound('Version does not exist')
592
    else:
593
        try:
594
            s, h = backend.get_object_hashmap(request.user, v_account, v_container, v_object, version)
595
            sizes.append(s)
596
            hashmaps.append(h)
597
        except NotAllowedError:
598
            raise Unauthorized('Access denied')
599
        except NameError:
600
            raise ItemNotFound('Object does not exist')
601
        except IndexError:
602
            raise ItemNotFound('Version does not exist')
603
    
604
    # Reply with the hashmap.
605
    if request.serialization != 'text':
606
        size = sum(sizes)
607
        hashmap = sum(hashmaps, [])
608
        d = {'block_size': backend.block_size, 'block_hash': backend.hash_algorithm, 'bytes': size, 'hashes': hashmap}
609
        if request.serialization == 'xml':
610
            d['object'] = v_object
611
            data = render_to_string('hashes.xml', d)
612
        elif request.serialization  == 'json':
613
            data = json.dumps(d)
614
        
615
        response = HttpResponse(data, status=200)
616
        put_object_headers(response, meta)
617
        response['Content-Length'] = len(data)
618
        return response
619
    
620
    return object_data_response(request, sizes, hashmaps, meta)
621

    
622
@api_method('PUT', format_allowed=True)
623
def object_write(request, v_account, v_container, v_object):
624
    # Normal Response Codes: 201
625
    # Error Response Codes: serviceUnavailable (503),
626
    #                       unprocessableEntity (422),
627
    #                       lengthRequired (411),
628
    #                       conflict (409),
629
    #                       itemNotFound (404),
630
    #                       unauthorized (401),
631
    #                       badRequest (400)
632
    
633
    if not request.GET.get('format'):
634
        request.serialization = 'text'
635
    
636
    copy_from = request.META.get('HTTP_X_COPY_FROM')
637
    move_from = request.META.get('HTTP_X_MOVE_FROM')
638
    if copy_from or move_from:
639
        content_length = get_content_length(request) # Required by the API.
640
        
641
        if move_from:
642
            try:
643
                src_container, src_name = split_container_object_string(move_from)
644
            except ValueError:
645
                raise BadRequest('Invalid X-Move-From header')
646
            copy_or_move_object(request, v_account, src_container, src_name, v_container, v_object, move=True)
647
        else:
648
            try:
649
                src_container, src_name = split_container_object_string(copy_from)
650
            except ValueError:
651
                raise BadRequest('Invalid X-Copy-From header')
652
            copy_or_move_object(request, v_account, src_container, src_name, v_container, v_object, move=False)
653
        return HttpResponse(status=201)
654
    
655
    meta, permissions, public = get_object_headers(request)
656
    content_length = -1
657
    if request.META.get('HTTP_TRANSFER_ENCODING') != 'chunked':
658
        content_length = get_content_length(request)
659
    # Should be BadRequest, but API says otherwise.
660
    if 'Content-Type' not in meta:
661
        raise LengthRequired('Missing Content-Type header')
662
    
663
    if request.serialization != 'text':
664
        data = ''
665
        sock = raw_input_socket(request)
666
        for block in socket_read_iterator(sock, content_length, backend.block_size):
667
            data = '%s%s' % (data, block)
668
        
669
        if request.serialization == 'json':
670
            d = json.loads(data)
671
            if not hasattr(d, '__getitem__'):
672
                raise BadRequest('Invalid data formating')
673
            try:
674
                hashmap = d['hashes']
675
                size = d['bytes']
676
            except KeyError:
677
                raise BadRequest('Invalid data formatting')
678
        elif request.serialization == 'xml':
679
            try:
680
                xml = minidom.parseString(data)
681
                obj = xml.getElementsByTagName('object')[0]
682
                size = obj.attributes['bytes'].value
683
                
684
                hashes = xml.getElementsByTagName('hash')
685
                hashmap = []
686
                for hash in hashes:
687
                    hashmap.append(hash.firstChild.data)
688
            except Exception:
689
                raise BadRequest('Invalid data formatting')
690
        
691
        meta.update({'hash': hashmap_hash(hashmap)}) # Update ETag.
692
    else:
693
        md5 = hashlib.md5()
694
        size = 0
695
        hashmap = []
696
        sock = raw_input_socket(request)
697
        for data in socket_read_iterator(sock, content_length, backend.block_size):
698
            # TODO: Raise 408 (Request Timeout) if this takes too long.
699
            # TODO: Raise 499 (Client Disconnect) if a length is defined and we stop before getting this much data.
700
            size += len(data)
701
            hashmap.append(backend.put_block(data))
702
            md5.update(data)
703
        
704
        meta['hash'] = md5.hexdigest().lower()
705
        etag = request.META.get('HTTP_ETAG')
706
        if etag and parse_etags(etag)[0].lower() != meta['hash']:
707
            raise UnprocessableEntity('Object ETag does not match')
708
    
709
    try:
710
        backend.update_object_hashmap(request.user, v_account, v_container, v_object, size, hashmap, meta, True, permissions)
711
    except NotAllowedError:
712
        raise Unauthorized('Access denied')
713
    except IndexError, e:
714
        raise Conflict(json.dumps(e.data))
715
    except NameError:
716
        raise ItemNotFound('Container does not exist')
717
    except ValueError:
718
        raise BadRequest('Invalid sharing header')
719
    except AttributeError, e:
720
        raise Conflict(json.dumps(e.data))
721
    if public is not None:
722
        try:
723
            backend.update_object_public(request.user, v_account, v_container, v_object, public)
724
        except NotAllowedError:
725
            raise Unauthorized('Access denied')
726
        except NameError:
727
            raise ItemNotFound('Object does not exist')
728
    
729
    response = HttpResponse(status=201)
730
    response['ETag'] = meta['hash']
731
    return response
732

    
733
@api_method('POST')
734
def object_write_form(request, v_account, v_container, v_object):
735
    # Normal Response Codes: 201
736
    # Error Response Codes: serviceUnavailable (503),
737
    #                       itemNotFound (404),
738
    #                       unauthorized (401),
739
    #                       badRequest (400)
740
    
741
    if not request.FILES.has_key('X-Object-Data'):
742
        raise BadRequest('Missing X-Object-Data field')
743
    file = request.FILES['X-Object-Data']
744
    
745
    meta = {}
746
    meta['Content-Type'] = file.content_type
747
    
748
    md5 = hashlib.md5()
749
    size = 0
750
    hashmap = []
751
    for data in file.chunks(backend.block_size):
752
        size += len(data)
753
        hashmap.append(backend.put_block(data))
754
        md5.update(data)
755
    
756
    meta['hash'] = md5.hexdigest().lower()
757
    
758
    try:
759
        backend.update_object_hashmap(request.user, v_account, v_container, v_object, size, hashmap, meta, True)
760
    except NotAllowedError:
761
        raise Unauthorized('Access denied')
762
    except NameError:
763
        raise ItemNotFound('Container does not exist')
764
    
765
    response = HttpResponse(status=201)
766
    response['ETag'] = meta['hash']
767
    return response
768

    
769
@api_method('COPY')
770
def object_copy(request, v_account, v_container, v_object):
771
    # Normal Response Codes: 201
772
    # Error Response Codes: serviceUnavailable (503),
773
    #                       itemNotFound (404),
774
    #                       unauthorized (401),
775
    #                       badRequest (400)
776
    
777
    dest_path = request.META.get('HTTP_DESTINATION')
778
    if not dest_path:
779
        raise BadRequest('Missing Destination header')
780
    try:
781
        dest_container, dest_name = split_container_object_string(dest_path)
782
    except ValueError:
783
        raise BadRequest('Invalid Destination header')
784
    copy_or_move_object(request, v_account, v_container, v_object, dest_container, dest_name, move=False)
785
    return HttpResponse(status=201)
786

    
787
@api_method('MOVE')
788
def object_move(request, v_account, v_container, v_object):
789
    # Normal Response Codes: 201
790
    # Error Response Codes: serviceUnavailable (503),
791
    #                       itemNotFound (404),
792
    #                       unauthorized (401),
793
    #                       badRequest (400)
794
    
795
    dest_path = request.META.get('HTTP_DESTINATION')
796
    if not dest_path:
797
        raise BadRequest('Missing Destination header')
798
    try:
799
        dest_container, dest_name = split_container_object_string(dest_path)
800
    except ValueError:
801
        raise BadRequest('Invalid Destination header')
802
    copy_or_move_object(request, v_account, v_container, v_object, dest_container, dest_name, move=True)
803
    return HttpResponse(status=201)
804

    
805
@api_method('POST')
806
def object_update(request, v_account, v_container, v_object):
807
    # Normal Response Codes: 202, 204
808
    # Error Response Codes: serviceUnavailable (503),
809
    #                       conflict (409),
810
    #                       itemNotFound (404),
811
    #                       unauthorized (401),
812
    #                       badRequest (400)
813
    meta, permissions, public = get_object_headers(request)
814
    content_type = meta.get('Content-Type')
815
    if content_type:
816
        del(meta['Content-Type']) # Do not allow changing the Content-Type.
817
    
818
    try:
819
        prev_meta = backend.get_object_meta(request.user, v_account, v_container, v_object)
820
    except NotAllowedError:
821
        raise Unauthorized('Access denied')
822
    except NameError:
823
        raise ItemNotFound('Object does not exist')
824
    # If replacing, keep previous values of 'Content-Type' and 'hash'.
825
    replace = True
826
    if 'update' in request.GET:
827
        replace = False
828
    if replace:
829
        for k in ('Content-Type', 'hash'):
830
            if k in prev_meta:
831
                meta[k] = prev_meta[k]
832
    
833
    # A Content-Type or X-Source-Object header indicates data updates.
834
    src_object = request.META.get('HTTP_X_SOURCE_OBJECT')
835
    if (not content_type or content_type != 'application/octet-stream') and not src_object:
836
        # Do permissions first, as it may fail easier.
837
        if permissions is not None:
838
            try:
839
                backend.update_object_permissions(request.user, v_account, v_container, v_object, permissions)
840
            except NotAllowedError:
841
                raise Unauthorized('Access denied')
842
            except NameError:
843
                raise ItemNotFound('Object does not exist')
844
            except ValueError:
845
                raise BadRequest('Invalid sharing header')
846
            except AttributeError, e:
847
                raise Conflict(json.dumps(e.data))
848
        if public is not None:
849
            try:
850
                backend.update_object_public(request.user, v_account, v_container, v_object, public)
851
            except NotAllowedError:
852
                raise Unauthorized('Access denied')
853
            except NameError:
854
                raise ItemNotFound('Object does not exist')
855
        try:
856
            backend.update_object_meta(request.user, v_account, v_container, v_object, meta, replace)
857
        except NotAllowedError:
858
            raise Unauthorized('Access denied')
859
        except NameError:
860
            raise ItemNotFound('Object does not exist')
861
        return HttpResponse(status=202)
862
    
863
    # Single range update. Range must be in Content-Range.
864
    # Based on: http://code.google.com/p/gears/wiki/ContentRangePostProposal
865
    # (with the addition that '*' is allowed for the range - will append).
866
    content_range = request.META.get('HTTP_CONTENT_RANGE')
867
    if not content_range:
868
        raise BadRequest('Missing Content-Range header')
869
    ranges = get_content_range(request)
870
    if not ranges:
871
        raise RangeNotSatisfiable('Invalid Content-Range header')
872
    
873
    try:
874
        size, hashmap = backend.get_object_hashmap(request.user, v_account, v_container, v_object)
875
    except NotAllowedError:
876
        raise Unauthorized('Access denied')
877
    except NameError:
878
        raise ItemNotFound('Object does not exist')
879
    
880
    offset, length, total = ranges
881
    if offset is None:
882
        offset = size
883
    elif offset > size:
884
        raise RangeNotSatisfiable('Supplied offset is beyond object limits')
885
    if src_object:
886
        src_container, src_name = split_container_object_string(src_object)
887
        src_version = request.META.get('HTTP_X_SOURCE_VERSION')
888
        try:
889
            src_size, src_hashmap = backend.get_object_hashmap(request.user, v_account, src_container, src_name, src_version)
890
        except NotAllowedError:
891
            raise Unauthorized('Access denied')
892
        except NameError:
893
            raise ItemNotFound('Source object does not exist')
894
        
895
        if length is None:
896
            length = src_size
897
        elif length > src_size:
898
            raise BadRequest('Object length is smaller than range length')
899
    else:
900
        # Require either a Content-Length, or 'chunked' Transfer-Encoding.
901
        content_length = -1
902
        if request.META.get('HTTP_TRANSFER_ENCODING') != 'chunked':
903
            content_length = get_content_length(request)
904
            
905
        if length is None:
906
            length = content_length
907
        else:
908
            if content_length == -1:
909
                # TODO: Get up to length bytes in chunks.
910
                length = content_length
911
            elif length != content_length:
912
                raise BadRequest('Content length does not match range length')
913
    if total is not None and (total != size or offset >= size or (length > 0 and offset + length >= size)):
914
        raise RangeNotSatisfiable('Supplied range will change provided object limits')
915
    
916
    dest_bytes = request.META.get('HTTP_X_OBJECT_BYTES')
917
    if dest_bytes is not None:
918
        dest_bytes = get_int_parameter(dest_bytes)
919
        if dest_bytes is None:
920
            raise BadRequest('Invalid X-Object-Bytes header')
921
    
922
    if src_object:
923
        if offset % backend.block_size == 0:
924
            # Update the hashes only.
925
            sbi = 0
926
            while length > 0:
927
                bi = int(offset / backend.block_size)
928
                bl = min(length, backend.block_size)
929
                if bi < len(hashmap):
930
                    if bl == backend.block_size:
931
                        hashmap[bi] = src_hashmap[sbi]
932
                    else:
933
                        data = backend.get_block(src_hashmap[sbi])
934
                        hashmap[bi] = backend.update_block(hashmap[bi], data[:bl], 0)
935
                else:
936
                    hashmap.append(src_hashmap[sbi])
937
                offset += bl
938
                length -= bl
939
                sbi += 1
940
        else:
941
            data = ''
942
            sbi = 0
943
            while length > 0:
944
                data += backend.get_block(src_hashmap[sbi])
945
                if length < backend.block_size:
946
                    data = data[:length]
947
                bytes = put_object_block(hashmap, data, offset)
948
                offset += bytes
949
                data = data[bytes:]
950
                length -= bytes
951
                sbi += 1
952
    else:
953
        sock = raw_input_socket(request)
954
        data = ''
955
        for d in socket_read_iterator(sock, length, backend.block_size):
956
            # TODO: Raise 408 (Request Timeout) if this takes too long.
957
            # TODO: Raise 499 (Client Disconnect) if a length is defined and we stop before getting this much data.
958
            data += d
959
            bytes = put_object_block(hashmap, data, offset)
960
            offset += bytes
961
            data = data[bytes:]
962
        if len(data) > 0:
963
            put_object_block(hashmap, data, offset)
964
    
965
    if offset > size:
966
        size = offset
967
    if dest_bytes is not None and dest_bytes < size:
968
        size = dest_bytes
969
        hashmap = hashmap[:(int((size - 1) / backend.block_size) + 1)]
970
    meta.update({'hash': hashmap_hash(hashmap)}) # Update ETag.
971
    try:
972
        backend.update_object_hashmap(request.user, v_account, v_container, v_object, size, hashmap, meta, replace, permissions)
973
    except NotAllowedError:
974
        raise Unauthorized('Access denied')
975
    except NameError:
976
        raise ItemNotFound('Container does not exist')
977
    except ValueError:
978
        raise BadRequest('Invalid sharing header')
979
    except AttributeError, e:
980
        raise Conflict(json.dumps(e.data))
981
    if public is not None:
982
        try:
983
            backend.update_object_public(request.user, v_account, v_container, v_object, public)
984
        except NotAllowedError:
985
            raise Unauthorized('Access denied')
986
        except NameError:
987
            raise ItemNotFound('Object does not exist')
988
    
989
    response = HttpResponse(status=204)
990
    response['ETag'] = meta['hash']
991
    return response
992

    
993
@api_method('DELETE')
994
def object_delete(request, v_account, v_container, v_object):
995
    # Normal Response Codes: 204
996
    # Error Response Codes: serviceUnavailable (503),
997
    #                       itemNotFound (404),
998
    #                       unauthorized (401),
999
    #                       badRequest (400)
1000
    
1001
    until = get_int_parameter(request.GET.get('until'))
1002
    try:
1003
        backend.delete_object(request.user, v_account, v_container, v_object, until)
1004
    except NotAllowedError:
1005
        raise Unauthorized('Access denied')
1006
    except NameError:
1007
        raise ItemNotFound('Object does not exist')
1008
    return HttpResponse(status=204)
1009

    
1010
@api_method()
1011
def method_not_allowed(request):
1012
    raise BadRequest('Method not allowed')