Websocket - WebSocket client library

Users » samux » Code » Websocket

WebSocket client library

Websocket.cpp@6:01a1eb7c0145, 2011-08-24 (annotated)

Committer:: samux
Date:: Wed Aug 24 10:49:36 2011 +0000
Revision:: 6:01a1eb7c0145
Parent:: 4:1d3e630b8f9c
Child:: 7:b15978708360

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	6:01a1eb7c0145	5	this->wifi = wifi;
samux	6:01a1eb7c0145	6	wifi_use = true;
samux	6:01a1eb7c0145	7	eth_use = false;
samux	6:01a1eb7c0145	8	fillFields(wifi_use, url);
samux	6:01a1eb7c0145	9	}
samux	6:01a1eb7c0145	10
samux	6:01a1eb7c0145	11	void Websocket::fillFields(bool wifi, char * url) {
samux	0:21fe3b05249f	12	char *res = NULL;
samux	6:01a1eb7c0145	13	char *res1 = NULL;
samux	6:01a1eb7c0145	14
samux	0:21fe3b05249f	15	char buf[30];
samux	0:21fe3b05249f	16	strcpy(buf, url);
samux	6:01a1eb7c0145	17
samux	0:21fe3b05249f	18	res = strtok(buf, ":");
samux	6:01a1eb7c0145	19	if (strcmp(res, "ws"))
samux	0:21fe3b05249f	20	{
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	6:01a1eb7c0145	24	}
samux	6:01a1eb7c0145	25	else
samux	0:21fe3b05249f	26	{
samux	6:01a1eb7c0145	27	//ip_domain and port
samux	6:01a1eb7c0145	28	res = strtok(NULL, "/");
samux	6:01a1eb7c0145	29
samux	6:01a1eb7c0145	30	//path
samux	6:01a1eb7c0145	31	res1 = strtok(NULL, " ");
samux	6:01a1eb7c0145	32	if (res1 != NULL) {
samux	6:01a1eb7c0145	33	path = (char ) malloc (sizeof(char) strlen(res1));
samux	6:01a1eb7c0145	34	strcpy(this->path, res1);
samux	6:01a1eb7c0145	35	}
samux	6:01a1eb7c0145	36
samux	6:01a1eb7c0145	37	//ip_domain
samux	6:01a1eb7c0145	38	res = strtok(res, ":");
samux	6:01a1eb7c0145	39	if (res != NULL) {
samux	0:21fe3b05249f	40	ip_domain = (char ) malloc (sizeof(char) strlen(res));
samux	0:21fe3b05249f	41	strcpy(this->ip_domain, res);
samux	6:01a1eb7c0145	42	//if we use ethernet, we must decode ip address or use dnsresolver
samux	6:01a1eb7c0145	43	if (!wifi) {
samux	6:01a1eb7c0145	44	printf("ethernet use\r\n");
samux	6:01a1eb7c0145	45	strcpy(buf, res);
samux	6:01a1eb7c0145	46	//we try to decode the ip address
samux	6:01a1eb7c0145	47	if (buf[0] >= '0' && buf[0] <= '9') {
samux	6:01a1eb7c0145	48	res = strtok(buf, ".");
samux	6:01a1eb7c0145	49	int i = 0;
samux	6:01a1eb7c0145	50	int ip[4];
samux	6:01a1eb7c0145	51	while (res != NULL) {
samux	6:01a1eb7c0145	52	ip[i] = atoi(res);
samux	6:01a1eb7c0145	53	res = strtok(NULL, ".");
samux	6:01a1eb7c0145	54	i++;
samux	6:01a1eb7c0145	55	}
samux	6:01a1eb7c0145	56	server_ip = IpAddr(ip[0], ip[1], ip[2], ip[3]);
samux	6:01a1eb7c0145	57	printf("server=%i.%i.%i.%i\n",server_ip[0],server_ip[1],server_ip[2],server_ip[3]);
samux	6:01a1eb7c0145	58	}
samux	6:01a1eb7c0145	59	//we must use dnsresolver to find the ip address
samux	6:01a1eb7c0145	60	else {
samux	6:01a1eb7c0145	61	DNSResolver dr;
samux	6:01a1eb7c0145	62	server_ip = dr.resolveName(buf);
samux	6:01a1eb7c0145	63	printf("server=%i.%i.%i.%i\n",server_ip[0],server_ip[1],server_ip[2],server_ip[3]);
samux	6:01a1eb7c0145	64	}
samux	6:01a1eb7c0145	65	}
samux	0:21fe3b05249f	66	}
samux	6:01a1eb7c0145	67
samux	6:01a1eb7c0145	68	//port
samux	6:01a1eb7c0145	69	res = strtok(NULL, ":");
samux	6:01a1eb7c0145	70	if (res != NULL) {
samux	6:01a1eb7c0145	71	port = (char ) malloc (sizeof(char) strlen(res));
samux	6:01a1eb7c0145	72	strcpy(this->port, res);
samux	6:01a1eb7c0145	73	} else {
samux	6:01a1eb7c0145	74	port = (char ) malloc (sizeof(char) 3);
samux	6:01a1eb7c0145	75	strcpy(this->port, "80");
samux	0:21fe3b05249f	76	}
samux	0:21fe3b05249f	77	}
samux	0:21fe3b05249f	78	}
samux	0:21fe3b05249f	79
samux	6:01a1eb7c0145	80
samux	6:01a1eb7c0145	81	Websocket::Websocket(char * url) {
samux	6:01a1eb7c0145	82	wifi_use = false;
samux	6:01a1eb7c0145	83	eth_use = true;
samux	6:01a1eb7c0145	84	eth_writeable = false;
samux	6:01a1eb7c0145	85	eth_readable = false;
samux	6:01a1eb7c0145	86	eth_connected = false;
samux	6:01a1eb7c0145	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	6:01a1eb7c0145	158	char msg[10];
samux	0:21fe3b05249f	159
samux	6:01a1eb7c0145	160	while (true) {
samux	6:01a1eb7c0145	161	Net::poll();
samux	6:01a1eb7c0145	162	if (tmr.read() > 0.1) {
samux	6:01a1eb7c0145	163	tmr.reset();
samux	6:01a1eb7c0145	164	if (eth_connected) {
samux	6:01a1eb7c0145	165	switch (i) {
samux	6:01a1eb7c0145	166	case 0:
samux	6:01a1eb7c0145	167	sprintf(cmd, "GET /%s HTTP/1.1\r\n", path);
samux	6:01a1eb7c0145	168	sock.send(cmd, strlen(cmd));
samux	6:01a1eb7c0145	169	i++;
samux	6:01a1eb7c0145	170	break;
samux	6:01a1eb7c0145	171	case 1:
samux	6:01a1eb7c0145	172	sprintf(cmd, "Host: %s:%s\r\n", ip_domain, port);
samux	6:01a1eb7c0145	173	sock.send(cmd, strlen(cmd));
samux	6:01a1eb7c0145	174	i++;
samux	6:01a1eb7c0145	175	break;
samux	6:01a1eb7c0145	176	case 2:
samux	6:01a1eb7c0145	177	sprintf(cmd, "Upgrade: WebSocket\r\n");
samux	6:01a1eb7c0145	178	sock.send(cmd, strlen(cmd));
samux	6:01a1eb7c0145	179	i++;
samux	6:01a1eb7c0145	180	break;
samux	6:01a1eb7c0145	181	case 3:
samux	6:01a1eb7c0145	182	sprintf(cmd, "Origin: http:%s:%s\r\n", ip_domain, port);
samux	6:01a1eb7c0145	183	sock.send(cmd, strlen(cmd));
samux	6:01a1eb7c0145	184	i++;
samux	6:01a1eb7c0145	185	break;
samux	6:01a1eb7c0145	186	case 4:
samux	6:01a1eb7c0145	187	sprintf(cmd, "Connection: Upgrade\r\n");
samux	6:01a1eb7c0145	188	sock.send(cmd, strlen(cmd));
samux	6:01a1eb7c0145	189	i++;
samux	6:01a1eb7c0145	190	break;
samux	6:01a1eb7c0145	191	case 5:
samux	6:01a1eb7c0145	192	sprintf(cmd, "Sec-WebSocket-Key1: 4 @1 46546xW%0l 1 5\r\n");
samux	6:01a1eb7c0145	193	sock.send(cmd, strlen(cmd));
samux	6:01a1eb7c0145	194	i++;
samux	6:01a1eb7c0145	195	break;
samux	6:01a1eb7c0145	196	case 6:
samux	6:01a1eb7c0145	197	sprintf(cmd, "Sec-WebSocket-key2: 12998 5 Y3 1 .P00\r\n\r\n");
samux	6:01a1eb7c0145	198	sock.send(cmd, strlen(cmd));
samux	6:01a1eb7c0145	199	i++;
samux	6:01a1eb7c0145	200	break;
samux	6:01a1eb7c0145	201	case 7:
samux	6:01a1eb7c0145	202	sock.send("^n:ds[4U", 8);
samux	6:01a1eb7c0145	203	i++;
samux	6:01a1eb7c0145	204	break;
samux	6:01a1eb7c0145	205	case 8:
samux	6:01a1eb7c0145	206	if(eth_readable)
samux	6:01a1eb7c0145	207	{
samux	6:01a1eb7c0145	208	string checking;
samux	6:01a1eb7c0145	209	size_t found = string::npos;
samux	6:01a1eb7c0145	210	sock.recv(msg, 10);
samux	6:01a1eb7c0145	211	Net::poll();
samux	6:01a1eb7c0145	212	checking = msg;
samux	6:01a1eb7c0145	213	found = checking.find("HTTP");
samux	6:01a1eb7c0145	214	if(found != string::npos)
samux	6:01a1eb7c0145	215	{
samux	6:01a1eb7c0145	216	//printf("checking: %s\r\n", checking);
samux	6:01a1eb7c0145	217	i++;
samux	6:01a1eb7c0145	218	break;
samux	6:01a1eb7c0145	219	}
samux	6:01a1eb7c0145	220	else
samux	6:01a1eb7c0145	221	return false;
samux	6:01a1eb7c0145	222	}
samux	6:01a1eb7c0145	223	break;
samux	6:01a1eb7c0145	224
samux	6:01a1eb7c0145	225	default:
samux	6:01a1eb7c0145	226	break;
samux	6:01a1eb7c0145	227	}
samux	6:01a1eb7c0145	228	}
samux	6:01a1eb7c0145	229	if (i==9)
samux	6:01a1eb7c0145	230	{
samux	6:01a1eb7c0145	231	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	232	return true;
samux	6:01a1eb7c0145	233	}
samux	6:01a1eb7c0145	234	}
samux	6:01a1eb7c0145	235	}
samux	6:01a1eb7c0145	236
samux	0:21fe3b05249f	237	}
samux	0:21fe3b05249f	238	return true;
samux	0:21fe3b05249f	239	}
samux	0:21fe3b05249f	240
samux	6:01a1eb7c0145	241	void Websocket::Send(char * str) {
samux	0:21fe3b05249f	242	char cmd[100];
samux	6:01a1eb7c0145	243	if (wifi_use) {
samux	6:01a1eb7c0145	244	wifi->putc('\x00');
samux	6:01a1eb7c0145	245	sprintf(cmd, "%s%c", str, '\xff');
samux	6:01a1eb7c0145	246	wifi->Send(cmd, "NO");
samux	6:01a1eb7c0145	247	} else if (eth_use) {
samux	6:01a1eb7c0145	248	sprintf(cmd, "%c%s%c", 0x00, str, 0xff);
samux	6:01a1eb7c0145	249	Net::poll();
samux	6:01a1eb7c0145	250	sock.send(cmd, strlen(cmd + 1) + 1);
samux	6:01a1eb7c0145	251	}
samux	0:21fe3b05249f	252	}
samux	0:21fe3b05249f	253
samux	6:01a1eb7c0145	254	bool Websocket::read(char * message) {
samux	0:21fe3b05249f	255	int i = 0;
samux	0:21fe3b05249f	256	char char_read;
samux	6:01a1eb7c0145	257
samux	6:01a1eb7c0145	258	if (!wifi->readable()) {
samux	0:21fe3b05249f	259	message = NULL;
samux	0:21fe3b05249f	260	return false;
samux	0:21fe3b05249f	261	}
samux	6:01a1eb7c0145	262
samux	6:01a1eb7c0145	263	if (wifi->getc() != 0x00) {
samux	0:21fe3b05249f	264	message = NULL;
samux	0:21fe3b05249f	265	return false;
samux	0:21fe3b05249f	266	}
samux	6:01a1eb7c0145	267
samux	0:21fe3b05249f	268	while ( (char_read = wifi->getc()) != 0xff)
samux	0:21fe3b05249f	269	message[i++] = char_read;
samux	6:01a1eb7c0145	270
samux	0:21fe3b05249f	271	return true;
samux	0:21fe3b05249f	272	}
samux	0:21fe3b05249f	273
samux	6:01a1eb7c0145	274	bool Websocket::close() {
samux	6:01a1eb7c0145	275	if (!wifi->CmdMode()) {
samux	0:21fe3b05249f	276	printf("Websocket::close: cannot enter in cmd mode\r\n");
samux	0:21fe3b05249f	277	return false;
samux	0:21fe3b05249f	278	}
samux	6:01a1eb7c0145	279
samux	0:21fe3b05249f	280	wifi->Send("close\r", "NO");
samux	6:01a1eb7c0145	281
samux	6:01a1eb7c0145	282	if (!wifi->exit())
samux	0:21fe3b05249f	283	return false;
samux	6:01a1eb7c0145	284
samux	0:21fe3b05249f	285	return true;
samux	0:21fe3b05249f	286	}
samux	0:21fe3b05249f	287
samux	0:21fe3b05249f	288
samux	0:21fe3b05249f	289
samux	6:01a1eb7c0145	290	bool Websocket::connected() {
samux	6:01a1eb7c0145	291	if (wifi_use) {
samux	6:01a1eb7c0145	292	char str[10];
samux	6:01a1eb7c0145	293
samux	6:01a1eb7c0145	294	wait(0.25);
samux	6:01a1eb7c0145	295	if (!wifi->CmdMode()) {
samux	6:01a1eb7c0145	296	printf("Websocket::connected: cannot enter in cmd mode\r\n");
samux	0:21fe3b05249f	297	return false;
samux	3:9b00db719afa	298	}
samux	6:01a1eb7c0145	299	wait(0.25);
samux	6:01a1eb7c0145	300
samux	6:01a1eb7c0145	301	wifi->Send("show c\r\n", "NO");
samux	6:01a1eb7c0145	302	wifi->read(str);
samux	6:01a1eb7c0145	303
samux	6:01a1eb7c0145	304	if (str[3] == '1') {
samux	6:01a1eb7c0145	305	if (!wifi->exit()) {
samux	6:01a1eb7c0145	306	printf("Websocket::connected: cannot exit\r\n");
samux	6:01a1eb7c0145	307	return false;
samux	6:01a1eb7c0145	308	}
samux	6:01a1eb7c0145	309	return true;
samux	6:01a1eb7c0145	310	}
samux	6:01a1eb7c0145	311	if (!wifi->exit())
samux	6:01a1eb7c0145	312	printf("Websocket::connected: cannot exit\r\n");
samux	6:01a1eb7c0145	313	return false;
samux	6:01a1eb7c0145	314	}
samux	6:01a1eb7c0145	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	6:01a1eb7c0145	326	switch (e) {
samux	6:01a1eb7c0145	327	case TCPSOCKET_CONNECTED:
samux	6:01a1eb7c0145	328	eth_connected = true;
samux	6:01a1eb7c0145	329	printf("TCP Socket Connected\r\n");
samux	6:01a1eb7c0145	330	break;
samux	6:01a1eb7c0145	331	case TCPSOCKET_WRITEABLE:
samux	6:01a1eb7c0145	332	eth_writeable = true;
samux	6:01a1eb7c0145	333	break;
samux	6:01a1eb7c0145	334	case TCPSOCKET_READABLE:
samux	6:01a1eb7c0145	335	printf("TCP Socket Readable\r\n");
samux	6:01a1eb7c0145	336	eth_readable = true;
samux	6:01a1eb7c0145	337	break;
samux	6:01a1eb7c0145	338	case TCPSOCKET_CONTIMEOUT:
samux	6:01a1eb7c0145	339	printf("TCP Socket Timeout\r\n");
samux	6:01a1eb7c0145	340	eth_connected = false;
samux	6:01a1eb7c0145	341	break;
samux	6:01a1eb7c0145	342	case TCPSOCKET_CONRST:
samux	6:01a1eb7c0145	343	printf("TCP Socket CONRST\r\n");
samux	6:01a1eb7c0145	344	eth_connected = false;
samux	6:01a1eb7c0145	345	break;
samux	6:01a1eb7c0145	346	case TCPSOCKET_CONABRT:
samux	6:01a1eb7c0145	347	printf("TCP Socket CONABRT\r\n");
samux	6:01a1eb7c0145	348	eth_connected = false;
samux	6:01a1eb7c0145	349	break;
samux	6:01a1eb7c0145	350	case TCPSOCKET_ERROR:
samux	6:01a1eb7c0145	351	printf("TCP Socket Error\r\n");
samux	6:01a1eb7c0145	352	eth_connected = false;
samux	6:01a1eb7c0145	353	break;
samux	6:01a1eb7c0145	354	case TCPSOCKET_DISCONNECTED:
samux	6:01a1eb7c0145	355	eth_connected = false;
samux	6:01a1eb7c0145	356	break;
samux	6:01a1eb7c0145	357	default:
samux	6:01a1eb7c0145	358	printf("DEFAULT\r\n");
samux	0:21fe3b05249f	359	}
samux	6:01a1eb7c0145	360	}
samux	6:01a1eb7c0145	361
samux	6:01a1eb7c0145	362

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@6:01a1eb7c0145, 2011-08-24 (annotated)

Who changed what in which revision?

Repository toolbox

Repository details

Important Information for this Arm website

Access Warning