74 lines
2.0 KiB
C++
Executable File
74 lines
2.0 KiB
C++
Executable File
/*
|
|
Copyright (C) 2016 Davidson Francis <davidsondfgl@gmail.com>
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>
|
|
*/
|
|
#include "../hsmodem.h"
|
|
#include "sha1.h"
|
|
#include "base64.h"
|
|
|
|
/**
|
|
* Gets the field Sec-WebSocket-Accept on response, by
|
|
* an previously informed key.
|
|
* @param wsKey Sec-WebSocket-Key
|
|
* @param dest source to be stored the value.
|
|
*/
|
|
int getHSaccept(char *wsKey, unsigned char **dest)
|
|
{
|
|
SHA1Context ctx;
|
|
char *str = (char *)malloc( sizeof(char) * (WS_KEY_LEN + WS_MS_LEN + 1) );
|
|
unsigned char hash[SHA1HashSize];
|
|
|
|
strcpy(str, wsKey);
|
|
strcat(str, MAGIC_STRING);
|
|
|
|
SHA1Reset(&ctx);
|
|
SHA1Input(&ctx, (const uint8_t *)str, WS_KEYMS_LEN);
|
|
SHA1Result(&ctx, hash);
|
|
|
|
*dest = base64_encode(hash, SHA1HashSize, NULL);
|
|
*(*dest + strlen((const char *)*dest) - 1) = '\0';
|
|
free(str);
|
|
return (0);
|
|
}
|
|
|
|
/**
|
|
* Gets the complete response to accomplish a succesfully
|
|
* handshake.
|
|
* @param hsrequest Client request.
|
|
* @param hsresponse Server response.
|
|
*/
|
|
int getHSresponse(char *hsrequest, char **hsresponse)
|
|
{
|
|
char *s;
|
|
unsigned char *accept;
|
|
|
|
for (s = strtok(hsrequest, "\r\n"); s != NULL; s = strtok(NULL, "\r\n") )
|
|
if (strstr(s, WS_HS_REQ) != NULL)
|
|
break;
|
|
|
|
s = strtok(s, " ");
|
|
s = strtok(NULL, " ");
|
|
|
|
getHSaccept(s, &accept);
|
|
|
|
*hsresponse = (char*)malloc(sizeof(char) * WS_HS_ACCLEN);
|
|
strcpy(*hsresponse, WS_HS_ACCEPT);
|
|
strcat(*hsresponse, (const char *)accept);
|
|
strcat(*hsresponse, "\r\n\r\n");
|
|
|
|
free(accept);
|
|
return (0);
|
|
}
|