开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 通讯/手机编程 > 手机WAP编程 > 查看源码
test_ppg.c
资源名称：gateway-1.2.1　[点击查看]
上传用户：gzpyjq
上传日期：2013-01-31
资源大小：1852k
文件大小：26k
源码类别：

手机WAP编程

开发平台：

WINDOWS

test_ppg.c：源码内容
  1. /*
  2.  * A very simple push initiator for testing a push proxy gateway
  3.  *
  4.  * Read pap control content and push content from files, pack them into a PAP
  5.  * protocol MIME message and invoke push services specified by an url. Use a 
  6.  * hardcoded message boundary (asdlfkjiurwgasf), for simpler command line 
  7.  * interface.
  8.  * Repetitions and use of multiple threads can be requested, in addition of 
  9.  * setting of some headers. 
  10.  *
  11.  * By Aarno Syv鋘en for Wiral Ltd and Global Networks Inc.
  12.  */
  14. #define MAX_THREADS 1024
  15. #define MAX_IN_QUEUE 128
  17. #include <unistd.h>
  18. #include <stdlib.h>
  19. #include <string.h>
  21. #include "gwlib/gwlib.h"
  22. #include "gw/wap_push_pap_compiler.h"
  24. static long max_pushes = 1;
  25. static int verbose = 1,
  26.            use_hardcoded = 0,
  27.            num_urls = 0,
  28.            wait = 0,
  29.            use_headers = 0,
  30.            use_config = 0;
  31. static double wait_seconds = 0.0;
  32. static Counter *counter = NULL;
  33. static char **push_data = NULL;
  34. static char *boundary = NULL;
  35. static Octstr *content_flag = NULL;
  36. static Octstr *appid_flag = NULL;
  37. static Octstr *content_transfer_encoding = NULL;
  38. static Octstr *connection = NULL;
  40. enum { SSL_CONNECTION_OFF = 0,
  41.        DEFAULT_NUMBER_OF_RELOGS = 2};
  43. /*
  44.  * Configuration variables
  45.  */
  46. static int pi_ssl = SSL_CONNECTION_OFF;
  47. static long retries = DEFAULT_NUMBER_OF_RELOGS;
  48. static Octstr *ssl_client_certkey_file = NULL;
  49. static Octstr *push_url = NULL;
  50. static Octstr *pap_file = NULL;
  51. static Octstr *content_file = NULL;
  52. static Octstr *username = NULL;
  53. static Octstr *password = NULL;
  55. static void read_test_ppg_config(Octstr *name)
  56. {
  57.     Cfg *cfg;
  58.     CfgGroup *grp;
  60.     cfg = cfg_create(name);
  61.     if (cfg_read(cfg) == -1)
  62.         panic(0, "Cannot read a configuration file %s, exiting",
  63.               octstr_get_cstr(name));
  64.     cfg_dump(cfg);
  65.     grp = cfg_get_single_group(cfg, octstr_imm("test-ppg"));
  66.     cfg_get_integer(&retries, grp, octstr_imm("retries"));
  67.     cfg_get_bool(&pi_ssl, grp, octstr_imm("pi-ssl"));
  68. #ifdef HAVE_LIBSSL    
  69.     if (pi_ssl) {
  70.         ssl_client_certkey_file = cfg_get(grp, 
  71.             octstr_imm("ssl-client-certkey-file"));
  72.         if (ssl_client_certkey_file != NULL) {
  73.             use_global_client_certkey_file(ssl_client_certkey_file);
  74.         } else { 
  75.             error(0, "cannot set up SSL without client certkey file");
  76.             exit(1);
  77.         }
  78.     }
  79. #endif
  81.     grp = cfg_get_single_group(cfg, octstr_imm("configuration"));
  82.     push_url = cfg_get(grp, octstr_imm("push-url"));
  83.     pap_file =  cfg_get(grp, octstr_imm("pap-file"));
  84.     content_file =  cfg_get(grp, octstr_imm("content-file"));
  85.     if (!use_hardcoded) {
  86.         username = cfg_get(grp, octstr_imm("username"));
  87.         password = cfg_get(grp, octstr_imm("password"));
  88.     }
  90.     cfg_destroy(cfg);
  91. }
  93. static void add_push_application_id(List **push_headers, Octstr *appid_flag)
  94. {
  95.     if (octstr_compare(appid_flag, octstr_imm("any")) == 0)
  96.         http_header_add(*push_headers, "X-WAP-Application-Id", 
  97.                         "http://www.wiral.com:*");
  98.     else if (octstr_compare(appid_flag, octstr_imm("ua")) == 0)
  99.         http_header_add(*push_headers, "X-WAP-Application-Id", 
  100.                         "http://www.wiral.com:wml.ua");
  101.     else if (octstr_compare(appid_flag, octstr_imm("mms")) == 0)
  102.         http_header_add(*push_headers, "X-WAP-Application-Id", 
  103.                         "mms.ua");
  104.     else if (octstr_compare(appid_flag, octstr_imm("scrap")) == 0)
  105.         http_header_add(*push_headers, "X-WAP-Application-Id", 
  106.                         "no appid at all");
  107. }
  109. static void add_content_type(Octstr *content_flag, Octstr **wap_content)
  110. {
  111.     if (octstr_compare(content_flag, octstr_imm("wml")) == 0)
  112.         *wap_content = octstr_format("%s", 
  113.             "Content-Type: text/vnd.wap.wmlrn");
  114.     else if (octstr_compare(content_flag, octstr_imm("si")) == 0)
  115.     *wap_content = octstr_format("%s",
  116.             "Content-Type: text/vnd.wap.sirn");
  117.     else if (octstr_compare(content_flag, octstr_imm("sl")) == 0)
  118.     *wap_content = octstr_format("%s",
  119.             "Content-Type: text/vnd.wap.slrn");
  120.     else if (octstr_compare(content_flag, octstr_imm("multipart")) == 0)
  121.         *wap_content = octstr_format("%s",
  122.             "Content-Type: multipart/related; boundary=fsahgwruijkfldsarn");
  123.     else if (octstr_compare(content_flag, octstr_imm("mms")) == 0) 
  124.         *wap_content = octstr_format("%s", 
  125.             "Content-Type: application/vnd.wap.mms-messagern"); 
  126.     else if (octstr_compare(content_flag, octstr_imm("scrap")) == 0)
  127.         *wap_content = octstr_format("%s", "no type at allrn"); 
  128.     else if (octstr_compare(content_flag, octstr_imm("nil")) == 0)
  129.         *wap_content = octstr_create("");
  130. }
  132. static void add_content_transfer_encoding_type(Octstr *content_flag, 
  133.                                                Octstr *wap_content)
  134. {
  135.     if (!content_flag)
  136.      return;
  138.     if (octstr_compare(content_flag, octstr_imm("base64")) == 0)
  139.     octstr_append_cstr(wap_content, "Content-transfer-encoding: base64rn");
  140. }
  142. static void add_connection_header(Octstr *connection, Octstr *wap_content)
  143. {
  144.     if (!connection)
  145.         return;
  147.     if (octstr_compare(connection, octstr_imm("close")) == 0)
  148.         octstr_format_append(wap_content, "%s", "Connection: closern");
  149.     else if (octstr_compare(connection, octstr_imm("keep-alive")) == 0) 
  150.         octstr_format_append(wap_content, "%s", "Connection: keep-alivern");
  151. }
  153. static void transfer_encode (Octstr *cte, Octstr *content)
  154. {
  155.     if (!cte)
  156.     return;
  157.     
  158.     if (octstr_compare(cte, octstr_imm("base64")) == 0) {
  159.     octstr_binary_to_base64(content);
  160.     }
  161. }
  164. /*
  165.  * Add boundary value to the multipart header.
  166.  */
  167. static Octstr *make_multipart_value(const char *boundary)
  168. {
  169.     Octstr *hos;
  170.     
  171.     hos = octstr_format("%s", "multipart/related; boundary=");
  172.     octstr_append(hos, octstr_imm(boundary));
  173.     octstr_append(hos, octstr_imm("; type="application/xml""));
  174.     
  175.     return hos;
  176. }
  178. static Octstr *make_part_delimiter(Octstr *boundary)
  179. {
  180.     Octstr *part_delimiter;
  182.     part_delimiter = octstr_create("");
  183.     octstr_format_append(part_delimiter, "%c", 'r');
  184.     octstr_format_append(part_delimiter, "%c", 'n');
  185.     octstr_format_append(part_delimiter, "%s", "--");
  186.     octstr_append(part_delimiter, boundary);
  187.     octstr_format_append(part_delimiter, "%c", 'r');
  188.     octstr_format_append(part_delimiter, "%c", 'n');
  189.     
  190.     return part_delimiter;
  191. }
  193. static Octstr *make_close_delimiter(Octstr *boundary)
  194. {
  195.     Octstr *close_delimiter;
  197.     close_delimiter = octstr_create("");
  198.     octstr_format_append(close_delimiter, "%c", 'r');
  199.     octstr_format_append(close_delimiter, "%c", 'n');
  200.     octstr_format_append(close_delimiter, "%s", "--");
  201.     octstr_append(close_delimiter, boundary);
  202.     octstr_format_append(close_delimiter, "%s", "--");
  203.     octstr_format_append(close_delimiter, "%c", 'r');
  204.     octstr_format_append(close_delimiter, "%c", 'n');
  205.     
  207.     return close_delimiter;
  208. }
  210. static List *push_headers_create(size_t content_len)
  211. {
  212.     List *push_headers;
  213.     Octstr *mos;
  215.     mos = NULL;
  216.     push_headers = http_create_empty_headers();
  217.     if (use_hardcoded)
  218.         http_header_add(push_headers, "Content-Type", "multipart/related;" 
  219.                         " boundary=asdlfkjiurwgasf; type="application/xml"");
  220.     else
  221.         http_header_add(push_headers, "Content-Type", 
  222.                         octstr_get_cstr(mos = make_multipart_value(boundary)));
  223.     if (use_headers)
  224.         http_add_basic_auth(push_headers, username, password);
  225.     add_push_application_id(&push_headers, appid_flag);
  227.     octstr_destroy(mos);
  229.     return push_headers;
  230. }
  232. static Octstr *push_content_create(void)
  233. {
  234.     Octstr *push_content, 
  235.            *wap_content;
  236.     Octstr *wap_file_content,
  237.            *pap_content,
  238.            *pap_file_content,
  239.            *bpos,
  240.            *bcos;
  242.     wap_content = NULL;
  243.     push_content = NULL;
  244.     if (use_hardcoded) {
  245.         push_content = octstr_create("rnrn"
  246.                   "--asdlfkjiurwgasfrn"
  247.                   "Content-Type: application/xmlrnrn"
  248.                   "<?xml version="1.0"?>"
  249.                   "<!DOCTYPE pap PUBLIC "-//WAPFORUM//DTD PAP//EN""
  250.                              " "http://www.wapforum.org/DTD/pap_1.0.dtd">"
  251.                   "<pap>"
  252.                         "<push-message push-id="9fjeo39jf084@pi.com""
  253.                           " deliver-before-timestamp="2002-11-01T06:45:00Z""
  254.                           " deliver-after-timestamp="2000-02-27T06:45:00Z""
  255.                           " progress-notes-requested="false">"
  256.      "<address address-value="WAPPUSH=+358408676001/"
  257.   "TYPE=PLMN@ppg.carrier.com">"
  258.                              "</address>"
  259.                              "<quality-of-service"
  260.                                " priority="low""
  261.                                " delivery-method="unconfirmed""
  262.                                " network-required="true""
  263.                                " network="GSM""
  264.                                " bearer-required="true""
  265.                                " bearer="SMS">"
  266.                              "</quality-of-service>"
  267.                         "</push-message>"
  268.                   "</pap>rnrn"         
  269.                   "--asdlfkjiurwgasfrn"
  270.                   "Content-Type: text/vnd.wap.sirnrn"
  271.                   "<?xml version="1.0"?>"
  272.                   "<!DOCTYPE si PUBLIC "-//WAPFORUM//DTD SI 1.0//EN" "
  273.                     " "http://www.wapforum.org/DTD/si.dtd">"
  274.                   "<si>"
  275.                       "<indication href="http://wap.iobox.fi""
  276.                           " si-id="1@wiral.com""
  277.                           " action="signal-high""
  278.                           " created="1999-06-25T15:23:15Z""
  279.                           " si-expires="2002-12-30T00:00:00Z">"
  280.                           "Want to test a fetch?"
  281.                       "</indication>"
  282.                    "</si>rnrn"
  283.                  "--asdlfkjiurwgasf--rnrn"
  284.                  "");
  285.     } else {
  286.         add_content_type(content_flag, &wap_content);
  287.         add_content_transfer_encoding_type(content_transfer_encoding, 
  288.                                            wap_content);
  289.         add_connection_header(connection, wap_content);
  291.         if ((wap_file_content = 
  292.                 octstr_read_file(octstr_get_cstr(content_file))) == NULL)
  293.          panic(0, "Stopping");
  295.     transfer_encode (content_transfer_encoding, wap_file_content);
  296.         octstr_append(wap_content, wap_file_content);
  297.         octstr_destroy(wap_file_content);
  299.         pap_content = octstr_format("%s", "Content-Type: application/xmlrn");
  300.         if ((pap_file_content = 
  301.                 octstr_read_file(octstr_get_cstr(pap_file))) ==  NULL)
  302.         panic(0, "Stopping");
  303.         octstr_append(pap_content, pap_file_content);
  304.         octstr_destroy(pap_file_content);
  306.         if (wap_content == NULL || pap_content == NULL)
  307.         panic(0, "Cannot open the push content files");
  309.         push_content = octstr_create("");
  310.         octstr_append(push_content, 
  311.             bpos = make_part_delimiter(octstr_imm(boundary)));
  312.         octstr_append(push_content, pap_content);
  313.         octstr_append(push_content, bpos);
  314.         octstr_destroy(bpos);
  315.         octstr_append(push_content, wap_content);
  316.         octstr_append(push_content, 
  317.             bcos = make_close_delimiter(octstr_imm(boundary)));
  318.         octstr_destroy(bcos);
  319.         octstr_destroy(pap_content);
  320.         octstr_destroy(wap_content);
  321.     }
  323.     return push_content;
  324. }
  326. static void make_url(Octstr **url)
  327. {
  328.     if (use_config && !use_headers) {
  329.         octstr_append(*url, octstr_imm("?username="));
  330.         octstr_append(*url, username ? username : octstr_imm("default"));
  331.         octstr_append(*url, octstr_imm("&password="));
  332.         octstr_append(*url, password ? password: octstr_imm("default"));
  333.     }
  334. }
  336. static void start_push(HTTPCaller *caller, long i)   
  337. {
  338.     List *push_headers;
  339.     Octstr *push_content;
  340.     long *id;
  341.     
  342.     push_content = push_content_create();
  343.     push_headers = push_headers_create(octstr_len(push_content));
  344.     if (verbose) {
  345.        debug("test.ppg", 0, "we have push content");
  346.        octstr_dump(push_content, 0);
  347.        debug("test.ppg", 0, "and headers");
  348.        http_header_dump(push_headers);
  349.     }
  351.     id = gw_malloc(sizeof(long));
  352.     *id = i;
  353.     make_url(&push_url);
  354.     debug("test.ppg", 0, "TEST_PPG: starting to push job %ld", i);
  355.     http_start_request(caller, HTTP_METHOD_POST, push_url, push_headers, 
  356.                        push_content, 0, id, ssl_client_certkey_file);
  357.     debug("test.ppg", 0, "push done");
  358.     octstr_destroy(push_content);
  359.     http_destroy_headers(push_headers);
  360. }
  362. /*
  363.  * Try log in defined number of times, when got response 401 and authentica-
  364.  * tion info is in headers.
  365.  */
  366. static int receive_push_reply(HTTPCaller *caller)
  367. {
  368.     void *id;
  369.     long *trid;
  370.     int http_status,
  371.         tries;
  372.     List *reply_headers;
  373.     Octstr *final_url,
  374.            *auth_url,
  375.            *reply_body,
  376.            *os,
  377.            *push_content,
  378.            *auth_reply_body;
  379.     WAPEvent *e;
  380.     List *retry_headers;
  381.     
  382.     http_status = HTTP_UNAUTHORIZED;
  383.     tries = 0;
  385.     id = http_receive_result(caller, &http_status, &final_url, &reply_headers,
  386.                              &reply_body);
  388.     if (id == NULL || http_status == -1 || final_url == NULL) {
  389.         error(0, "push failed, no reason found");
  390.         goto push_failed;
  391.     }
  393.     while (use_headers && http_status == HTTP_UNAUTHORIZED && tries < retries) {
  394.         debug("test.ppg", 0, "try number %d", tries);
  395.         debug("test.ppg", 0, "authentication failure, get a challenge");
  396.         http_destroy_headers(reply_headers);
  397.         push_content = push_content_create();
  398.         retry_headers = push_headers_create(octstr_len(push_content));
  399.         http_add_basic_auth(retry_headers, username, password);
  400.         trid = gw_malloc(sizeof(long));
  401.         *trid = tries;
  402.         http_start_request(caller, HTTP_METHOD_POST, final_url, retry_headers, 
  403.                            push_content, 0, trid, NULL);
  404.         debug("test.ppg ", 0, "TEST_PPG: doing response to %s", 
  405.               octstr_get_cstr(final_url));
  407.         octstr_destroy(push_content);
  408.         http_destroy_headers(retry_headers);
  409.         
  410.         trid = http_receive_result(caller, &http_status, &auth_url, 
  411.                                    &reply_headers, &auth_reply_body);
  413.         if (trid == NULL || http_status == -1 || auth_url == NULL) {
  414.             error(0, "unable to send authorisation, no reason found");
  415.             goto push_failed;
  416.         }   
  418.         debug("test.ppg", 0, "TEST_PPG: send authentication to %s, retry %ld", 
  419.                octstr_get_cstr(auth_url), *(long *) trid);
  420.         gw_free(trid);
  421.         octstr_destroy(auth_reply_body);
  422.         octstr_destroy(auth_url);
  423.         ++tries;
  424.     }
  426.     if (http_status == HTTP_NOT_FOUND) {
  427.         error(0, "push failed, service not found");
  428.         goto push_failed;
  429.     }
  431.     if (http_status == HTTP_FORBIDDEN) {
  432.         error(0, "push failed, service forbidden");
  433.         goto push_failed;
  434.     }
  436.     if (http_status == HTTP_UNAUTHORIZED) {
  437.         if (use_headers)
  438.             error(0, "tried %ld times, stopping", retries);
  439.         else
  440.         error(0, "push failed, authorisation failure");
  441.         goto push_failed;
  442.     }
  443.         
  444.     debug("test.ppg", 0, "TEST_PPG: push %ld done: reply from,  %s", 
  445.           *(long *) id, octstr_get_cstr(final_url));
  446.     gw_free(id);
  447.     octstr_destroy(final_url);
  449.     if (verbose)
  450.         debug("test.ppg", 0, "TEST_PPG: reply headers were");
  452.     while ((os = list_extract_first(reply_headers)) != NULL) {
  453.         if (verbose)
  454.             octstr_dump(os, 0); 
  455.         octstr_destroy(os);
  456.     }
  458.     if (verbose) {
  459.         debug("test.ppg", 0, "TEST_PPG: reply body was");
  460.         octstr_dump(reply_body, 0);
  461.     }
  463.     e = NULL;
  464.     if (pap_compile(reply_body, &e) < 0) {
  465.         warning(0, "TEST_PPG: receive_push_reply: cannot compile pap message");
  466.         goto parse_error;
  467.     }
  469.     switch (e->type) {
  470.         case Push_Response:
  471.         debug("test.ppg", 0, "TEST_PPG: and type push response");
  472.     break;
  474.         case Bad_Message_Response:
  475.         debug("test.ppg", 0, "TEST_PPG: and type bad message response");
  476.         break;
  478.         default:
  479.             warning(0, "TEST_PPG: unknown event received from %s", 
  480.                     octstr_get_cstr(final_url));
  481.         break;
  482.     }
  484.     octstr_destroy(reply_body);
  485.     wap_event_destroy(e);
  486.     http_destroy_headers(reply_headers);
  487.     return 0;
  489. push_failed:
  490.     gw_free(id);
  491.     octstr_destroy(final_url);
  492.     octstr_destroy(reply_body);
  493.     http_destroy_headers(reply_headers);
  494.     return -1;
  496. parse_error:
  497.     octstr_destroy(reply_body);
  498.     http_destroy_headers(reply_headers);
  499.     wap_event_destroy(e);
  500.     return -1;
  501. }
  503. static void push_thread(void *arg)
  504. {
  505.     HTTPCaller *caller;
  506.     long succeeded, failed, in_queue;
  507.     unsigned long i;
  509.     caller = arg;
  510.     succeeded = 0;
  511.     failed = 0;   
  512.     in_queue = 0;
  513.     i = 0;
  515.     for (;;) {
  516.         while (in_queue < MAX_IN_QUEUE) {
  517.         i = counter_increase(counter);
  518.             if (i >= max_pushes)
  519.             goto receive_rest;
  520.             start_push(caller, i);
  521.             if (wait)
  522.                 gwthread_sleep(wait_seconds);
  523.             ++in_queue;
  524.         }
  526.         while (in_queue >= MAX_IN_QUEUE) {
  527.         if (receive_push_reply(caller) == -1)
  528.             ++failed;
  529.             else
  530.             ++succeeded;
  531.             --in_queue;
  532.         }
  533.     }
  535. receive_rest:
  536.     while (in_queue > 0) {
  537.         if (receive_push_reply(caller) == -1)
  538.         ++failed;
  539.         else
  540.         ++succeeded;
  541.         --in_queue;
  542.     }
  544.     http_caller_destroy(caller);
  545.     info(0, "TEST_PPG: In thread %ld %ld succeeded, %ld failed", 
  546.          (long) gwthread_self(), succeeded, failed);
  547. }
  549. static void help(void) 
  550. {
  551.     info(0, "Usage: test_ppg [options] push_url [content_file pap_file]");
  552.     info(0, "      or");
  553.     info(0, "Usage: test_ppg [options] [conf_file]");
  554.     info(0, "Implements push initiator for wap push. Push services are ");
  555.     info(0, "located in push_url, push content in the file content file.");
  556.     info(0, "File pap_file contains pap control document that controls");
  557.     info(0, "pushing");
  558.     info(0, "If option -H is not used, command line has either three or one");
  559.     info(0, "arguments:");
  560.     info(0, "      a) the url of the push proxy gateway");
  561.     info(0, "      b) a file containing the content to be pushed");
  562.     info(0, "      c) a pap document controlling pushing");
  563.     info(0, "     or");
  564.     info(0, "      a) a test configuration file, containing all these");
  565.     info(0, "Option -H cannot be used with a configuration file. If it is");
  566.     info(0, "used, the push url is the only argument.");
  567.     info(0, "Options are:");
  568.     info(0, "-h");
  569.     info(0, "print this info");
  570.     info(0, "-c content qualifier");
  571.     info(0, "Define content type of the push content. Wml, multipart, nil,"); 
  572.     info(0, "scrap, sl, and si accepted. Si is default, nil (no content"); 
  573.     info(0, " type at all) and scrap (random string) are used for debugging");
  574.     info(0, "-a application id");
  575.     info(0, "Define the client application that will handle the push. Any,"); 
  576.     info(0, "ua, mms, nil and scrap accepted, default ua.");
  577.     info(0, "-b");
  578.     info(0, "If true, send username/password in headers. Default false");
  579.     info(0, "-v number");
  580.     info(0, "    Set log level for stderr logging. Default 0 (debug)");
  581.     info(0, "-q");
  582.     info(0, "    Do not print debugging information");
  583.     info(0, "Default: print it");
  584.     info(0, "-r number");
  585.     info(0, "    Make `number' requests. Default one request");
  586.     info(0, "-i seconds");
  587.     info(0, "    Wait 'seconds' seconds between pushes. Default: do not wait");
  588.     info(0, "-e transfer encoding");
  589.     info(0, "    use transfer encoding to send push contents.");
  590.     info(0, "    Currently supported is base64.");
  591.     info(0, "-k connection header");
  592.     info(0, "Use the connection header. Keep-alive and close accepted,");
  593.     info(0, "default close");
  594.     info(0, "-H");
  595.     info(0, "Use hardcoded MIME message, containing a pap control document.");
  596.     info(0, "In addition, use hardcoded username/password in headers (if ");
  597.     info(0, "flag -b is set, too");
  598.     info(0, "Default: read components from files");
  599.     info(0, "-t");
  600.     info(0, "number of threads, maximum 1024, default 1");
  601. }
  603. int main(int argc, char **argv)
  604. {
  605.     int opt,
  606.         num_threads;
  607.     time_t start,
  608.            end;
  609.     double run_time;
  610.     long threads[MAX_THREADS];
  611.     long i;
  612.     Octstr *fos;
  614.     gwlib_init();
  615.     num_threads = 1;
  617.     while ((opt = getopt(argc, argv, "Hhbv:qr:t:c:a:i:e:k:")) != EOF) {
  618.         switch(opt) {
  619.     case 'v':
  620.         log_set_output_level(atoi(optarg));
  621.     break;
  623.     case 'q': 
  624.         verbose = 0;
  625.     break;  
  627.     case 'r':
  628.         max_pushes = atoi(optarg);      
  629.     break; 
  630.             
  631.     case 'i': 
  632.         wait = 1;
  633.             wait_seconds = atof(optarg);
  634.     break;
  636.         case 't': 
  637.         num_threads = atoi(optarg);
  638.             if (num_threads > MAX_THREADS)
  639.         num_threads = MAX_THREADS;
  640.     break;
  642.     case 'H': 
  643.         use_hardcoded = 1;
  644.     break;
  646.     case 'c':
  647.         content_flag = octstr_create(optarg);
  648.             if (octstr_compare(content_flag, octstr_imm("wml")) != 0 && 
  649.                     octstr_compare(content_flag, octstr_imm("si")) != 0 &&
  650.                     octstr_compare(content_flag, octstr_imm("sl")) != 0 &&
  651.                     octstr_compare(content_flag, octstr_imm("nil")) != 0 &&
  652.                     octstr_compare(content_flag, octstr_imm("mms")) != 0 &&
  653.                     octstr_compare(content_flag, octstr_imm("scrap")) != 0 &&
  654.                     octstr_compare(content_flag, 
  655.                     octstr_imm("multipart")) != 0){
  656.         octstr_destroy(content_flag);
  657.         error(0, "TEST_PPG: Content type not known");
  658.         help();
  659.                 exit(1);
  660.             }
  661.     break;
  663.     case 'a':
  664.         appid_flag = octstr_create(optarg);
  665.             if (octstr_compare(appid_flag, octstr_imm("any")) != 0 && 
  666.                     octstr_compare(appid_flag, octstr_imm("ua")) != 0 &&
  667.                     octstr_compare(appid_flag, octstr_imm("mms")) != 0 &&
  668.                     octstr_compare(appid_flag, octstr_imm("nil")) != 0 &&
  669.                     octstr_compare(appid_flag, octstr_imm("scrap")) != 0) {
  670.         octstr_destroy(appid_flag);
  671.         error(0, "TEST_PPG: Push application id not known");
  672.         help();
  673.                 exit(1);
  674.            }
  675.     break;
  677.     case 'e':
  678.     content_transfer_encoding = octstr_create(optarg);
  679.             if (octstr_compare(content_transfer_encoding, 
  680.                                octstr_imm("base64")) != 0) {
  681.         octstr_destroy(content_transfer_encoding);
  682.         error(0, "TEST_PPG: unknown content transfer" 
  683.                       " encoding "%s"",
  684.           octstr_get_cstr(content_transfer_encoding));
  685.         help();
  686.                 exit(1);
  687.     }
  688.     break;
  690.     case 'k':
  691.         connection = octstr_create(optarg);
  692.             if (octstr_compare(connection, octstr_imm("close")) != 0 && 
  693.                     octstr_compare(connection, 
  694.                         octstr_imm("keep-alive")) != 0) {
  695.         octstr_destroy(connection);
  696.         error(0, "TEST_PPG: Connection-header unacceptable");
  697.         help();
  698.                 exit(1);
  699.            }
  700.     break;
  702.     case 'h':
  703.         help();
  704.             exit(1);
  706.     case 'b':
  707.         use_headers = 1;
  708.     break;
  710.     case '?':
  711.     default:
  712.         error(0, "TEST_PPG: Invalid option %c", opt);
  713.             help();
  714.             error(0, "Stopping");
  715.             exit(1);
  716.         }
  717.     }
  719.     if (optind == argc) {
  720.         help();
  721.         exit(1);
  722.     }
  723.     
  724.     push_data = argv + optind;
  725.     num_urls = argc - optind;
  727.     if (content_flag == NULL)
  728.         content_flag = octstr_imm("si");
  730.     if (appid_flag == NULL)
  731.         appid_flag = octstr_imm("ua");
  733.     if (use_hardcoded) {
  734.         username = octstr_imm("troo");
  735.         password = octstr_imm("far");
  736.     }
  738.     if (push_data[0] == NULL) {
  739.         error(0, "No ppg address or config file, stopping");
  740.         exit(1);
  741.     }
  742.            
  743.     use_config = 0;
  744.     if (!use_hardcoded) {
  745.         if (push_data[1] == NULL) {
  746.             info(0, "a configuration file input assumed");
  747.             read_test_ppg_config(fos = octstr_format("%s", push_data[0]));
  748.             octstr_destroy(fos);
  749.             use_config = 1;
  750.         }
  751.     }
  753.     if (!use_config)
  754.         push_url = octstr_format("%s", push_data[0]);
  756.     if (!use_hardcoded && !use_config && push_data[1] != NULL) {
  757.         if (push_data[2] == NULL) {
  758.         error(0, "no pap control document, stopping");
  759.             exit(1);
  760.         } else {
  761.            info(0, "an input without a configuration file assumed");
  762.            content_file = octstr_create(push_data[1]);
  763.            pap_file = octstr_create(push_data[2]);
  764.            debug("test.ppg", 0, "using %s as a content file", push_data[1]);
  765.            debug("test.ppg", 0, "using %s as a control file", push_data[2]);
  766.         }
  767.     }
  769.     boundary = "asdlfkjiurwghasf";
  770.     counter = counter_create();
  772.     time(&start);
  773.     if (num_threads == 0)
  774.         push_thread(http_caller_create());
  775.     else {
  776.         for (i = 0; i < num_threads; ++i)
  777.         threads[i] = gwthread_create(push_thread, http_caller_create());
  778.     for (i = 0; i < num_threads; ++i)
  779.         gwthread_join(threads[i]);
  780.     }
  781.     time(&end);
  782.     run_time = difftime(end, start);
  783.     info(0, "TEST_PPG: %ld requests in %f seconds, %f requests per second",
  784.          max_pushes, run_time, max_pushes / run_time);
  786.     octstr_destroy(content_flag);
  787.     octstr_destroy(appid_flag);
  788.     octstr_destroy(content_file);
  789.     octstr_destroy(pap_file);
  790.     octstr_destroy(ssl_client_certkey_file);
  791.     octstr_destroy(username);
  792.     octstr_destroy(password);
  793.     octstr_destroy(push_url);
  794.     octstr_destroy(connection);
  795.     counter_destroy(counter);
  796.     gwlib_shutdown();
  798.     exit(0);
  799. }