//// C++ Implementation: Command//// Description:////// Author: Jally <jallyx@163.com>, (C) 2008//// Copyright: See COPYING file that comes with this distribution////#include "Command.h"#include "Control.h"#include "Pal.h"#include "utils.h"#include "support.h"#include "baling.h"uint32_t Command::packetn = 0; Command::Command():size(0){}Command::~Command(){}//廣播void Command::BroadCast(int sock){	extern struct interactive inter;	extern Control ctr;	GSList *list, *tmp;	SI addr;	CreateCommand(IPMSG_ABSENCEOPT | IPMSG_BR_ENTRY, ctr.myname);	TransferEncode(ctr.encode);	CreateIptuxExtra();	bzero(&addr, sizeof(addr));	addr.sin_family = AF_INET;	addr.sin_port = htons(IPTUX_DEFAULT_PORT);	list = tmp = get_sys_broadcast_addr(inter.udpsock);	while (tmp) {		inet_pton(AF_INET, (char *)tmp->data, &addr.sin_addr);		sendto(sock, buf, size, 0, (SA *) & addr, sizeof(addr));		my_delay(0, 9999999);		tmp = tmp->next;	}	g_slist_foreach(list, remove_foreach, GINT_TO_POINTER(UNKNOWN));	g_slist_free(list);}//單點廣播void Command::DialUp(int sock){	extern Control ctr;	in_addr_t ip1, ip2, ip;	GSList *tmp;	SI addr;	CreateCommand(IPMSG_DIALUPOPT | IPMSG_ABSENCEOPT | IPMSG_BR_ENTRY,		      ctr.myname);	TransferEncode(ctr.encode);	CreateIptuxExtra();	bzero(&addr, sizeof(addr));	addr.sin_family = AF_INET;	addr.sin_port = htons(IPTUX_DEFAULT_PORT);	pthread_mutex_lock(&ctr.mutex);	tmp = ctr.ipseg;	while (tmp) {		inet_pton(AF_INET, (char *)tmp->data, &ip1);		ip1 = ntohl(ip1), tmp = tmp->next;		inet_pton(AF_INET, (char *)tmp->data, &ip2);		ip2 = ntohl(ip2), tmp = tmp->next;		data_order(&ip1, &ip2);		ip = ip1;		while (ip <= ip2) {			addr.sin_addr.s_addr = htonl(ip);			sendto(sock, buf, size, 0, (SA *) & addr, sizeof(addr));			my_delay(0, 999999);			ip++;		}	}	pthread_mutex_unlock(&ctr.mutex);}//通知在線void Command::SendAnsentry(int sock, pointer data){	extern Control ctr;	Pal *pal;	SI addr;	pal = (Pal *) data;	CreateCommand(IPMSG_ABSENCEOPT | IPMSG_ANSENTRY, ctr.myname);	TransferEncode(pal->encode);	CreateIptuxExtra();	bzero(&addr, sizeof(addr));	addr.sin_family = AF_INET;	addr.sin_port = htons(IPTUX_DEFAULT_PORT);	addr.sin_addr.s_addr = pal->ipv4;	sendto(sock, buf, size, 0, (SA *) & addr, sizeof(addr));}//通告退出void Command::SendExit(int sock, pointer data){	Pal *pal;	SI addr;	pal = (Pal *) data;	CreateCommand(IPMSG_DIALUPOPT | IPMSG_BR_EXIT, NULL);	TransferEncode(pal->encode);	bzero(&addr, sizeof(addr));	addr.sin_family = AF_INET;	addr.sin_port = htons(IPTUX_DEFAULT_PORT);	addr.sin_addr.s_addr = pal->ipv4;	sendto(sock, buf, size, 0, (SA *) & addr, sizeof(addr));}//更改我的信息void Command::SendAbsence(int sock, pointer data){	extern Control ctr;	Pal *pal;	SI addr;	pal = (Pal *) data;	CreateCommand(IPMSG_ABSENCEOPT | IPMSG_BR_ABSENCE, ctr.myname);	TransferEncode(pal->encode);	CreateIptuxExtra();	bzero(&addr, sizeof(addr));	addr.sin_family = AF_INET;	addr.sin_port = htons(IPTUX_DEFAULT_PORT);	addr.sin_addr.s_addr = pal->ipv4;	sendto(sock, buf, size, 0, (SA *) & addr, sizeof(addr));}//探測好友void Command::SendDetectPacket(int sock, in_addr_t ipv4){	extern Control ctr;	SI addr;	CreateCommand(IPMSG_DIALUPOPT | IPMSG_ABSENCEOPT | IPMSG_BR_ENTRY,		      ctr.myname);	TransferEncode(ctr.encode);	CreateIptuxExtra();	bzero(&addr, sizeof(addr));	addr.sin_family = AF_INET;	addr.sin_port = htons(IPTUX_DEFAULT_PORT);	addr.sin_addr.s_addr = ipv4;	sendto(sock, buf, size, 0, (SA *) & addr, sizeof(addr));}//發送消息void Command::SendMessage(int sock, pointer data, const char *msg){	uint32_t packetno;	uint8_t count;	Pal *pal;	SI addr;	pal = (Pal *) data, packetno = packetn;	pal->CheckReply(packetno, true);	CreateCommand(IPMSG_SENDCHECKOPT | IPMSG_SENDMSG, msg);	TransferEncode(pal->encode);	bzero(&addr, sizeof(addr));	addr.sin_family = AF_INET;	addr.sin_port = htons(IPTUX_DEFAULT_PORT);	addr.sin_addr.s_addr = pal->ipv4;	count = 0;	do {		sendto(sock, buf, size, 0, (SA *) & addr, sizeof(addr));		my_delay(1, 0);		count++;	} while (!pal->CheckReply(packetno, false) && count < MAX_RETRYTIMES);	if (count >= MAX_RETRYTIMES)		pal->BufferInsertData(NULL, ERROR);}//回復消息void Command::SendReply(int sock, pointer data, uint32_t packetno){	char packetstr[5];	Pal *pal;	SI addr;	pal = (Pal *) data;	snprintf(packetstr, 5, "%u", packetno);	CreateCommand(IPMSG_SENDCHECKOPT | IPMSG_RECVMSG, packetstr);	TransferEncode(pal->encode);	bzero(&addr, sizeof(addr));	addr.sin_family = AF_INET;	addr.sin_port = htons(IPTUX_DEFAULT_PORT);	addr.sin_addr.s_addr = pal->ipv4;	sendto(sock, buf, size, 0, (SA *) & addr, sizeof(addr));}//群發消息void Command::SendGroupMsg(int sock, pointer data, const char *msg){	Pal *pal;	SI addr;	pal = (Pal *) data;	CreateCommand(IPMSG_BROADCASTOPT | IPMSG_SENDMSG, msg);	TransferEncode(pal->encode);	bzero(&addr, sizeof(addr));	addr.sin_family = AF_INET;	addr.sin_port = htons(IPTUX_DEFAULT_PORT);	addr.sin_addr.s_addr = pal->ipv4;	sendto(sock, buf, size, 0, (SA *) & addr, sizeof(addr));}bool Command::SendAskData(int sock, pointer data, uint32_t packetno,			  uint32_t fileid, uint32_t offset){	char attrstr[15];	Pal *pal;	SI addr;	pal = (Pal *) data;	snprintf(attrstr, 15, "%x:%x:%x", packetno, fileid, offset);	CreateCommand(IPMSG_FILEATTACHOPT | IPMSG_GETFILEDATA, attrstr);	TransferEncode(pal->encode);	bzero(&addr, sizeof(addr));	addr.sin_family = AF_INET;	addr.sin_port = htons(IPTUX_DEFAULT_PORT);	addr.sin_addr.s_addr = pal->ipv4;	if (Connect(sock, (SA *) & addr, sizeof(addr)) == -1)		return false;	if (Write(sock, buf, size) == -1)		return false;	return true;}bool Command::SendAskFiles(int sock, pointer data, uint32_t packetno,			   uint32_t fileid){	char attrstr[15];	Pal *pal;	SI addr;	pal = (Pal *) data;	snprintf(attrstr, 15, "%x:%x:0", packetno, fileid);	//兼容 LanQQ 軟件	CreateCommand(IPMSG_FILEATTACHOPT | IPMSG_GETDIRFILES, attrstr);	TransferEncode(pal->encode);	bzero(&addr, sizeof(addr));	addr.sin_family = AF_INET;	addr.sin_port = htons(IPTUX_DEFAULT_PORT);	addr.sin_addr.s_addr = pal->ipv4;	if (Connect(sock, (SA *) & addr, sizeof(addr)) == -1)		return false;	if (Write(sock, buf, size) == -1)		return false;	return true;}void Command::SendAskShared(int sock, pointer data){	Pal *pal;	SI addr;	pal = (Pal *) data;	CreateCommand(IPTUX_ASKSHARED, NULL);	TransferEncode(pal->encode);	bzero(&addr, sizeof(addr));	addr.sin_family = AF_INET;	addr.sin_port = htons(IPTUX_DEFAULT_PORT);	addr.sin_addr.s_addr = pal->ipv4;	sendto(sock, buf, size, 0, (SA *) & addr, sizeof(addr));}void Command::SendSharedInfo(int sock, pointer data, const char *extra){	char *ptr;	Pal *pal;	SI addr;	pal = (Pal *) data;	CreateCommand(IPMSG_FILEATTACHOPT | IPMSG_SENDMSG, NULL);	TransferEncode(pal->encode);	ptr = transfer_encode(extra, pal->encode, true);	CreateIpmsgExtra(ptr);	free(ptr);	bzero(&addr, sizeof(addr));	addr.sin_family = AF_INET;	addr.sin_port = htons(IPTUX_DEFAULT_PORT);	addr.sin_addr.s_addr = pal->ipv4;	sendto(sock, buf, size, 0, (SA *) & addr, sizeof(addr));}void Command::SendMyIcon(int sock, pointer data){	Pal *pal;	SI addr;	pal = (Pal *) data;	CreateCommand(IPTUX_SENDICON, NULL);	TransferEncode(pal->encode);	CreateIconExtra();	bzero(&addr, sizeof(addr));	addr.sin_family = AF_INET;	addr.sin_port = htons(IPTUX_DEFAULT_PORT);	addr.sin_addr.s_addr = pal->ipv4;	sendto(sock, buf, size, 0, (SA *) & addr, sizeof(addr));}void Command::SendMySign(int sock, pointer data){	extern Control ctr;	Pal *pal;	SI addr;	pal = (Pal *) data;	CreateCommand(IPTUX_SENDSIGN, ctr.sign);	TransferEncode(pal->encode);	bzero(&addr, sizeof(addr));	addr.sin_family = AF_INET;	addr.sin_port = htons(IPTUX_DEFAULT_PORT);	addr.sin_addr.s_addr = pal->ipv4;	sendto(sock, buf, size, 0, (SA *) & addr, sizeof(addr));}void Command::SendSublayer(int sock, pointer data, uint32_t opttype, const char *path){	Pal *pal;	SI addr;	int fd;	pal = (Pal *) data;	CreateCommand(opttype|IPTUX_SENDSUBLAYER, NULL);	TransferEncode(pal->encode);	bzero(&addr, sizeof(addr));	addr.sin_family = AF_INET;	addr.sin_port = htons(IPTUX_DEFAULT_PORT);	addr.sin_addr.s_addr = pal->ipv4;	if (Connect(sock, (SA *) & addr, sizeof(addr)) == -1 ||		   Write(sock, buf, size) == -1 ||		   (fd = Open(path, O_RDONLY)) == -1)		return;	SendSublayerData(sock, fd);	close(fd);}void Command::SendSublayerData(int sock, int fd){	ssize_t len;	do {		len = Read(fd, buf, MAX_UDPBUF);		if (len == 0 || len == -1)			break;		len = Write(sock, buf, len);		if (len == -1)			break;	} while (1);}void Command::CreateCommand(uint32_t command, const char *attach){	char *ptr, *env;	snprintf(buf, MAX_UDPBUF, "%s", IPTUX_VERSION);	size = strlen(buf);	ptr = buf + size;	snprintf(ptr, MAX_UDPBUF - size, ":%u", packetn);	packetn++;	size += strlen(ptr);	ptr = buf + size;	env = getenv("USERNAME");	snprintf(ptr, MAX_UDPBUF - size, ":%s", env);	size += strlen(ptr);	ptr = buf + size;	env = getenv("HOSTNAME");	snprintf(ptr, MAX_UDPBUF - size, ":%s", env);	size += strlen(ptr);	ptr = buf + size;	snprintf(ptr, MAX_UDPBUF - size, ":%u", command);	size += strlen(ptr);	ptr = buf + size;	snprintf(ptr, MAX_UDPBUF - size, ":%s", attach ? attach : "");	size += strlen(ptr) + 1;}void Command::TransferEncode(const char *encode){	char *ptr;	ptr = transfer_encode(buf, encode, true);	size = strlen(ptr) + 1;	memcpy(buf, ptr, size);	free(ptr);}void Command::CreateIptuxExtra(){	extern Control ctr;	char *ptr, *tmp;	ptr = buf + size;	tmp = strrchr(ctr.myicon, '/');	snprintf(ptr, MAX_UDPBUF - size, "%s", tmp ? tmp + 1 : ctr.myicon);	size += strlen(ptr) + 1;	ptr = buf + size;	snprintf(ptr, MAX_UDPBUF - size, "UTF-8");	size += strlen(ptr) + 1;}void Command::CreateIpmsgExtra(const char *extra){	char *ptr, *tmp;	ptr = buf + size;	snprintf(ptr, MAX_UDPBUF - size, "%s", extra);	if (tmp = strrchr(ptr, '\a'))		*(tmp + 1) = '\0';	size += strlen(ptr) + 1;}void Command::CreateIconExtra(){	char path[MAX_PATHBUF], *env;	ssize_t len;	int fd;	env = getenv("HOME");	snprintf(path, MAX_PATHBUF, "%s/.iptux/myicon", env);	if ((fd = Open(path, O_RDONLY)) == -1)		return;	len = Read(fd, buf + size, MAX_UDPBUF - size);	close(fd);	if (len != -1)		size += len;}