108 lines
		
	
	
		
			2.9 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			108 lines
		
	
	
		
			2.9 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/*****************************************************************************
 | 
						|
 *                                  _   _ ____  _
 | 
						|
 *  Project                     ___| | | |  _ \| |
 | 
						|
 *                             / __| | | | |_) | |
 | 
						|
 *                            | (__| |_| |  _ <| |___
 | 
						|
 *                             \___|\___/|_| \_\_____|
 | 
						|
 *
 | 
						|
 * $Id: getinmemory.c,v 1.13 2008-09-06 04:28:45 yangtse Exp $
 | 
						|
 *
 | 
						|
 * Example source code to show how the callback function can be used to
 | 
						|
 * download data into a chunk of memory instead of storing it in a file.
 | 
						|
 *
 | 
						|
 * This exact source code has not been verified to work.
 | 
						|
 */
 | 
						|
 | 
						|
#include <stdio.h>
 | 
						|
#include <stdlib.h>
 | 
						|
#include <string.h>
 | 
						|
 | 
						|
#include <curl/curl.h>
 | 
						|
#include <curl/types.h>
 | 
						|
#include <curl/easy.h>
 | 
						|
 | 
						|
struct MemoryStruct {
 | 
						|
  char *memory;
 | 
						|
  size_t size;
 | 
						|
};
 | 
						|
 | 
						|
static void *myrealloc(void *ptr, size_t size);
 | 
						|
 | 
						|
static void *myrealloc(void *ptr, size_t size)
 | 
						|
{
 | 
						|
  /* There might be a realloc() out there that doesn't like reallocing
 | 
						|
     NULL pointers, so we take care of it here */
 | 
						|
  if(ptr)
 | 
						|
    return realloc(ptr, size);
 | 
						|
  else
 | 
						|
    return malloc(size);
 | 
						|
}
 | 
						|
 | 
						|
static size_t
 | 
						|
WriteMemoryCallback(void *ptr, size_t size, size_t nmemb, void *data)
 | 
						|
{
 | 
						|
  size_t realsize = size * nmemb;
 | 
						|
  struct MemoryStruct *mem = (struct MemoryStruct *)data;
 | 
						|
 | 
						|
  mem->memory = myrealloc(mem->memory, mem->size + realsize + 1);
 | 
						|
  if (mem->memory) {
 | 
						|
    memcpy(&(mem->memory[mem->size]), ptr, realsize);
 | 
						|
    mem->size += realsize;
 | 
						|
    mem->memory[mem->size] = 0;
 | 
						|
  }
 | 
						|
  return realsize;
 | 
						|
}
 | 
						|
 | 
						|
int main(int argc, char **argv)
 | 
						|
{
 | 
						|
  CURL *curl_handle;
 | 
						|
 | 
						|
  struct MemoryStruct chunk;
 | 
						|
 | 
						|
  chunk.memory=NULL; /* we expect realloc(NULL, size) to work */
 | 
						|
  chunk.size = 0;    /* no data at this point */
 | 
						|
 | 
						|
  curl_global_init(CURL_GLOBAL_ALL);
 | 
						|
 | 
						|
  /* init the curl session */
 | 
						|
  curl_handle = curl_easy_init();
 | 
						|
 | 
						|
  /* specify URL to get */
 | 
						|
  curl_easy_setopt(curl_handle, CURLOPT_URL, "http://cool.haxx.se/");
 | 
						|
 | 
						|
  /* send all data to this function  */
 | 
						|
  curl_easy_setopt(curl_handle, CURLOPT_WRITEFUNCTION, WriteMemoryCallback);
 | 
						|
 | 
						|
  /* we pass our 'chunk' struct to the callback function */
 | 
						|
  curl_easy_setopt(curl_handle, CURLOPT_WRITEDATA, (void *)&chunk);
 | 
						|
 | 
						|
  /* some servers don't like requests that are made without a user-agent
 | 
						|
     field, so we provide one */
 | 
						|
  curl_easy_setopt(curl_handle, CURLOPT_USERAGENT, "libcurl-agent/1.0");
 | 
						|
 | 
						|
  /* get it! */
 | 
						|
  curl_easy_perform(curl_handle);
 | 
						|
 | 
						|
  /* cleanup curl stuff */
 | 
						|
  curl_easy_cleanup(curl_handle);
 | 
						|
 | 
						|
  /*
 | 
						|
   * Now, our chunk.memory points to a memory block that is chunk.size
 | 
						|
   * bytes big and contains the remote file.
 | 
						|
   *
 | 
						|
   * Do something nice with it!
 | 
						|
   *
 | 
						|
   * You should be aware of the fact that at this point we might have an
 | 
						|
   * allocated data block, and nothing has yet deallocated that data. So when
 | 
						|
   * you're done with it, you should free() it as a nice application.
 | 
						|
   */
 | 
						|
 | 
						|
  if(chunk.memory)
 | 
						|
    free(chunk.memory);
 | 
						|
 | 
						|
  /* we're done with libcurl, so clean it up */
 | 
						|
  curl_global_cleanup();
 | 
						|
 | 
						|
  return 0;
 | 
						|
}
 |