git.gag.com Git - debian/amanda/blob - device-src/s3.h

   1 /*
   2  * Copyright (c) 2008, 2009, 2010 Zmanda, Inc.  All Rights Reserved.
   3  *
   4  * This program is free software; you can redistribute it and/or modify it
   5  * under the terms of the GNU General Public License version 2 as published
   6  * by the Free Software Foundation.
   7  *
   8  * This program is distributed in the hope that it will be useful, but
   9  * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  10  * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  11  * for more details.
  12  *
  13  * You should have received a copy of the GNU General Public License along
  14  * with this program; if not, write to the Free Software Foundation, Inc.,
  15  * 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16  *
  17  * Contact information: Zmanda Inc., 465 S. Mathilda Ave., Suite 300
  18  * Sunnyvale, CA 94085, USA, or: http://www.zmanda.com
  19  */
  20
  21 #ifndef __S3_H__
  22 #define __S3_H__
  23 #include <glib.h>
  24 #include <curl/curl.h>
  25
  26 /*
  27  * Data types
  28  */
  29
  30 /* An opaque handle.  S3Handles should only be accessed from a single
  31  * thread at any given time, although it is fine to use different handles
  32  * in different threads simultaneously. */
  33 typedef struct S3Handle S3Handle;
  34
  35 /* Callback function to read data to upload
  36  *
  37  * @note this is the same as CURLOPT_READFUNCTION
  38  *
  39  * @param data: The pointer to write data to
  40  * @param size: The size of each "element" of the data buffer in bytes
  41  * @param nmemb: The number of elements in the data buffer.
  42  * So, the buffer's size is size*nmemb bytes.
  43  * @param stream: The read_data (an opaque pointer)
  44  *
  45  * @return The number of bytes written to the buffer,
  46  * CURL_READFUNC_PAUSE to pause, or CURL_READFUNC_ABORT to abort.
  47  * Return 0 only if there's no more data to be uploaded.
  48  */
  49 typedef size_t (*s3_read_func)(void *data, size_t size, size_t nmemb, void *stream);
  50
  51 /* This function is called to get size of the upload data
  52  *
  53  * @param data: The write_data (opaque pointer)
  54  *
  55  * @return The number of bytes of data, negative for error
  56  */
  57 typedef size_t (*s3_size_func)(void *data);
  58
  59 /* This function is called to get MD5 hash of the upload data
  60  *
  61  * @param data: The write_data (opaque pointer)
  62  *
  63  * @return The MD5 hash, NULL on error
  64  */
  65 typedef GByteArray* (*s3_md5_func)(void *data);
  66
  67 /* This function is called to reset an upload or download data stream
  68  * to the beginning
  69  *
  70  * @param data: The read_data or write_data (opaque pointer)
  71  *
  72  * @return The number of bytes of data, negative for error
  73  */
  74 typedef void (*s3_reset_func)(void *data);
  75
  76 /* Callback function to write data that's been downloaded
  77  *
  78  * @note this is the same as CURLOPT_WRITEFUNCTION
  79  *
  80  * @param data: The pointer to read data from
  81  * @param size: The size of each "element" of the data buffer in bytes
  82  * @param nmemb: The number of elements in the data buffer.
  83  * So, the buffer's size is size*nmemb bytes.
  84  * @param stream: the write_data (an opaque pointer)
  85  *
  86  * @return The number of bytes written to the buffer or
  87  * CURL_WRITEFUNC_PAUSE to pause.
  88  * If it's the number of bytes written, it should match the buffer size
  89  */
  90 typedef size_t (*s3_write_func)(void *data, size_t size, size_t nmemb, void *stream);
  91
  92 /**
  93  * Callback function to track progress
  94  *
  95  * @note this is the same as CURLOPT_PROGRESSFUNCTION
  96  *
  97  * @param data: The progress_data
  98  * @param dltotal: The total number of bytes to downloaded
  99  * @param dlnow: The current number of bytes downloaded
 100  * @param ultotal: The total number of bytes to downloaded
 101  * @param ulnow: The current number of bytes downloaded
 102  *
 103  * @return 0 to continue, non-zero to abort.
 104  */
 105 typedef curl_progress_callback s3_progress_func;
 106
 107 /*
 108  * Constants
 109  */
 110
 111 /* These are assumed to be already URL-escaped. */
 112 # define STS_BASE_URL "https://ls.amazonaws.com/"
 113 # define STS_PRODUCT_TOKEN "{ProductToken}AAAGQXBwVGtu4geoGybuwuk8VEEPzJ9ZANpu0yzbf9g4Gs5Iarzff9B7qaDBEEaWcAzWpcN7zmdMO765jOtEFc4DWTRNkpPSzUnTdkHbdYUamath73OreaZtB86jy/JF0gsHZfhxeKc/3aLr8HNT//DsX3r272zYHLDPWWUbFguOwqNjllnt6BshYREx59l8RrWABLSa37dyJeN+faGvz3uQxiDakZRn3LfInOE6d9+fTFl50LPoP08LCqI/SJfpouzWix7D/cep3Jq8yYNyM1rgAOTF7/wh7r8OuPDLJ/xZUDLfykePIAM="
 114
 115 /* This preprocessor magic will enumerate constants named S3_ERROR_XxxYyy for
 116  * each of the errors in parentheses.
 117  *
 118  * see http://docs.amazonwebservices.com/AmazonS3/2006-03-01/ErrorCodeList.html
 119  * for Amazon's breakdown of error responses.
 120  */
 121 #define S3_ERROR_LIST \
 122     S3_ERROR(None), \
 123     S3_ERROR(AccountProblem), \
 124     S3_ERROR(AllAccessDisabled), \
 125     S3_ERROR(AmbiguousGrantByEmailAddress), \
 126     S3_ERROR(OperationAborted), \
 127     S3_ERROR(BadDigest), \
 128     S3_ERROR(BucketAlreadyExists), \
 129     S3_ERROR(BucketAlreadyOwnedByYou), \
 130     S3_ERROR(BucketNotEmpty), \
 131     S3_ERROR(CredentialsNotSupported), \
 132     S3_ERROR(EntityTooLarge), \
 133     S3_ERROR(IncompleteBody), \
 134     S3_ERROR(InternalError), \
 135     S3_ERROR(InvalidAccessKeyId), \
 136     S3_ERROR(InvalidArgument), \
 137     S3_ERROR(InvalidBucketName), \
 138     S3_ERROR(InvalidDigest), \
 139     S3_ERROR(InvalidRange), \
 140     S3_ERROR(InvalidSecurity), \
 141     S3_ERROR(InvalidSOAPRequest), \
 142     S3_ERROR(InvalidStorageClass), \
 143     S3_ERROR(InvalidTargetBucketForLogging), \
 144     S3_ERROR(KeyTooLong), \
 145     S3_ERROR(InvalidURI), \
 146     S3_ERROR(MalformedACLError), \
 147     S3_ERROR(MaxMessageLengthExceeded), \
 148     S3_ERROR(MetadataTooLarge), \
 149     S3_ERROR(MethodNotAllowed), \
 150     S3_ERROR(MissingAttachment), \
 151     S3_ERROR(MissingContentLength), \
 152     S3_ERROR(MissingSecurityElement), \
 153     S3_ERROR(MissingSecurityHeader), \
 154     S3_ERROR(NoLoggingStatusForKey), \
 155     S3_ERROR(NoSuchBucket), \
 156     S3_ERROR(NoSuchKey), \
 157     S3_ERROR(NotImplemented), \
 158     S3_ERROR(NotSignedUp), \
 159     S3_ERROR(PreconditionFailed), \
 160     S3_ERROR(RequestTimeout), \
 161     S3_ERROR(RequestTimeTooSkewed), \
 162     S3_ERROR(RequestTorrentOfBucketError), \
 163     S3_ERROR(SignatureDoesNotMatch), \
 164     S3_ERROR(TooManyBuckets), \
 165     S3_ERROR(UnexpectedContent), \
 166     S3_ERROR(UnresolvableGrantByEmailAddress), \
 167     S3_ERROR(Unknown), \
 168     S3_ERROR(END)
 169
 170 typedef enum {
 171 #define S3_ERROR(NAME) S3_ERROR_ ## NAME
 172     S3_ERROR_LIST
 173 #undef S3_ERROR
 174 } s3_error_code_t;
 175
 176 /*
 177  * Functions
 178  */
 179
 180 /* Does this install of curl support SSL?
 181  *
 182  * @returns: boolean
 183  */
 184 gboolean
 185 s3_curl_supports_ssl(void);
 186
 187 /* Checks if the version of libcurl being used supports and checks
 188  * wildcard certificates correctly (used for the subdomains required
 189  * by location constraints).
 190  *
 191  * @returns: true if the version of libcurl is new enough
 192  */
 193 gboolean
 194 s3_curl_location_compat(void);
 195
 196 /* Checks if a bucket name is compatible with setting a location
 197  * constraint.
 198  *
 199  * @note This doesn't guarantee that bucket name is entirely valid,
 200  * just that using it as one (or more) subdomain(s) of s3.amazonaws.com
 201  * won't fail; that would prevent the reporting of useful messages from
 202  * the service.
 203  *
 204  * @param bucket: the bucket name
 205  * @returns: true if the bucket name is compatible
 206  */
 207 gboolean
 208 s3_bucket_location_compat(const char *bucket);
 209
 210 /* Initialize S3 operation
 211  *
 212  * If an error occurs in this function, diagnostic information is
 213  * printed to stderr.
 214  *
 215  * @returns: false if an error occurred
 216  */
 217 gboolean
 218 s3_init(void);
 219
 220 /* Set up an S3Handle.
 221  *
 222  * The concept of a bucket is defined by the Amazon S3 API.
 223  * See: "Components of Amazon S3" - API Version 2006-03-01 pg. 8
 224  *
 225  * @param access_key: the secret key for Amazon Web Services
 226  * @param secret_key: the secret key for Amazon Web Services
 227  * @param user_token: the user token for Amazon DevPay
 228  * @param bucket_location: the location constraint for buckets
 229  * @param storage_class: the storage class for new objects
 230  * @param ca_info: the path to pass to libcurl as the certificate authority.
 231  *                 see curl_easy_setopt() CURLOPT_CAINFO for more
 232  * @returns: the new S3Handle
 233  */
 234 S3Handle *
 235 s3_open(const char * access_key, const char *secret_key, const char * user_token,
 236         const char * bucket_location, const char * storage_class, const char * ca_info);
 237
 238 /* Deallocate an S3Handle
 239  *
 240  * @param hdl: the S3Handle object
 241  */
 242 void
 243 s3_free(S3Handle *hdl);
 244
 245 /* Reset the information about the last request, including
 246  * freeing any allocated memory.  The S3Handle itself is not
 247  * freed and may be used again.  This function is called
 248  * automatically as needed, and should be called to free memory
 249  * when the handle will not be used for some time.
 250  *
 251  * @param hdl: the S3Handle object
 252  */
 253 void
 254 s3_reset(S3Handle *hdl);
 255
 256 /* Get the error information for the last operation
 257  *
 258  * All results are returned via result parameters.  If any parameter is
 259  * NULL, that result will not be returned.  Caller is not responsible for
 260  * freeing any returned strings, although the results are only valid until
 261  * the next call to an S3 function with this handle.
 262  *
 263  * @param hdl: the S3Handle object
 264  * @param message: (result) the error message, or NULL if none exists
 265  * @param response_code: (result) the HTTP response code (or 0 if none exists)
 266  * @param s3_error_code: (result) the S3 error code (see constants, above)
 267  * @param s3_error_name: (result) the S3 error name (e.g., "RequestTimeout"),
 268  * or NULL if none exists
 269  * @param curl_code: (result) the curl error code (or 0 if none exists)
 270  * @param num_retries: (result) number of retries
 271  */
 272 void
 273 s3_error(S3Handle *hdl,
 274          const char **message,
 275          guint *response_code,
 276          s3_error_code_t *s3_error_code,
 277          const char **s3_error_name,
 278          CURLcode *curl_code,
 279          guint *num_retries);
 280
 281 /* Control verbose output of HTTP transactions, etc.
 282  *
 283  * @param hdl: the S3Handle object
 284  * @param verbose: if true, send HTTP transactions, etc. to debug output
 285  */
 286 void
 287 s3_verbose(S3Handle *hdl,
 288        gboolean verbose);
 289
 290 /* Control the use of SSL with HTTP transactions.
 291  *
 292  * @param hdl: the S3Handle object
 293  * @param use_ssl: if true, use SSL (if curl supports it)
 294  * @returns: true if the setting is valid
 295  */
 296 gboolean
 297 s3_use_ssl(S3Handle *hdl, gboolean use_ssl);
 298
 299 /* Control the throttling of S3 uploads.  Only supported with curl >= 7.15.5.
 300  *
 301  * @param hdl: the S3Handle object
 302  * @param max_send_speed: max speed (bytes/sec) at which to send
 303  * @returns: true if the setting is valid
 304  */
 305 gboolean
 306 s3_set_max_send_speed(S3Handle *hdl, guint64 max_send_speed);
 307
 308 /* Control the throttling of S3 downloads.  Only supported with curl >= 7.15.5.
 309  *
 310  * @param hdl: the S3Handle object
 311  * @param max_recv_speed: max speed (bytes/sec) at which to receive
 312  * @returns: true if the setting is valid
 313  */
 314 gboolean
 315 s3_set_max_recv_speed(S3Handle *hdl, guint64 max_recv_speed);
 316
 317 /* Get the error information from the last operation on this handle,
 318  * formatted as a string.
 319  *
 320  * Caller is responsible for freeing the resulting string.
 321  *
 322  * @param hdl: the S3Handle object
 323  * @returns: string, or NULL if no error occurred
 324  */
 325 char *
 326 s3_strerror(S3Handle *hdl);
 327
 328 /* Perform an upload.
 329  *
 330  * When this function returns, KEY and BUFFER remain the
 331  * responsibility of the caller.
 332  *
 333  * @param hdl: the S3Handle object
 334  * @param bucket: the bucket to which the upload should be made
 335  * @param key: the key to which the upload should be made
 336  * @param read_func: the callback for reading data
 337  * @param reset_func: the callback for to reset reading data
 338  * @param size_func: the callback to get the number of bytes to upload
 339  * @param md5_func: the callback to get the MD5 hash of the data to upload
 340  * @param read_data: pointer to pass to the above functions
 341  * @param progress_func: the callback for progress information
 342  * @param progress_data: pointer to pass to C{progress_func}
 343  *
 344  * @returns: false if an error ocurred
 345  */
 346 gboolean
 347 s3_upload(S3Handle *hdl,
 348           const char *bucket,
 349           const char *key,
 350           s3_read_func read_func,
 351           s3_reset_func reset_func,
 352           s3_size_func size_func,
 353           s3_md5_func md5_func,
 354           gpointer read_data,
 355           s3_progress_func progress_func,
 356           gpointer progress_data);
 357
 358 /* List all of the files matching the pseudo-glob C{PREFIX*DELIMITER*},
 359  * returning only that portion which matches C{PREFIX*DELIMITER}.  S3 supports
 360  * this particular semantics, making it quite efficient.  The returned list
 361  * should be freed by the caller.
 362  *
 363  * @param hdl: the S3Handle object
 364  * @param bucket: the bucket to list
 365  * @param prefix: the prefix
 366  * @param delimiter: delimiter (any length string)
 367  * @param list: (output) the list of files
 368  * @returns: FALSE if an error occurs
 369  */
 370 gboolean
 371 s3_list_keys(S3Handle *hdl,
 372               const char *bucket,
 373               const char *prefix,
 374               const char *delimiter,
 375               GSList **list);
 376
 377 /* Read an entire file, passing the contents to write_func buffer
 378  * by buffer.
 379  *
 380  * @param hdl: the S3Handle object
 381  * @param bucket: the bucket to read from
 382  * @param key: the key to read from
 383  * @param write_func: the callback for writing data
 384  * @param reset_func: the callback for to reset writing data
 385  * @param write_data: pointer to pass to C{write_func}
 386  * @param progress_func: the callback for progress information
 387  * @param progress_data: pointer to pass to C{progress_func}
 388  * @returns: FALSE if an error occurs
 389  */
 390 gboolean
 391 s3_read(S3Handle *hdl,
 392         const char *bucket,
 393         const char *key,
 394         s3_write_func write_func,
 395         s3_reset_func reset_func,
 396         gpointer write_data,
 397         s3_progress_func progress_func,
 398         gpointer progress_data);
 399
 400 /* Delete a file.
 401  *
 402  * @param hdl: the S3Handle object
 403  * @param bucket: the bucket to delete from
 404  * @param key: the key to delete
 405  * @returns: FALSE if an error occurs; a non-existent file is I{not} considered an error.
 406  */
 407 gboolean
 408 s3_delete(S3Handle *hdl,
 409           const char *bucket,
 410           const char *key);
 411
 412 /* Create a bucket.
 413  *
 414  * @param hdl: the S3Handle object
 415  * @param bucket: the bucket to create
 416  * @returns: FALSE if an error occurs
 417  */
 418 gboolean
 419 s3_make_bucket(S3Handle *hdl,
 420                const char *bucket);
 421
 422 /* Delete a bucket
 423  *
 424  * @note A bucket can not be deleted if it still contains keys
 425  *
 426  * @param hdl: the S3Handle object
 427  * @param bucket: the bucket to delete
 428  * @returns: FALSE if an error occurs
 429  */
 430 gboolean
 431 s3_delete_bucket(S3Handle *hdl,
 432                  const char *bucket);
 433
 434 /* Attempt a RefreshAWSSecurityToken on a token; if it succeeds, the old
 435  * token will be freed and replaced by the new. If it fails, the old
 436  * token is left unchanged and FALSE is returned. */
 437 gboolean sts_refresh_token(char ** token, const char * directory);
 438
 439 /* These functions are for if you want to use curl on your own. You get more
 440  * control, but it's a lot of work that way: */
 441 typedef struct {
 442     char *buffer;
 443     guint buffer_len;
 444     guint buffer_pos;
 445     guint max_buffer_size;
 446 } CurlBuffer;
 447
 448 #define S3_BUFFER_READ_FUNCS s3_buffer_read_func, s3_buffer_reset_func, s3_buffer_size_func, s3_buffer_md5_func
 449
 450 #define S3_BUFFER_WRITE_FUNCS s3_buffer_write_func, s3_buffer_reset_func
 451
 452 /* a CURLOPT_READFUNCTION to read data from a buffer. */
 453 size_t
 454 s3_buffer_read_func(void *ptr, size_t size, size_t nmemb, void * stream);
 455
 456 size_t
 457 s3_buffer_size_func(void *stream);
 458
 459 GByteArray*
 460 s3_buffer_md5_func(void *stream);
 461
 462 void
 463 s3_buffer_reset_func(void *stream);
 464
 465 #define S3_EMPTY_READ_FUNCS s3_empty_read_func, NULL, s3_empty_size_func, s3_empty_md5_func
 466
 467 /* a CURLOPT_WRITEFUNCTION to write data to a buffer. */
 468 size_t
 469 s3_buffer_write_func(void *ptr, size_t size, size_t nmemb, void *stream);
 470
 471 /* a CURLOPT_READFUNCTION that writes nothing. */
 472 size_t
 473 s3_empty_read_func(void *ptr, size_t size, size_t nmemb, void * stream);
 474
 475 size_t
 476 s3_empty_size_func(void *stream);
 477
 478 GByteArray*
 479 s3_empty_md5_func(void *stream);
 480
 481 #define S3_COUNTER_WRITE_FUNCS s3_counter_write_func, s3_counter_reset_func
 482
 483 /* a CURLOPT_WRITEFUNCTION to write data that just counts data.
 484  * s3_write_data should be NULL or a pointer to an gint64.
 485  */
 486 size_t
 487 s3_counter_write_func(void *ptr, size_t size, size_t nmemb, void *stream);
 488
 489 void
 490 s3_counter_reset_func(void *stream);
 491
 492 #ifdef _WIN32
 493 /* a CURLOPT_READFUNCTION to read data from a file. */
 494 size_t
 495 s3_file_read_func(void *ptr, size_t size, size_t nmemb, void * stream);
 496
 497 size_t
 498 s3_file_size_func(void *stream);
 499
 500 GByteArray*
 501 s3_file_md5_func(void *stream);
 502
 503 size_t
 504 s3_file_reset_func(void *stream);
 505
 506 /* a CURLOPT_WRITEFUNCTION to write data to a file. */
 507 size_t
 508 s3_file_write_func(void *ptr, size_t size, size_t nmemb, void *stream);
 509 #endif
 510
 511 /* Adds a null termination to a buffer. */
 512 void terminate_buffer(CurlBuffer *);
 513
 514 #endif