109 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			109 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/*****************************************************************************
 | 
						|
 *                                  _   _ ____  _
 | 
						|
 *  Project                     ___| | | |  _ \| |
 | 
						|
 *                             / __| | | | |_) | |
 | 
						|
 *                            | (__| |_| |  _ <| |___
 | 
						|
 *                             \___|\___/|_| \_\_____|
 | 
						|
 *
 | 
						|
 * $Id: lib547.c,v 1.6 2008-09-20 04:26:57 yangtse Exp $
 | 
						|
 *
 | 
						|
 * argv1 = URL
 | 
						|
 * argv2 = proxy
 | 
						|
 * argv3 = proxyuser:password
 | 
						|
 */
 | 
						|
 | 
						|
#include "test.h"
 | 
						|
 | 
						|
#include "memdebug.h"
 | 
						|
 | 
						|
#define UPLOADTHIS "this is the blurb we want to upload\n"
 | 
						|
 | 
						|
#ifndef LIB548
 | 
						|
static size_t readcallback(void  *ptr,
 | 
						|
                           size_t size,
 | 
						|
                           size_t nmemb,
 | 
						|
                           void *clientp)
 | 
						|
{
 | 
						|
  int *counter = (int *)clientp;
 | 
						|
 | 
						|
  if(*counter) {
 | 
						|
    /* only do this once and then require a clearing of this */
 | 
						|
    fprintf(stderr, "READ ALREADY DONE!\n");
 | 
						|
    return 0;
 | 
						|
  }
 | 
						|
  (*counter)++; /* bump */
 | 
						|
 | 
						|
  if(size * nmemb > strlen(UPLOADTHIS)) {
 | 
						|
    fprintf(stderr, "READ!\n");
 | 
						|
    strcpy(ptr, UPLOADTHIS);
 | 
						|
    return strlen(UPLOADTHIS);
 | 
						|
  }
 | 
						|
  fprintf(stderr, "READ NOT FINE!\n");
 | 
						|
  return 0;
 | 
						|
}
 | 
						|
static curlioerr ioctlcallback(CURL *handle,
 | 
						|
                               int cmd,
 | 
						|
                               void *clientp)
 | 
						|
{
 | 
						|
  int *counter = (int *)clientp;
 | 
						|
  (void)handle; /* unused */
 | 
						|
  if(cmd == CURLIOCMD_RESTARTREAD) {
 | 
						|
    fprintf(stderr, "REWIND!\n");
 | 
						|
    *counter = 0; /* clear counter to make the read callback restart */
 | 
						|
  }
 | 
						|
  return CURLIOE_OK;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
 | 
						|
#endif
 | 
						|
 | 
						|
int test(char *URL)
 | 
						|
{
 | 
						|
  CURLcode res;
 | 
						|
  CURL *curl;
 | 
						|
#ifndef LIB548
 | 
						|
  int counter=0;
 | 
						|
#endif
 | 
						|
 | 
						|
  if (curl_global_init(CURL_GLOBAL_ALL) != CURLE_OK) {
 | 
						|
    fprintf(stderr, "curl_global_init() failed\n");
 | 
						|
    return TEST_ERR_MAJOR_BAD;
 | 
						|
  }
 | 
						|
 | 
						|
  if ((curl = curl_easy_init()) == NULL) {
 | 
						|
    fprintf(stderr, "curl_easy_init() failed\n");
 | 
						|
    curl_global_cleanup();
 | 
						|
    return TEST_ERR_MAJOR_BAD;
 | 
						|
  }
 | 
						|
 | 
						|
  curl_easy_setopt(curl, CURLOPT_URL, URL);
 | 
						|
  curl_easy_setopt(curl, CURLOPT_VERBOSE, 1L);
 | 
						|
  curl_easy_setopt(curl, CURLOPT_HEADER, 1L);
 | 
						|
#ifdef LIB548
 | 
						|
  /* set the data to POST with a mere pointer to a zero-terminated string */
 | 
						|
  curl_easy_setopt(curl, CURLOPT_POSTFIELDS, UPLOADTHIS);
 | 
						|
#else
 | 
						|
  /* 547 style, which means reading the POST data from a callback */
 | 
						|
  curl_easy_setopt(curl, CURLOPT_IOCTLFUNCTION, ioctlcallback);
 | 
						|
  curl_easy_setopt(curl, CURLOPT_IOCTLDATA, &counter);
 | 
						|
  curl_easy_setopt(curl, CURLOPT_READFUNCTION, readcallback);
 | 
						|
  curl_easy_setopt(curl, CURLOPT_READDATA, &counter);
 | 
						|
  /* We CANNOT do the POST fine without setting the size (or choose chunked)! */
 | 
						|
  curl_easy_setopt(curl, CURLOPT_POSTFIELDSIZE, (long)strlen(UPLOADTHIS));
 | 
						|
#endif
 | 
						|
  curl_easy_setopt(curl, CURLOPT_POST, 1L);
 | 
						|
  curl_easy_setopt(curl, CURLOPT_PROXY, libtest_arg2);
 | 
						|
  curl_easy_setopt(curl, CURLOPT_PROXYUSERPWD, libtest_arg3);
 | 
						|
  curl_easy_setopt(curl, CURLOPT_PROXYAUTH,
 | 
						|
                   (long) (CURLAUTH_NTLM | CURLAUTH_DIGEST | CURLAUTH_BASIC) );
 | 
						|
 | 
						|
  res = curl_easy_perform(curl);
 | 
						|
 | 
						|
  curl_easy_cleanup(curl);
 | 
						|
  curl_global_cleanup();
 | 
						|
 | 
						|
  return (int)res;
 | 
						|
}
 | 
						|
 |