dms代码
转自：http://blog.csdn.net/bao_jinyu/article/details/7554293
/*
* date 20120510
* last author: baojinyu
* usage: successful as socket server
* successful as dlna media file server
* */
#include "Neptune.h"
#include "Platinum.h"
#include <sys/ioctl.h>
#include <sys/select.h>
#include <sys/time.h>
#include <sys/socket.h>
#include <netinet/in.h> /* definations of sa_family structre,htons,INADDR_ANY,... */
#include <arpa/inet.h> /* name ip format */
#include <sys/types.h> /*give the defination of pid_t*/
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <errno.h>
#include <stdlib.h>
#include <ctype.h> /* isspace() */
#define BUFFER_MAX 1024 /* input buffer for commands */
#define TOKEN_MAX 8 /* max number of arguments in buffer */
#define REPLY_MAX 256 /* largest reply allowed */
#define MAX_QUE_CONN_NM 10
NPT_SET_LOCAL_LOGGER("platinum.media.server.file")
//example:NPT_Debug("ERROR: failed to load entity (%d)\n", result);
int sockfd;
static NPT_Flags deviceStartFlag = 0; //服务启动状态为1，服务停止则状态为0
PLT_UPnP upnp;
PLT_DeviceHostReference device;
int init_sock()
{
sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (sockfd == -1)
{
NPT_LOG_INFO("baojinyu---------------create socket error! \n");
exit(1);
}
NPT_LOG_INFO("baojinyu---------------create socket successfully .\n");
return 0;
}
int init_device()
{
device = new PLT_FileMediaServer(
"/data/local/1.3gp",//"/data/local/if.mp3",
"Platinum UPnP Media Server",
false,
"SAMEDEVICEGUID", // NULL for random ID
(NPT_UInt16)1194);
NPT_List<NPT_IpAddress> list;
NPT_CHECK_SEVERE(PLT_UPnPMessageHelper::GetIPAddresses(list));
NPT_String ip = list.GetFirstItem()->ToString();
device->m_ModelDescription = "Platinum File Media Server";
device->m_ModelURL = "http://www.plutinosoft.com/";
device->m_ModelNumber = "1.0";
device->m_ModelName = "Platinum File Media Server";
device->m_Manufacturer = "Plutinosoft";
device->m_ManufacturerURL = "http://www.plutinosoft.com/";
upnp.AddDevice(device);
NPT_CHECK_SEVERE(upnp.Start());//Just for test.
return 0;
}
int init()
{
init_sock();
init_device();
return 0;
}
//the following are commands.
static int HandleCmd_dmsStartStop(char **arg, char reply[REPLY_MAX])
{
if (!deviceStartFlag){
//start service
NPT_LOG_INFO("baojinyu------------socket cmd Start。");
NPT_CHECK_SEVERE(upnp.Start());
NPT_LOG_INFO("baojinyu------------upnp Start success!");
deviceStartFlag = 1;
}
else{
//stop service
NPT_LOG_INFO("baojinyu------------socket cmd stop。");
upnp.Stop();
NPT_LOG_INFO("baojinyu------------upnp stop success!");
deviceStartFlag = 0;
}
return deviceStartFlag;
}
struct cmdinfo {
const char *name;
unsigned numargs;
int (*func)(char **arg, char reply[REPLY_MAX]);
};
struct cmdinfo cmds[] = {
//{ "isMouse", 1, isMouse },
//{ "mousePrecision", 1, mousePrecision },
{ "dmsStartStop", 0, HandleCmd_dmsStartStop },
//{ "mediaManage", 1, HandleCmd_mediaManage },
//{ "serverConfig", 1, HandleCmd_serverConfig },
};
static int readx(int fd, void *_buf, int count)
{
char *buf =(char *) _buf;
int n = 0, r;
if (count < 0) return -1;
while (n < count) {
r = read(fd, buf + n, count - n);
if (r < 0) {
if (errno == EINTR) continue;
NPT_LOG_INFO("read error\n");
return -1;
}
if (r == 0) {
NPT_LOG_INFO("eof");
return -1; /* EOF */
}
n += r;
}
return 0;
}
static int writex(int fd, const void *_buf, int count)
{
const char *buf =(char *) _buf;
int n = 0, r;
if (count < 0) return -1;
while (n < count) {
r = write(fd, buf + n, count - n);
if (r < 0) {
if (errno == EINTR) continue;
NPT_LOG_INFO("write error");
return -1;
}
n += r;
}
return 0;
}
/* Tokenize the command buffer, locate a matching command,
* ensure that the required number of arguments are provided,
* call the function(), return the result.
*/
static int execute(int s, char cmd[BUFFER_MAX])
{
char reply[REPLY_MAX];
char *arg[TOKEN_MAX+1];
unsigned i;
unsigned n = 0;
unsigned short count, le_count;
int ret = -1;
/* default reply is "" */
reply[0] = 0;
/* n is number of args (not counting arg[0]) */
arg[0] = cmd;
while (*cmd) {
if (isspace(*cmd)) {
*cmd++ = 0;
n++;
arg[n] = cmd;
if (n == TOKEN_MAX) {
NPT_LOG_INFO("too many arguments");
goto done;
}
}
cmd++;
}
for (i = 0; i < sizeof(cmds) / sizeof(cmds[0]); i++) {
if (!strcmp(cmds[i].name,arg[0])) {
if (n != cmds[i].numargs) {
NPT_LOG_INFO("numargs doesnot equal requires arguments ");
} else {
ret = cmds[i].func(arg + 1, reply);
}
goto done;
}
}
NPT_LOG_INFO("unsupported command");
done:
if (reply[0]) {
n = snprintf(cmd, BUFFER_MAX, "%d %s", ret, reply);
} else {
n = snprintf(cmd, BUFFER_MAX, "%d", ret);
}
if (n > BUFFER_MAX) n = BUFFER_MAX;
count = n;
/* receiver expects count in little-endian order */
le_count = htole16(count);
// LOGI("reply: '%s'\n", cmd);
if (writex(s, &le_count, sizeof(le_count))) return -1;
if (writex(s, cmd, count)) return -1;
return 0;
}
int
main(int argc, char** argv)
{
//setup Neptune logging
NPT_LogManager::GetDefault().Configure("plist:.level=INFO;.handlers=ConsoleHandler;.ConsoleHandler.colors=off;.ConsoleHandler.filter=42");
NPT_LOG_INFO("baojinyu---------------media server just begin\n ");
init();
char buf[BUFFER_MAX];
struct sockaddr_in server_sockaddr, client_sockaddr;
int sin_size = sizeof(struct sockaddr_in);
/* socket配置 */
bzero(&server_sockaddr, sizeof(server_sockaddr));
server_sockaddr.sin_family = AF_INET; /* 地址族 */
server_sockaddr.sin_port = htons(7933);/* 端口号 */
//server_sockaddr.sin_addr.s_addr = inet_addr("127.0.0.1");/* 系统会自动填入本机的IP地址 */
server_sockaddr.sin_addr.s_addr = INADDR_ANY;
//bzero(&(server_sockaddr.sin_zero),8);/*填充0和struct sockaddr 同样大小*/
int i = 1; /* 允许重复使用本地地址与套接字进行绑定 */
setsockopt(sockfd,SOL_SOCKET,SO_REUSEADDR, (const char*)&i, sizeof(i));
//if(bind(sockfd,(struct sockaddr*)&server_sockaddr,len) < 0){
if(bind(sockfd,(struct sockaddr*)&server_sockaddr, sin_size) == -1){
NPT_LOG_INFO("baojinyu---------------Bind failure.\n ");
exit(1);
}
NPT_LOG_INFO("baojinyu--------------- Bind successfully.\n ");
/* listen for connections on a socket */
if(listen(sockfd, MAX_QUE_CONN_NM) == -1){
NPT_LOG_INFO("baojinyu--------------- Listen failure.\n ");
exit(1);
}
NPT_LOG_INFO("baojinyu--------------- Listen successfully ");
int client_fd, fd;
int result;
int readflag;
int maxfd=sockfd;
fd_set tmp_readfds;
fd_set readfds;
FD_ZERO(&readfds);
FD_SET(sockfd, &readfds);
fd_set tmp_writefds;
fd_set writefds;
FD_ZERO(&writefds);
FD_SET(sockfd, &writefds);
struct timeval timeout;
fd_set fdset;
while(true)
{
tmp_readfds = readfds;
tmp_writefds = writefds;
NPT_LOG_INFO("baojinyu--------------- enter while loop ");
timeout={5,0};
result = select(maxfd+1, &tmp_readfds, &tmp_writefds, NULL, &timeout);
if(result == 0)
{
NPT_LOG_INFO("baojinyu----------------------Select timeout!---no change happens.\n");
continue;
}else if(result == -1)
{
NPT_LOG_INFO("baojinyu----------------------Select failure!\n");
break;
}
NPT_LOG_INFO("baojinyu-----------------Select successfully\n");
for(fd=0; fd < FD_SETSIZE; fd++)
{
if(FD_ISSET(fd,&tmp_readfds))
{
NPT_LOG_INFO("baojinyu-----------------FD_ISSET\n");
if(fd == sockfd)
{ //服务端接收客户端连接请求
NPT_LOG_INFO("baojinyu-----------------fd == sockfd\n");
client_fd = accept(sockfd, (struct sockaddr*)&client_sockaddr, &sin_size);
sleep(1);
if(client_fd > 0)
{
if(client_fd > maxfd){
maxfd = client_fd;
}
FD_SET(client_fd,&readfds);
/*
NPT_Debug("baojinyu--accept success--------Got connection from %s, port %d,socket %d\n",
inet_ntoa(client_sockaddr.sin_addr),ntohs(client_sockaddr.sin_port),client_fd);
* */
printf("baojinyu--accept success--------Got connection from %s, port %d,socket %d\n",
inet_ntoa(client_sockaddr.sin_addr), ntohs(client_sockaddr.sin_port), client_fd);
}else {
NPT_LOG_INFO("baojinyu-----------------accept failure.\n");
break;
}
}else{
ioctl(fd,FIONREAD,&readflag);
if(readflag == 0){
NPT_LOG_INFO("baojinyu----------------readflag == 0 \n");
close(fd);
FD_CLR(fd,&readfds);
}
else{
bzero(buf, BUFFER_MAX);
//处理客户端发来的消息-read size
unsigned short count;
if (readx(fd, &count, sizeof(count))){
NPT_LOG_INFO("baojinyu-------------------failed to read size\n");
break;
}
count = letoh16(count);
if ((count < 1) || (count >= BUFFER_MAX)) {
//NPT_LOG_INFO("baojinyu-------------------count error!\n");
printf("baojinyu--------count error!--count= %d\n",count);
break;
}
if (readx(fd, buf, count)) {
NPT_LOG_INFO("baojinyu----------------failed to read command\n");
break;
}
buf[count] = 0;
if (execute(fd, buf)) break;
}
}
} //end of if(FD_ISSET(fd,&tmp_readfds))
} //end for
}
for(fd = 0; fd < FD_SETSIZE; fd++)
{
close(fd);
FD_CLR(fd,&readfds);
}
return 0;
}