Websocket - WebSocket client library

Users » samux » Code » Websocket

WebSocket client library

Websocket.cpp@9:9fa055ed54b4, 2011-08-24 (annotated)

Committer:: samux
Date:: Wed Aug 24 15:22:52 2011 +0000
Revision:: 9:9fa055ed54b4
Parent:: 8:ecd914cffdf6
Child:: 10:4f02275c34ee

Who changed what in which revision?

User	Revision	Line number	New contents of line
samux	0:21fe3b05249f	1	#include "Websocket.h"
samux	6:01a1eb7c0145	2	#include <string>
samux	0:21fe3b05249f	3
samux	6:01a1eb7c0145	4	Websocket::Websocket(char * url, Wifly * wifi) {
samux	9:9fa055ed54b4	5	this->wifi = wifi;
samux	9:9fa055ed54b4	6	wifi_use = true;
samux	9:9fa055ed54b4	7	eth_use = false;
samux	9:9fa055ed54b4	8	response_server_eth = false;
samux	9:9fa055ed54b4	9	fillFields(wifi_use, url);
samux	6:01a1eb7c0145	10	}
samux	6:01a1eb7c0145	11
samux	6:01a1eb7c0145	12	void Websocket::fillFields(bool wifi, char * url) {
samux	0:21fe3b05249f	13	char *res = NULL;
samux	6:01a1eb7c0145	14	char *res1 = NULL;
samux	6:01a1eb7c0145	15
samux	0:21fe3b05249f	16	char buf[30];
samux	0:21fe3b05249f	17	strcpy(buf, url);
samux	6:01a1eb7c0145	18
samux	0:21fe3b05249f	19	res = strtok(buf, ":");
samux	9:9fa055ed54b4	20	if (strcmp(res, "ws")) {
samux	0:21fe3b05249f	21	this->ip_domain = NULL;
samux	0:21fe3b05249f	22	this->path = NULL;
samux	6:01a1eb7c0145	23	printf("\r\nFormat error: please use: \"ws://ip-or-domain[:port]/path\"\r\n\r\n");
samux	9:9fa055ed54b4	24	} else {
samux	6:01a1eb7c0145	25	//ip_domain and port
samux	6:01a1eb7c0145	26	res = strtok(NULL, "/");
samux	6:01a1eb7c0145	27
samux	6:01a1eb7c0145	28	//path
samux	6:01a1eb7c0145	29	res1 = strtok(NULL, " ");
samux	6:01a1eb7c0145	30	if (res1 != NULL) {
samux	6:01a1eb7c0145	31	path = (char ) malloc (sizeof(char) strlen(res1));
samux	6:01a1eb7c0145	32	strcpy(this->path, res1);
samux	6:01a1eb7c0145	33	}
samux	6:01a1eb7c0145	34
samux	6:01a1eb7c0145	35	//ip_domain
samux	6:01a1eb7c0145	36	res = strtok(res, ":");
samux	9:9fa055ed54b4	37
samux	7:b15978708360	38	//port
samux	7:b15978708360	39	res1 = strtok(NULL, " ");
samux	7:b15978708360	40	//port
samux	7:b15978708360	41	if (res1 != NULL) {
samux	7:b15978708360	42	port = (char ) malloc (sizeof(char) strlen(res1));
samux	7:b15978708360	43	strcpy(this->port, res1);
samux	7:b15978708360	44	} else {
samux	7:b15978708360	45	port = (char ) malloc (sizeof(char) 3);
samux	7:b15978708360	46	strcpy(this->port, "80");
samux	7:b15978708360	47	}
samux	9:9fa055ed54b4	48
samux	6:01a1eb7c0145	49	if (res != NULL) {
samux	0:21fe3b05249f	50	ip_domain = (char ) malloc (sizeof(char) strlen(res));
samux	0:21fe3b05249f	51	strcpy(this->ip_domain, res);
samux	6:01a1eb7c0145	52	//if we use ethernet, we must decode ip address or use dnsresolver
samux	6:01a1eb7c0145	53	if (!wifi) {
samux	6:01a1eb7c0145	54	strcpy(buf, res);
samux	6:01a1eb7c0145	55	//we try to decode the ip address
samux	6:01a1eb7c0145	56	if (buf[0] >= '0' && buf[0] <= '9') {
samux	6:01a1eb7c0145	57	res = strtok(buf, ".");
samux	6:01a1eb7c0145	58	int i = 0;
samux	6:01a1eb7c0145	59	int ip[4];
samux	6:01a1eb7c0145	60	while (res != NULL) {
samux	6:01a1eb7c0145	61	ip[i] = atoi(res);
samux	6:01a1eb7c0145	62	res = strtok(NULL, ".");
samux	6:01a1eb7c0145	63	i++;
samux	6:01a1eb7c0145	64	}
samux	6:01a1eb7c0145	65	server_ip = IpAddr(ip[0], ip[1], ip[2], ip[3]);
samux	6:01a1eb7c0145	66	printf("server=%i.%i.%i.%i\n",server_ip[0],server_ip[1],server_ip[2],server_ip[3]);
samux	6:01a1eb7c0145	67	}
samux	6:01a1eb7c0145	68	//we must use dnsresolver to find the ip address
samux	6:01a1eb7c0145	69	else {
samux	6:01a1eb7c0145	70	DNSResolver dr;
samux	6:01a1eb7c0145	71	server_ip = dr.resolveName(buf);
samux	6:01a1eb7c0145	72	printf("server=%i.%i.%i.%i\n",server_ip[0],server_ip[1],server_ip[2],server_ip[3]);
samux	6:01a1eb7c0145	73	}
samux	6:01a1eb7c0145	74	}
samux	0:21fe3b05249f	75	}
samux	0:21fe3b05249f	76	}
samux	0:21fe3b05249f	77	}
samux	0:21fe3b05249f	78
samux	6:01a1eb7c0145	79
samux	6:01a1eb7c0145	80	Websocket::Websocket(char * url) {
samux	9:9fa055ed54b4	81	wifi_use = false;
samux	9:9fa055ed54b4	82	eth_use = true;
samux	9:9fa055ed54b4	83	eth_writeable = false;
samux	9:9fa055ed54b4	84	eth_readable = false;
samux	9:9fa055ed54b4	85	eth_connected = false;
samux	9:9fa055ed54b4	86	response_server_eth = false;
samux	9:9fa055ed54b4	87	fillFields(wifi_use, url);
samux	6:01a1eb7c0145	88	}
samux	6:01a1eb7c0145	89
samux	6:01a1eb7c0145	90
samux	6:01a1eb7c0145	91
samux	6:01a1eb7c0145	92
samux	6:01a1eb7c0145	93	bool Websocket::connect() {
samux	0:21fe3b05249f	94	char cmd[50];
samux	6:01a1eb7c0145	95	if ( wifi_use ) {
samux	6:01a1eb7c0145	96	wifi->Send("exit\r", "NO");
samux	6:01a1eb7c0145	97	//enter in cmd mode
samux	6:01a1eb7c0145	98	while (!wifi->Send("$$$", "CMD")) {
samux	6:01a1eb7c0145	99	printf("cannot enter in CMD mode\r\n");
samux	6:01a1eb7c0145	100	wifi->exit();
samux	6:01a1eb7c0145	101	}
samux	6:01a1eb7c0145	102
samux	6:01a1eb7c0145	103
samux	6:01a1eb7c0145	104	//open the connection
samux	6:01a1eb7c0145	105	sprintf(cmd, "open %s %s\r\n", ip_domain, port);
samux	6:01a1eb7c0145	106	if (!wifi->Send(cmd, "OPEN")) {
samux	6:01a1eb7c0145	107	printf("Websocket::connect cannot open\r\n");
samux	6:01a1eb7c0145	108	return false;
samux	6:01a1eb7c0145	109	}
samux	6:01a1eb7c0145	110
samux	6:01a1eb7c0145	111
samux	6:01a1eb7c0145	112	//send websocket HTTP header
samux	6:01a1eb7c0145	113	sprintf(cmd, "GET /%s HTTP/1.1\r\n", path);
samux	6:01a1eb7c0145	114	wifi->Send(cmd, "NO");
samux	6:01a1eb7c0145	115
samux	6:01a1eb7c0145	116	sprintf(cmd, "Host: %s:%s\r\n", ip_domain, port);
samux	6:01a1eb7c0145	117	wifi->Send(cmd, "NO");
samux	6:01a1eb7c0145	118
samux	6:01a1eb7c0145	119	wifi->Send("Upgrade: WebSocket\r\n", "NO");
samux	6:01a1eb7c0145	120
samux	6:01a1eb7c0145	121	sprintf(cmd, "Origin: http:%s:%s\r\n", ip_domain, port);
samux	6:01a1eb7c0145	122	wifi->Send(cmd, "NO");
samux	6:01a1eb7c0145	123
samux	6:01a1eb7c0145	124
samux	6:01a1eb7c0145	125	wifi->Send("Connection: Upgrade\r\n", "NO");
samux	6:01a1eb7c0145	126	wifi->Send("Sec-WebSocket-Key1: 4 @1 46546xW%0l 1 5\r\n", "NO");
samux	6:01a1eb7c0145	127	wifi->Send("Sec-WebSocket-key2: 12998 5 Y3 1 .P00\r\n\r\n", "NO");
samux	6:01a1eb7c0145	128	if (!wifi->Send("^n:ds[4U", "8jKS'y:G*Co,Wxa-"))
samux	6:01a1eb7c0145	129	return false;
samux	6:01a1eb7c0145	130
samux	6:01a1eb7c0145	131	printf("\r\nip_domain: %s\r\npath: /%s\r\nport: %s\r\n\r\n",this->ip_domain, this->path, this->port);
samux	6:01a1eb7c0145	132	return true;
samux	6:01a1eb7c0145	133	} else if ( eth_use ) {
samux	6:01a1eb7c0145	134	int i = 0;
samux	6:01a1eb7c0145	135
samux	6:01a1eb7c0145	136	EthernetErr ethErr = eth.setup();
samux	6:01a1eb7c0145	137	if (ethErr) {
samux	6:01a1eb7c0145	138	printf("Error %d in setup.\r\n", ethErr);
samux	6:01a1eb7c0145	139	return false;
samux	6:01a1eb7c0145	140	}
samux	6:01a1eb7c0145	141
samux	6:01a1eb7c0145	142	printf("setup OK\r\n");
samux	6:01a1eb7c0145	143
samux	6:01a1eb7c0145	144	IpAddr ipt = eth.getIp();
samux	6:01a1eb7c0145	145	printf("mbed IP Address is %d.%d.%d.%d\r\n", ipt[0], ipt[1], ipt[2], ipt[3]);
samux	6:01a1eb7c0145	146
samux	6:01a1eb7c0145	147	sock.setOnEvent(this, &Websocket::onTCPSocketEvent);
samux	6:01a1eb7c0145	148	Host server (server_ip, atoi(port));
samux	6:01a1eb7c0145	149	TCPSocketErr bindErr = sock.connect(server);
samux	6:01a1eb7c0145	150	if (bindErr) {
samux	6:01a1eb7c0145	151	printf("binderr: %d", bindErr);
samux	6:01a1eb7c0145	152	return false;
samux	6:01a1eb7c0145	153	}
samux	6:01a1eb7c0145	154
samux	6:01a1eb7c0145	155
samux	6:01a1eb7c0145	156	Timer tmr;
samux	6:01a1eb7c0145	157	tmr.start();
samux	0:21fe3b05249f	158
samux	6:01a1eb7c0145	159	while (true) {
samux	6:01a1eb7c0145	160	Net::poll();
samux	6:01a1eb7c0145	161	if (tmr.read() > 0.1) {
samux	6:01a1eb7c0145	162	tmr.reset();
samux	6:01a1eb7c0145	163	if (eth_connected) {
samux	6:01a1eb7c0145	164	switch (i) {
samux	6:01a1eb7c0145	165	case 0:
samux	6:01a1eb7c0145	166	sprintf(cmd, "GET /%s HTTP/1.1\r\n", path);
samux	6:01a1eb7c0145	167	sock.send(cmd, strlen(cmd));
samux	6:01a1eb7c0145	168	i++;
samux	6:01a1eb7c0145	169	break;
samux	6:01a1eb7c0145	170	case 1:
samux	6:01a1eb7c0145	171	sprintf(cmd, "Host: %s:%s\r\n", ip_domain, port);
samux	6:01a1eb7c0145	172	sock.send(cmd, strlen(cmd));
samux	6:01a1eb7c0145	173	i++;
samux	6:01a1eb7c0145	174	break;
samux	6:01a1eb7c0145	175	case 2:
samux	6:01a1eb7c0145	176	sprintf(cmd, "Upgrade: WebSocket\r\n");
samux	6:01a1eb7c0145	177	sock.send(cmd, strlen(cmd));
samux	6:01a1eb7c0145	178	i++;
samux	6:01a1eb7c0145	179	break;
samux	6:01a1eb7c0145	180	case 3:
samux	6:01a1eb7c0145	181	sprintf(cmd, "Origin: http:%s:%s\r\n", ip_domain, port);
samux	6:01a1eb7c0145	182	sock.send(cmd, strlen(cmd));
samux	6:01a1eb7c0145	183	i++;
samux	6:01a1eb7c0145	184	break;
samux	6:01a1eb7c0145	185	case 4:
samux	6:01a1eb7c0145	186	sprintf(cmd, "Connection: Upgrade\r\n");
samux	6:01a1eb7c0145	187	sock.send(cmd, strlen(cmd));
samux	6:01a1eb7c0145	188	i++;
samux	6:01a1eb7c0145	189	break;
samux	6:01a1eb7c0145	190	case 5:
samux	6:01a1eb7c0145	191	sprintf(cmd, "Sec-WebSocket-Key1: 4 @1 46546xW%0l 1 5\r\n");
samux	6:01a1eb7c0145	192	sock.send(cmd, strlen(cmd));
samux	6:01a1eb7c0145	193	i++;
samux	6:01a1eb7c0145	194	break;
samux	6:01a1eb7c0145	195	case 6:
samux	6:01a1eb7c0145	196	sprintf(cmd, "Sec-WebSocket-key2: 12998 5 Y3 1 .P00\r\n\r\n");
samux	6:01a1eb7c0145	197	sock.send(cmd, strlen(cmd));
samux	6:01a1eb7c0145	198	i++;
samux	6:01a1eb7c0145	199	break;
samux	6:01a1eb7c0145	200	case 7:
samux	6:01a1eb7c0145	201	sock.send("^n:ds[4U", 8);
samux	6:01a1eb7c0145	202	i++;
samux	6:01a1eb7c0145	203	break;
samux	6:01a1eb7c0145	204	case 8:
samux	9:9fa055ed54b4	205	if (response_server_eth)
samux	9:9fa055ed54b4	206	i++;
samux	9:9fa055ed54b4	207	else
samux	9:9fa055ed54b4	208	break;
samux	9:9fa055ed54b4	209
samux	6:01a1eb7c0145	210	default:
samux	6:01a1eb7c0145	211	break;
samux	6:01a1eb7c0145	212	}
samux	6:01a1eb7c0145	213	}
samux	9:9fa055ed54b4	214	if (i==9) {
samux	6:01a1eb7c0145	215	printf("\r\nip_domain: %s\r\npath: /%s\r\nport: %s\r\n\r\n",this->ip_domain, this->path, this->port);
samux	6:01a1eb7c0145	216	return true;
samux	6:01a1eb7c0145	217	}
samux	6:01a1eb7c0145	218	}
samux	6:01a1eb7c0145	219	}
samux	6:01a1eb7c0145	220
samux	0:21fe3b05249f	221	}
samux	0:21fe3b05249f	222	return true;
samux	0:21fe3b05249f	223	}
samux	0:21fe3b05249f	224
samux	6:01a1eb7c0145	225	void Websocket::Send(char * str) {
samux	0:21fe3b05249f	226	char cmd[100];
samux	6:01a1eb7c0145	227	if (wifi_use) {
samux	6:01a1eb7c0145	228	wifi->putc('\x00');
samux	6:01a1eb7c0145	229	sprintf(cmd, "%s%c", str, '\xff');
samux	6:01a1eb7c0145	230	wifi->Send(cmd, "NO");
samux	6:01a1eb7c0145	231	} else if (eth_use) {
samux	6:01a1eb7c0145	232	sprintf(cmd, "%c%s%c", 0x00, str, 0xff);
samux	6:01a1eb7c0145	233	Net::poll();
samux	6:01a1eb7c0145	234	sock.send(cmd, strlen(cmd + 1) + 1);
samux	6:01a1eb7c0145	235	}
samux	0:21fe3b05249f	236	}
samux	0:21fe3b05249f	237
samux	6:01a1eb7c0145	238	bool Websocket::read(char * message) {
samux	0:21fe3b05249f	239	int i = 0;
samux	6:01a1eb7c0145	240
samux	9:9fa055ed54b4	241	if (wifi_use) {
samux	9:9fa055ed54b4	242	char char_read;
samux	9:9fa055ed54b4	243	if (!wifi->readable()) {
samux	9:9fa055ed54b4	244	message = NULL;
samux	9:9fa055ed54b4	245	return false;
samux	9:9fa055ed54b4	246	}
samux	9:9fa055ed54b4	247
samux	9:9fa055ed54b4	248	if (wifi->getc() != 0x00) {
samux	9:9fa055ed54b4	249	message = NULL;
samux	9:9fa055ed54b4	250	return false;
samux	9:9fa055ed54b4	251	}
samux	6:01a1eb7c0145	252
samux	9:9fa055ed54b4	253	while ( (char_read = wifi->getc()) != 0xff)
samux	9:9fa055ed54b4	254	message[i++] = char_read;
samux	9:9fa055ed54b4	255	message[i] = 0;
samux	9:9fa055ed54b4	256	} else if (eth_use) {
samux	9:9fa055ed54b4	257
samux	9:9fa055ed54b4	258	if (eth_rx[0] != 0x00) {
samux	9:9fa055ed54b4	259	message = NULL;
samux	9:9fa055ed54b4	260	return false;
samux	9:9fa055ed54b4	261	}
samux	9:9fa055ed54b4	262	while (eth_rx[i + 1] != 0xff) {
samux	9:9fa055ed54b4	263	message[i] = eth_rx[i + 1];
samux	9:9fa055ed54b4	264	i++;
samux	9:9fa055ed54b4	265	}
samux	9:9fa055ed54b4	266	message[i] = 0;
samux	0:21fe3b05249f	267	}
samux	0:21fe3b05249f	268	return true;
samux	0:21fe3b05249f	269	}
samux	0:21fe3b05249f	270
samux	6:01a1eb7c0145	271	bool Websocket::close() {
samux	9:9fa055ed54b4	272	if (wifi_use) {
samux	9:9fa055ed54b4	273	if (!wifi->CmdMode()) {
samux	9:9fa055ed54b4	274	printf("Websocket::close: cannot enter in cmd mode\r\n");
samux	9:9fa055ed54b4	275	return false;
samux	9:9fa055ed54b4	276	}
samux	6:01a1eb7c0145	277
samux	9:9fa055ed54b4	278	wifi->Send("close\r", "NO");
samux	6:01a1eb7c0145	279
samux	9:9fa055ed54b4	280	if (!wifi->exit())
samux	9:9fa055ed54b4	281	return false;
samux	9:9fa055ed54b4	282	} else if (eth_use) {
samux	9:9fa055ed54b4	283	if (sock.close())
samux	9:9fa055ed54b4	284	return false;
samux	9:9fa055ed54b4	285	}
samux	0:21fe3b05249f	286	return true;
samux	0:21fe3b05249f	287	}
samux	0:21fe3b05249f	288
samux	0:21fe3b05249f	289
samux	0:21fe3b05249f	290
samux	6:01a1eb7c0145	291	bool Websocket::connected() {
samux	6:01a1eb7c0145	292	if (wifi_use) {
samux	6:01a1eb7c0145	293	char str[10];
samux	6:01a1eb7c0145	294
samux	6:01a1eb7c0145	295	wait(0.25);
samux	6:01a1eb7c0145	296	if (!wifi->CmdMode()) {
samux	6:01a1eb7c0145	297	printf("Websocket::connected: cannot enter in cmd mode\r\n");
samux	0:21fe3b05249f	298	return false;
samux	3:9b00db719afa	299	}
samux	6:01a1eb7c0145	300	wait(0.25);
samux	6:01a1eb7c0145	301
samux	6:01a1eb7c0145	302	wifi->Send("show c\r\n", "NO");
samux	6:01a1eb7c0145	303	wifi->read(str);
samux	6:01a1eb7c0145	304
samux	6:01a1eb7c0145	305	if (str[3] == '1') {
samux	6:01a1eb7c0145	306	if (!wifi->exit()) {
samux	6:01a1eb7c0145	307	printf("Websocket::connected: cannot exit\r\n");
samux	6:01a1eb7c0145	308	return false;
samux	6:01a1eb7c0145	309	}
samux	6:01a1eb7c0145	310	return true;
samux	6:01a1eb7c0145	311	}
samux	6:01a1eb7c0145	312	if (!wifi->exit())
samux	6:01a1eb7c0145	313	printf("Websocket::connected: cannot exit\r\n");
samux	6:01a1eb7c0145	314	return false;
samux	9:9fa055ed54b4	315	} else if (eth_use)
samux	6:01a1eb7c0145	316	return eth_connected;
samux	6:01a1eb7c0145	317
samux	6:01a1eb7c0145	318	return true;
samux	6:01a1eb7c0145	319	}
samux	6:01a1eb7c0145	320
samux	6:01a1eb7c0145	321
samux	6:01a1eb7c0145	322
samux	6:01a1eb7c0145	323
samux	6:01a1eb7c0145	324
samux	6:01a1eb7c0145	325	void Websocket::onTCPSocketEvent(TCPSocketEvent e) {
samux	9:9fa055ed54b4	326	if (e == TCPSOCKET_CONNECTED) {
samux	9:9fa055ed54b4	327	eth_connected = true;
samux	9:9fa055ed54b4	328	printf("TCP Socket Connected\r\n");
samux	9:9fa055ed54b4	329	}
samux	9:9fa055ed54b4	330	else if (e == TCPSOCKET_WRITEABLE)
samux	9:9fa055ed54b4	331	{
samux	9:9fa055ed54b4	332	}
samux	9:9fa055ed54b4	333	else if (e == TCPSOCKET_READABLE)
samux	9:9fa055ed54b4	334	{
samux	9:9fa055ed54b4	335	int len = sock.recv(eth_rx, 512);
samux	9:9fa055ed54b4	336	eth_rx[len] = 0;
samux	9:9fa055ed54b4	337	if (!response_server_eth) {
samux	9:9fa055ed54b4	338	string checking;
samux	9:9fa055ed54b4	339	size_t found = string::npos;
samux	9:9fa055ed54b4	340	checking = eth_rx;
samux	9:9fa055ed54b4	341	found = checking.find("HTTP");
samux	9:9fa055ed54b4	342	if (found != string::npos)
samux	9:9fa055ed54b4	343	response_server_eth = true;
samux	9:9fa055ed54b4	344	}
samux	9:9fa055ed54b4	345	}
samux	9:9fa055ed54b4	346	else
samux	9:9fa055ed54b4	347	{
samux	9:9fa055ed54b4	348	printf("TCP Socket Fail\r\n");
samux	9:9fa055ed54b4	349	eth_connected = false;
samux	0:21fe3b05249f	350	}
samux	6:01a1eb7c0145	351	}
samux	6:01a1eb7c0145	352
samux	6:01a1eb7c0145	353

Repository toolbox

Export to desktop IDE

Repository details

Type:	Library
Created:	12 Aug 2011
Imports:	678
Forks:	0
Commits:	26
Dependents:	0
Dependencies:	0
Followers:	4

This repository is Public (Unlisted).

Websocket.cpp@9:9fa055ed54b4, 2011-08-24 (annotated)

Who changed what in which revision?

Repository toolbox

Repository details

Important Information for this Arm website

Access Warning