asiohiper.cpp: Updated for curl coding standards (193ba7b4) · Commits · CYBER - Cyber Security / TS 103 523 MSP / TLMSP / TLMSP curl

docs/examples/asiohiper.cpp

+73 −58

Original line number	Diff line number	Diff line
		@@ -108,21 +108,39 @@ static void mcode_or_die(const char *where, CURLMcode code)
		const char *s;
		switch(code)
		{
		case CURLM_CALL_MULTI_PERFORM: s="CURLM_CALL_MULTI_PERFORM"; break;
		case CURLM_BAD_HANDLE: s="CURLM_BAD_HANDLE"; break;
		case CURLM_BAD_EASY_HANDLE: s="CURLM_BAD_EASY_HANDLE"; break;
		case CURLM_OUT_OF_MEMORY: s="CURLM_OUT_OF_MEMORY"; break;
		case CURLM_INTERNAL_ERROR: s="CURLM_INTERNAL_ERROR"; break;
		case CURLM_UNKNOWN_OPTION: s="CURLM_UNKNOWN_OPTION"; break;
		case CURLM_LAST: s="CURLM_LAST"; break;
		default: s="CURLM_unknown";
		case CURLM_CALL_MULTI_PERFORM:
		s = "CURLM_CALL_MULTI_PERFORM";
		break;
		case CURLM_BAD_SOCKET: s="CURLM_BAD_SOCKET";
		case CURLM_BAD_HANDLE:
		s = "CURLM_BAD_HANDLE";
		break;
		case CURLM_BAD_EASY_HANDLE:
		s = "CURLM_BAD_EASY_HANDLE";
		break;
		case CURLM_OUT_OF_MEMORY:
		s = "CURLM_OUT_OF_MEMORY";
		break;
		case CURLM_INTERNAL_ERROR:
		s = "CURLM_INTERNAL_ERROR";
		break;
		case CURLM_UNKNOWN_OPTION:
		s = "CURLM_UNKNOWN_OPTION";
		break;
		case CURLM_LAST:
		s = "CURLM_LAST";
		break;
		default:
		s = "CURLM_unknown";
		break;
		case CURLM_BAD_SOCKET:
		s = "CURLM_BAD_SOCKET";
		fprintf(MSG_OUT, "\nERROR: %s returns %s", where, s);
		/* ignore this error */
		return;
		}

		fprintf(MSG_OUT, "\nERROR: %s returns %s", where, s);

		exit(code);
		}
		}
		@@ -157,12 +175,14 @@ static void check_multi_info(GlobalInfo *g)
		}

		/* Called by asio when there is an action on a socket */
		static void event_cb(GlobalInfo * g, boost::asio::ip::tcp::socket * tcp_socket, int action)
		static void event_cb(GlobalInfo g, boost::asio::ip::tcp::socket tcp_socket,
		int action)
		{
		fprintf(MSG_OUT, "\nevent_cb: action=%d", action);

		CURLMcode rc;
		rc = curl_multi_socket_action(g->multi, tcp_socket->native_handle(), action, &g->still_running);
		rc = curl_multi_socket_action(g->multi, tcp_socket->native_handle(), action,
		&g->still_running);

		mcode_or_die("event_cb: curl_multi_socket_action", rc);
		check_multi_info(g);
		@@ -209,6 +229,7 @@ static void setsock(int fdp, curl_socket_t s, CURLe, int act, GlobalInfo*g)
		if(it == socket_map.end())
		{
		fprintf(MSG_OUT, "\nsocket %d is a c-ares socket, ignoring", s);

		return;
		}

		@@ -221,39 +242,31 @@ static void setsock(int fdp, curl_socket_t s, CURLe, int act, GlobalInfo*g)
		fprintf(MSG_OUT, "\nwatching for socket to become readable");

		tcp_socket->async_read_some(boost::asio::null_buffers(),
		boost::bind(&event_cb, g,
		tcp_socket,
		act));
		boost::bind(&event_cb, g, tcp_socket, act));
		}
		else if (act == CURL_POLL_OUT)
		{
		fprintf(MSG_OUT, "\nwatching for socket to become writable");

		tcp_socket->async_write_some(boost::asio::null_buffers(),
		boost::bind(&event_cb, g,
		tcp_socket,
		act));
		boost::bind(&event_cb, g, tcp_socket, act));
		}
		else if(act == CURL_POLL_INOUT)
		{
		fprintf(MSG_OUT, "\nwatching for socket to become readable & writable");

		tcp_socket->async_read_some(boost::asio::null_buffers(),
		boost::bind(&event_cb, g,
		tcp_socket,
		act));
		boost::bind(&event_cb, g, tcp_socket, act));

		tcp_socket->async_write_some(boost::asio::null_buffers(),
		boost::bind(&event_cb, g,
		tcp_socket,
		act));
		boost::bind(&event_cb, g, tcp_socket, act));
		}
		}


		static void addsock(curl_socket_t s, CURL easy, int action, GlobalInfo g)
		{
		int fdp = (int )calloc(sizeof(int), 1); /* fdp is used to store current action */
		/* fdp is used to store current action */
		int fdp = (int ) calloc(sizeof(int), 1);

		setsock(fdp, s, easy, action, g);
		curl_multi_assign(g->multi, s, fdp);
		@@ -291,10 +304,10 @@ static int sock_cb(CURL e, curl_socket_t s, int what, void cbp, void *sockp)
		setsock(actionp, s, e, what, g);
		}
		}

		return 0;
		}


		/* CURLOPT_WRITEFUNCTION */
		static size_t write_cb(void ptr, size_t size, size_t nmemb, void data)
		{
		@@ -312,12 +325,12 @@ static size_t write_cb(void ptr, size_t size, size_t nmemb, void data)
		return written;
		}


		/* CURLOPT_PROGRESSFUNCTION */
		static int prog_cb(void *p, double dltotal, double dlnow, double ult,
		double uln)
		{
		ConnInfo conn = (ConnInfo )p;

		(void)ult;
		(void)uln;

		@@ -328,8 +341,7 @@ static int prog_cb (void *p, double dltotal, double dlnow, double ult,
		}

		/* CURLOPT_OPENSOCKETFUNCTION */
		static curl_socket_t opensocket(void *clientp,
		curlsocktype purpose,
		static curl_socket_t opensocket(void *clientp, curlsocktype purpose,
		struct curl_sockaddr *address)
		{
		fprintf(MSG_OUT, "\nopensocket :");
		@@ -348,7 +360,7 @@ static curl_socket_t opensocket(void *clientp,

		if(ec)
		{
		//An error occurred
		/* An error occurred */
		std::cout << std::endl << "Couldn't open socket [" << ec << "][" << ec.message() << "]";
		fprintf(MSG_OUT, "\nERROR: Returning CURL_SOCKET_BAD to signal error");
		}
		@@ -392,12 +404,13 @@ static void new_conn(char url, GlobalInfo g )
		conn->error[0] = '\0';

		conn->easy = curl_easy_init();

		if(!conn->easy)
		{
		fprintf(MSG_OUT, "\ncurl_easy_init() failed, exiting!");

		exit(2);
		}

		conn->global = g;
		conn->url = strdup(url);
		curl_easy_setopt(conn->easy, CURLOPT_URL, conn->url);
		@@ -431,6 +444,7 @@ int main(int argc, char **argv)
		{
		GlobalInfo g;
		CURLMcode rc;

		(void)argc;
		(void)argv;

		@@ -450,5 +464,6 @@ int main(int argc, char **argv)
		curl_multi_cleanup(g.multi);

		fprintf(MSG_OUT, "\ndone.\n");

		return 0;
		}