Statistics
| Branch: | Tag: | Revision:

root / pithos / api / functions.py @ 7912c32e

History | View | Annotate | Download (39.3 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
    shared = False
241
    if 'shared' in request.GET:
242
        shared = True
243
    
244
    try:
245
        containers = backend.list_containers(request.user, v_account, marker, limit, shared, until)
246
    except NotAllowedError:
247
        raise Unauthorized('Access denied')
248
    except NameError:
249
        containers = []
250
    
251
    if request.serialization == 'text':
252
        if len(containers) == 0:
253
            # The cloudfiles python bindings expect 200 if json/xml.
254
            response.status_code = 204
255
            return response
256
        response.status_code = 200
257
        response.content = '\n'.join(containers) + '\n'
258
        return response
259
    
260
    container_meta = []
261
    for x in containers:
262
        try:
263
            meta = backend.get_container_meta(request.user, v_account, x, until)
264
            policy = backend.get_container_policy(request.user, v_account, x)
265
        except NotAllowedError:
266
            raise Unauthorized('Access denied')
267
        except NameError:
268
            pass
269
        else:
270
            for k, v in policy.iteritems():
271
                meta['X-Container-Policy-' + k] = v
272
            container_meta.append(printable_header_dict(meta))
273
    if request.serialization == 'xml':
274
        data = render_to_string('containers.xml', {'account': v_account, 'containers': container_meta})
275
    elif request.serialization  == 'json':
276
        data = json.dumps(container_meta)
277
    response.status_code = 200
278
    response.content = data
279
    return response
280

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

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

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

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

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

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

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

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

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

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

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

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

    
1001
@api_method('DELETE')
1002
def object_delete(request, v_account, v_container, v_object):
1003
    # Normal Response Codes: 204
1004
    # Error Response Codes: serviceUnavailable (503),
1005
    #                       itemNotFound (404),
1006
    #                       unauthorized (401),
1007
    #                       badRequest (400)
1008
    
1009
    until = get_int_parameter(request.GET.get('until'))
1010
    try:
1011
        backend.delete_object(request.user, v_account, v_container, v_object, until)
1012
    except NotAllowedError:
1013
        raise Unauthorized('Access denied')
1014
    except NameError:
1015
        raise ItemNotFound('Object does not exist')
1016
    return HttpResponse(status=204)
1017

    
1018
@api_method()
1019
def method_not_allowed(request):
1020
    raise BadRequest('Method not allowed')