329 lines
8.3 KiB
C
329 lines
8.3 KiB
C
/*
|
|
* $Id: imgsock_manager.c 54 2011-05-10 13:22:47Z kaori $
|
|
*
|
|
* Copyright (c) 2002-2011, Communications and Remote Sensing Laboratory, Universite catholique de Louvain (UCL), Belgium
|
|
* Copyright (c) 2002-2011, Professor Benoit Macq
|
|
* Copyright (c) 2010-2011, Kaori Hagihara
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS `AS IS'
|
|
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
|
* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
* POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#ifdef _WIN32
|
|
#include <windows.h>
|
|
#define strcasecmp _stricmp
|
|
#else
|
|
#include <strings.h>
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/uio.h>
|
|
#include <sys/param.h>
|
|
#include <netinet/in.h>
|
|
#include <arpa/inet.h>
|
|
#include <netdb.h>
|
|
#endif
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include "imgsock_manager.h"
|
|
|
|
#define BUF_LEN 256
|
|
|
|
SOCKET open_listeningsocket()
|
|
{
|
|
SOCKET listening_socket;
|
|
struct sockaddr_in sin;
|
|
int sock_optval = 1;
|
|
int port = 5000;
|
|
|
|
listening_socket = socket(AF_INET, SOCK_STREAM, 0);
|
|
if ( listening_socket == -1 ){
|
|
perror("socket");
|
|
exit(1);
|
|
}
|
|
|
|
if ( setsockopt(listening_socket, SOL_SOCKET, SO_REUSEADDR,
|
|
&sock_optval, sizeof(sock_optval)) == -1 ){
|
|
perror("setsockopt");
|
|
exit(1);
|
|
}
|
|
|
|
memset(&sin, 0, sizeof(sin));
|
|
sin.sin_family = AF_INET;
|
|
sin.sin_port = htons(port);
|
|
sin.sin_addr.s_addr = htonl(INADDR_ANY);
|
|
|
|
if ( bind(listening_socket, (struct sockaddr *)&sin, sizeof(sin)) < 0 ){
|
|
perror("bind");
|
|
close_socket(listening_socket);
|
|
exit(1);
|
|
}
|
|
|
|
if( listen(listening_socket, SOMAXCONN) == -1){
|
|
perror("listen");
|
|
close_socket(listening_socket);
|
|
exit(1);
|
|
}
|
|
fprintf( stderr, "port %d is listened\n", port);
|
|
|
|
return listening_socket;
|
|
}
|
|
|
|
SOCKET accept_socket( SOCKET listening_socket)
|
|
{
|
|
struct sockaddr_in peer_sin;
|
|
unsigned int addrlen = sizeof(peer_sin);
|
|
|
|
return accept( listening_socket, (struct sockaddr *)&peer_sin, &addrlen);
|
|
}
|
|
|
|
msgtype_t identify_clientmsg( SOCKET connected_socket)
|
|
{
|
|
int receive_size;
|
|
char buf[BUF_LEN];
|
|
char *magicid[] = { "JPIP-stream", "PNM request", "XML request", "TID request", "CID request", "CID destroy", "SIZ request", "JP2 save", "QUIT"};
|
|
int i;
|
|
|
|
receive_size = receive_line( connected_socket, buf);
|
|
|
|
if( receive_size == 0){
|
|
fprintf( stderr, "Error to receive the header of client message\n");
|
|
return MSGERROR;
|
|
}
|
|
|
|
for( i=0; i<NUM_OF_MSGTYPES; i++){
|
|
if( strncasecmp( magicid[i], buf, strlen(magicid[i])) == 0){
|
|
fprintf( stderr, "%s\n", magicid[i]);
|
|
return i;
|
|
}
|
|
}
|
|
|
|
fprintf( stderr, "Cannot identify client message type %s\n", buf);
|
|
return MSGERROR;
|
|
}
|
|
|
|
Byte_t * receive_JPIPstream( SOCKET connected_socket, char **target, char **tid, char **cid, int *streamlen)
|
|
{
|
|
Byte_t *jpipstream=NULL, *ptr;
|
|
char buf[BUF_LEN], versionstring[] = "version 1.2";
|
|
int linelen, redlen, remlen;
|
|
|
|
*target = *cid = *tid = NULL;
|
|
|
|
if((linelen = receive_line( connected_socket, buf)) == 0)
|
|
return NULL;
|
|
if( strncmp( versionstring, buf, strlen(versionstring))!=0){
|
|
fprintf( stderr, "Wrong format\n");
|
|
return NULL;
|
|
}
|
|
|
|
if((linelen = receive_line( connected_socket, buf)) == 0)
|
|
return NULL;
|
|
|
|
if( strstr( buf, "jp2")){
|
|
// register cid option
|
|
*target = strdup( buf);
|
|
|
|
if((linelen = receive_line( connected_socket, buf)) == 0)
|
|
return NULL;
|
|
if( strcmp( buf, "0") != 0)
|
|
*tid = strdup( buf);
|
|
|
|
if((linelen = receive_line( connected_socket, buf)) == 0)
|
|
return NULL;
|
|
if( strcmp( buf, "0") != 0)
|
|
*cid = strdup( buf);
|
|
|
|
if((linelen = receive_line( connected_socket, buf)) == 0)
|
|
return NULL;
|
|
}
|
|
|
|
*streamlen = atoi( buf);
|
|
fprintf( stderr, "Receive Data: %d Bytes\n", *streamlen);
|
|
|
|
jpipstream = (unsigned char *)malloc( (*streamlen));
|
|
ptr = jpipstream;
|
|
remlen = (*streamlen);
|
|
while( remlen > 0){
|
|
redlen = recv( connected_socket, ptr, remlen, 0);
|
|
if( redlen == -1){
|
|
fprintf( stderr, "receive JPT- JPP- stream error\n");
|
|
break;
|
|
}
|
|
remlen -= redlen;
|
|
ptr = ptr + redlen;
|
|
}
|
|
|
|
return jpipstream;
|
|
}
|
|
|
|
void send_stream( SOCKET connected_socket, void *stream, int length);
|
|
|
|
void send_XMLstream( SOCKET connected_socket, Byte_t *xmlstream, int length)
|
|
{
|
|
Byte_t header[5];
|
|
|
|
header[0] = 'X';
|
|
header[1] = 'M';
|
|
header[2] = 'L';
|
|
header[3] = (length >> 8) & 0xff;
|
|
header[4] = length & 0xff;
|
|
|
|
send_stream( connected_socket, header, 5);
|
|
send_stream( connected_socket, xmlstream, length);
|
|
}
|
|
|
|
void send_IDstream( SOCKET connected_socket, char *id, int idlen, char *label);
|
|
|
|
void send_CIDstream( SOCKET connected_socket, char *cid, int cidlen)
|
|
{
|
|
send_IDstream( connected_socket, cid, cidlen, "CID");
|
|
}
|
|
|
|
void send_TIDstream( SOCKET connected_socket, char *tid, int tidlen)
|
|
{
|
|
send_IDstream( connected_socket, tid, tidlen, "TID");
|
|
}
|
|
|
|
void send_IDstream( SOCKET connected_socket, char *id, int idlen, char *label)
|
|
{
|
|
Byte_t header[4];
|
|
|
|
header[0] = label[0];
|
|
header[1] = label[1];
|
|
header[2] = label[2];
|
|
header[3] = idlen & 0xff;
|
|
|
|
send_stream( connected_socket, header, 4);
|
|
send_stream( connected_socket, id, idlen);
|
|
}
|
|
|
|
void send_PNMstream( SOCKET connected_socket, Byte_t *pnmstream, unsigned int width, unsigned int height, unsigned int numofcomp, Byte_t maxval)
|
|
{
|
|
int pnmlen = 0;
|
|
Byte_t header[7];
|
|
|
|
pnmlen = width*height*numofcomp;
|
|
|
|
header[0] = 'P';
|
|
header[1] = numofcomp==3 ? 6:5;
|
|
header[2] = (width >> 8) & 0xff;
|
|
header[3] = width & 0xff;
|
|
header[4] = (height >> 8) & 0xff;
|
|
header[5] = height & 0xff;
|
|
header[6] = maxval;
|
|
|
|
send_stream( connected_socket, header, 7);
|
|
send_stream( connected_socket, pnmstream, pnmlen);
|
|
}
|
|
|
|
void send_SIZstream( SOCKET connected_socket, unsigned int width, unsigned int height)
|
|
{
|
|
Byte_t responce[9];
|
|
|
|
responce[0] = 'S';
|
|
responce[1] = 'I';
|
|
responce[2] = 'Z';
|
|
responce[3] = (width >> 16) & 0xff;
|
|
responce[4] = (width >> 8) & 0xff;
|
|
responce[5] = width & 0xff;
|
|
responce[6] = (height >> 16) & 0xff;
|
|
responce[7] = (height >> 8) & 0xff;
|
|
responce[8] = height & 0xff;
|
|
|
|
send_stream( connected_socket, responce, 9);
|
|
}
|
|
|
|
void send_stream( SOCKET connected_socket, void *stream, int length)
|
|
{
|
|
void *ptr = stream;
|
|
int remlen = length;
|
|
|
|
while( remlen > 0){
|
|
int sentlen = send( connected_socket, ptr, remlen, 0);
|
|
if( sentlen == -1){
|
|
fprintf( stderr, "sending stream error\n");
|
|
break;
|
|
}
|
|
remlen = remlen - sentlen;
|
|
ptr = ptr + sentlen;
|
|
}
|
|
}
|
|
|
|
int receive_line(SOCKET connected_socket, char *p)
|
|
{
|
|
int len = 0;
|
|
while (1){
|
|
int ret;
|
|
ret = recv( connected_socket, p, 1, 0);
|
|
if ( ret == -1 ){
|
|
perror("receive");
|
|
exit(1);
|
|
} else if ( ret == 0 ){
|
|
break;
|
|
}
|
|
if ( *p == '\n' )
|
|
break;
|
|
p++;
|
|
len++;
|
|
}
|
|
*p = '\0';
|
|
|
|
if( len == 0)
|
|
fprintf( stderr, "Header receive error\n");
|
|
|
|
return len;
|
|
}
|
|
|
|
char * receive_string( SOCKET connected_socket)
|
|
{
|
|
char buf[BUF_LEN];
|
|
|
|
receive_line( connected_socket, buf);
|
|
|
|
return strdup(buf);
|
|
}
|
|
|
|
void response_signal( SOCKET connected_socket, bool succeed)
|
|
{
|
|
Byte_t code;
|
|
|
|
if( succeed)
|
|
code = 1;
|
|
else
|
|
code = 0;
|
|
|
|
if( send( connected_socket, &code, 1, 0) != 1)
|
|
fprintf( stderr, "Response signalling error\n");
|
|
}
|
|
|
|
int close_socket( SOCKET sock)
|
|
{
|
|
#ifdef _WIN32
|
|
return closesocket( sock);
|
|
#else
|
|
return close( sock);
|
|
#endif
|
|
}
|