C++文件上传、下载工具
更新时间:2020年4月25日 17:35 点击:1460
本文实例为大家分享了C++文件上传下载的实现代码,供大家参考,具体内容如下
#include <stdio.h> #include <stdlib.h> #include <string.h> #include <sys/timeb.h> #include <sys/ioctl.h> #include <string.h> #include <fcntl.h> #include <sys/wait.h> #include <sys/socket.h> #include <errno.h> #include <sys/types.h> #include <sys/stat.h> #include <unistd.h> #include <netinet/in.h> #include <netinet/tcp.h> #include <arpa/inet.h> #ifndef __TCPFILE_C__ #define __TCPFILE_C__ #include "libfunc.h" #include "vapi_log.h" #define CMDHEAD 6 // 通信报文头信息长度 #define CMDINFO 6 // 空值命令 PUT GET #define CMD #define MAXKB 16 #define NCMD 32 #define NFILE 128 // tcp 通信命令行缓冲区 typedef struct tcpCmd { int len; // 命令行长度 char cmd[NCMD+1]; // 命令行缓冲区 } tcpCmd; struct tcpBuffer { int rlen; // 接收数据长度 int wlen; // 发送数据长度 char rcvcmd[NCMD+1]; // 接收命令域数据 char sndcmd[NCMD+1]; // 接收命令域数据 tcpCmd rCmd ; // 接收缓冲区 tcpCmd sCmd ; // 发送缓冲区 char buff[1024 * MAXKB + 64 + 1 ]; // 报文缓冲区, 包含命令控制串和实际报文数据 } ncb; // ////////////////////////////////////////////////////////////////////////////////////////////// //// // // // 根据报文头数据要求,接收一个通信数据。 // // 程序首先按照要求读取 headlen 长度的长度包数据,然后再次从网络上读取真正长度的数据包。 // // // // 数据接收函数分两次进行处理,返回数据 sData 中已经不再包含 6位通信长度数据。 // // ///////////////////////////////////////////////////////////////////////////////////////////// //// int tcp_readbuf(int headlen, int sfd, char * sData, int MaxLen, int sTime) { int iRet = 0; int left_bytes, thisRead, sLen; char *ptr = sData; struct timeval tv; fd_set rfds; char temp[ NCMD + 1 ]; tv.tv_sec = (long )sTime ; tv.tv_usec = 0; FD_ZERO(&rfds); FD_SET(sfd, &rfds); do{ iRet = select(sfd+1, &rfds, NULL, NULL, &tv) ; }while (iRet < 0 && errno == EINTR ); if(iRet == 0){ wLog(LOGERROR,"tcp_readbuf select 延时[%d] 结束, faile [%d, %s]", sTime, errno, strerror(errno) ); return -1; } // 接受控制命令序列 memset(temp, 0x00 , sizeof (temp)); thisRead = read(sfd, temp, headlen); if( temp[0]=='0' )sLen = atoi(temp); else sLen = 0; if(thisRead != headlen && sLen ){ wLog(LOGERROR,"读取通信报文长度[%s]失败, faile [%d, %s]", temp, errno, strerror(errno) ); return -1; } if(sLen < 1 || sLen > MaxLen ){ if(sLen > MaxLen ) wLog(LOGERROR,"报文长度[%s]错误,数据非法. ", temp ); return -1; } left_bytes = sLen; while( left_bytes > 0 ){ if( (thisRead = read(sfd, ptr, left_bytes)) == 0) break ; if(thisRead < 0 ){ if( errno == EINTR ) continue; break; } left_bytes -= thisRead; ptr += thisRead; } if(left_bytes && ptr != sData ) wLog(LOGERROR,"[tcp_readbuf [%d] faile [%d, %s]", sLen, errno, strerror(errno) ); /* 数据没有处理完时,程序打印错误日志信息 */ return(sLen-left_bytes); } // // 数据发送程序,在指定的延时内将指定长度的数据包发送到 sfd 上。 // 发送数据需要将报文长度保存在 sData 中,发送长度比实际的报文多出长度域 6字节 // int tcp_writebuf(int sfd, char * sData, int sLen, int sTime) { int iRet = 0; int left_bytes, thisWrite; char *ptr = sData; fd_set wfds; struct timeval tv; FD_ZERO(&wfds); FD_SET(sfd, &wfds); do{ iRet = select(sfd+1, NULL, &wfds, NULL, &tv) ; }while (iRet < 0 && errno == EINTR ); if(iRet==0){ wLog(LOGERROR,"tcp_writebuf select 延时[%d] 结束, faile [%d, %s]", sTime, errno, strerror(errno) ); return -1; } // 检查通信链路的 写状态 left_bytes=sLen; while(left_bytes >0 ){ if( (thisWrite = write(sfd, ptr, left_bytes)) == 0) break ; if(thisWrite < 0 ){ if( errno == EINTR ) continue; break; } left_bytes -= thisWrite; ptr += thisWrite; } // 将数据发送到通信端口 if(left_bytes && ptr != sData ) wLog(LOGERROR,"[tcp_sendbuf left_bytes[%d] faile [%d, %s]", left_bytes, errno, strerror(errno) ); return(sLen-left_bytes); } // ============================= 客户端使用 文件发送程序 ================================= //// // // // socket 方式文件发送程序,根据输入的 netinfo 建立通信通道,然后按照以下要求发送文件 // // PUT100 发送文件基本信息 // // PUT200 发送文件内容,根据要求循环执行,...... // // PUT300 数据发送结束 // // // // ======================================================================================= //// int cli_put_sendFile(char *netinfo, char *localFile, char *remoteFile, int blockSize, int timeout) { int sfd, fd; struct stat sb; int iRet ; int chkflg = 0 ; int maxBlk, blknum = 0; long start; long fsize; sfd = fd = -1; start = time(NULL); sfd = make_tcpConnect (netinfo); // 申请 socket 描述符,使用 connect () 建立到服务器的连接通道 if(sfd < 0 ) { wLog(LOGERROR,"建立到[%s]连接失败 error [%d, %s]", netinfo , errno, strerror(errno) ); return -4; } wLog(LOGINFO, "成功建立到[%s]发送通道[%d]", netinfo, sfd ); fd = open(localFile, O_RDONLY); if(fd == -1){ wLog(LOGERROR,"本地文件[%s]打开失败 error [%d, %s]", localFile , errno, strerror(errno) ); close (sfd ); return -3; } if (fstat(fd, &sb) < 0) { wLog(LOGERROR,"取[%s]文件信息失败 error [%d, %s]", localFile , errno, strerror(errno) ); chkflg = -3; goto cli_put_sendFile_END; } fsize = sb.st_size; if(blockSize > 1024 * MAXKB ) blockSize = 1024 * MAXKB ; if(blockSize < 1024 ) blockSize = 1024; // 显示本地文件的基本信息 wLog(LOGINFO, "成功打开本地文件[%s], size[%ld] ", localFile, fsize ); maxBlk = (int ) ( (fsize ) / blockSize) ; // 计算本文件的最大传输次数 if( fsize % blockSize ) maxBlk += 1; // 不足整块的数据,需要按一块进行处理。 memset(&ncb , 0x00 , sizeof (struct tcpBuffer)); // 准备发送文件控制命令串, 告诉对方准备发送文件的基础信息 ncb.wlen = snprintf(ncb.buff+25, sizeof(ncb.buff)-1, "%s:%010ld:%010d:%010d", remoteFile, fsize , blockSize , maxBlk ); sprintf(ncb.sndcmd , "%06d%-16s%1s%2s", ncb.wlen+19, "PUT100FILEINFO", "1", "00"); memcpy(ncb.buff, ncb.sndcmd, 25); ncb.wlen += 25; iRet = tcp_writebuf (sfd , ncb.buff, ncb.wlen, timeout); if(iRet != ncb.wlen){ wLog(LOGERROR,"发送[%d] [%s]失败 error [%d, %s]", ncb.wlen, remoteFile , errno, strerror(errno) ); chkflg = -2; goto cli_put_sendFile_END; } wLog(LOGINFO,"发送报文头[%d] [%s]成功", ncb.wlen , ncb.buff ); ncb.rlen = tcp_readbuf(6, sfd, ncb.rcvcmd , 19, timeout); if(ncb.rlen != 19 || memcmp(ncb.rcvcmd+17, "00", 2)){ wLog(LOGERROR,"远程保存[%s]失败 error [%d, %s]", remoteFile , errno, strerror(errno) ); chkflg = -2; goto cli_put_sendFile_END; } wLog(LOGINFO,"接到返回数据 [%s]成功", ncb.rcvcmd ); // 循环将本地文件全部发送完毕 while ( 1 ){ blknum ++; memset(&ncb , 0x00 , sizeof (struct tcpBuffer )); ncb.rlen = read (fd, ncb.buff+25, blockSize ); if(ncb.rlen < 1 ) break ; // 本地文件已经发送结束 sprintf (ncb.sndcmd , "%06dPUT200BLK%07d%1s%2s", ncb.rlen+19, blknum, "1", "00"); memcpy(ncb.buff, ncb.sndcmd , 25); ncb.rlen += 25; iRet = tcp_writebuf (sfd , ncb.buff, ncb.rlen, timeout); if(iRet != ncb.rlen ){ wLog(LOGERROR,"发送 [%s] 失败 error [%d, %s]", ncb.sndcmd , errno, strerror(errno) ); chkflg = -1; goto cli_put_sendFile_END; } if( blknum == 1 || blknum == maxBlk ) wLog(LOGINFO,"发送数据[%d] [%s]成功", ncb.rlen, ncb.sndcmd ); iRet = tcp_readbuf(6, sfd, ncb.rcvcmd , 19, timeout); if(iRet != 19 || memcmp(ncb.rcvcmd+ 17, "00", 2)){ wLog(LOGERROR,"远程接收 [%s] 失败 error [%d, %s]", ncb.rcvcmd, errno, strerror(errno) ); chkflg = -1; goto cli_put_sendFile_END; } } memset(&ncb , 0x00 , sizeof (struct tcpBuffer )); ncb.wlen = snprintf (ncb.sndcmd , sizeof (ncb.sndcmd )-1, "%-16s%1s%2s", "PUT300FILEOVER", "1", "00" ); sprintf(ncb.buff, "%06d%s", ncb.wlen, ncb.sndcmd ); iRet = tcp_writebuf (sfd , ncb.buff, ncb.wlen+6, timeout); if(iRet != ncb.wlen + 6 ){ wLog(LOGERROR,"发送 FileOver 失败 error [%d, %s]", errno, strerror(errno) ); chkflg = -1; goto cli_put_sendFile_END; } wLog(LOGINFO,"发送数据[%d] [%s]成功", iRet , ncb.sndcmd ); iRet = tcp_readbuf(6, sfd, ncb.rcvcmd , 19, timeout); if(iRet != 19 || memcmp(ncb.rcvcmd+17, "00", 2)){ wLog(LOGERROR,"远程接收 FileOver 失败 error [%d, %s]", errno, strerror(errno) ); chkflg = -1; goto cli_put_sendFile_END; } wLog(LOGINFO, "接到返回数据[%s]成功", ncb.rcvcmd ); wLog(LOGINFO, "传输[%s]-->[%s] [%d]块,共 [%ld]字节, 耗时 %ld 秒\n\n", localFile , remoteFile , blknum, fsize , time(NULL) - start ); cli_put_sendFile_END: if(sfd > 0) close(sfd); if(fd > 0) close(fd); // 关闭本文描述符和通信连接通道描述符 return chkflg; } // ============================ 服务端使用 文件传输程序 ================================= //// // // // socket 文件接收服务程序,对收到的通信报文进行分析。 // // 在收到 PUT100 命令后,将该数据域中的文件名称与本地路径拼写得到完整的文件 // // 路径信息,打开本地文件。 // // 对 PUT200 数据域传输来的数据保存到本地文件中。 // // 收到 PUT300 命令,关闭本地文件,传输过程结束。 // // // // 文件传输服务端程序: // // sfd 在 accept() 后获取的新的客户端连接通道描述符 // // path 准备保存本地文件的路径信息 // // fileName 根据接收报文中的文件名称与本地路径拼串而得到的返回文件信息 // // timeout 数据传输需要使用的延时参数 // // // // 返回数据: // // 0 ---- 文件接收成功 // // -2 -- 文件无法创建,打开文件名称失败 // // -1 -- 文件内容内容保存失败 // // ====================================================================================== //// int srv_put_recvFile(int sfd , char *path, char *fileName, int timeout ) { int fd = -1; int blknum = 0, maxBlk=0 ; char tfile[NFILE+1], bfile[NFILE+1]; char *ptr; long fsize = 0; int chkflg = 0; memset(tfile, 0x00, sizeof(tfile)); while ( 1 ){ memset(&ncb, 0x00, sizeof (struct tcpBuffer)); ncb.rlen = tcp_readbuf (6, sfd, ncb.buff, sizeof(ncb.buff)-1, timeout); if(ncb.rlen < 0 ) break ; memcpy(ncb.sndcmd + 6, ncb.buff, 19); if(memcmp(ncb.buff, "PUT", 3) ) { wLog(LOGERROR,"接收命令序列 [%s] 错误 ,程序退出. ", ncb.sndcmd+6 ); memcpy(ncb.rcvcmd+23, "01", 2); chkflg = -3; } switch (Nstr_int(ncb.buff + 3, 3)){ // 获取 PUT 后面的命令控制字,下面分析该控制字进行工作 case 100 : // 开始接收文件,打开本地文件 wLog(LOGINFO, "接收管理报文[%s]成功", ncb.buff ); // 对于接收到的第一条命令,打印该控制命令的全部内容 ptr = strchr(ncb.buff, ':'); if(ptr) memcpy(tfile, ncb.buff+19, ptr - (char *)ncb.buff - 19); else strcpy(tfile, ncb.buff+19); // 获取传输来的文件名称 ptr = strrchr(tfile, '/'); if(ptr) strcpy(bfile, ptr+1); else strcpy(bfile, tfile); // 检查传输来文件名称中的路径信息,得到基本文件名称,将前面的路径 // 信息全部剔除,以保证本地文件的安全。 if( (ptr = strrchr(ncb.buff , ':') ) != NULL ) maxBlk = atoi (ptr +1); if( path ) sprintf(fileName, "%s/%s", path, bfile); else strcpy(fileName, bfile ); // 与本地保存路径拼串,获得本地文件名称 fd = open(fileName, O_CREAT|O_WRONLY|O_TRUNC, 0666 ); if(fd < 0 ) { wLog(LOGERROR,"生成本地文件 [%s] 失败 error [%d, %s]", fileName , errno, strerror(errno) ); memcpy(ncb.rcvcmd+23, "01", 2); chkflg = -2; } // 对需要保存的本地文件,使用清空方式,创建新文件,既是该文件已经存在 // 也可以保证数据处理 wLog(LOGINFO,"创建本地文件[%s]成功", fileName ); break ; case 200 : // 保存文件内容 blknum ++; maxBlk --; if(blknum == 1 || !maxBlk )wLog(LOGINFO, "接收数据[%s]成功", ncb.sndcmd+6 ); ncb.wlen = write(fd, ncb.buff+19, ncb.rlen - 19); if(ncb.wlen != ncb.rlen - 19 ) { memcpy(ncb.sndcmd + 23, "01", 2); chkflg = -1; } else fsize += ncb.wlen ; break ; case 300 : // 文件传输结束 if( !maxBlk ) wLog(LOGINFO, " 文件[%s]成功接收,共 [%d] 传输块 ", fileName, blknum ); else wLog(LOGERROR,"文件[%s]接收结束,差错 [%d] 传输块 ", fileName , maxBlk ); close(fd); chkflg = 1; break ; } memcpy(ncb.sndcmd , "000019", 6); ncb.sndcmd [22]='2'; ncb.wlen = tcp_writebuf (sfd , ncb.sndcmd , 25, timeout); if(ncb.wlen != 25 ){ wLog(LOGERROR,"发送返回信息 [%s] 失败 error [%d, %s]", ncb.sndcmd + 6, errno, strerror(errno) ); } if(chkflg ) break ; } if(fd) close (fd); wLog(LOGINFO,"成功接收[%s]文件, 共 [%ld] 字节", fileName , fsize ); return chkflg ; } // ============================= 客户端使用 多文件发送程序 =============================== //// // // // socket 方式文件发送程序,根据输入的 netinfo 建立通信通道,然后按照以下要求发送文件 // // PUT100 发送文件基本信息 // // PUT200 发送文件内容,根据要求循环执行,...... // // PUT300 数据发送结束 // // // // ======================================================================================= //// // // 在建立好的文件传输通道上,将一个文件数据发送到服务端,传输后,不需要关闭传输通道。 // int cli_mput_sendFile(int sfd, char *localFile, char *remoteFile, int blockSize, int timeout ) { int fd; struct stat sb; int iRet ; int chkflg = 0 ; int maxBlk, blknum = 0; char *ftr; long start; long fsize; fd = -1; start = time(NULL); fd = open(localFile, O_RDONLY); if(fd == -1){ wLog(LOGERROR,"本地文件[%s]打开失败 error [%d, %s]", localFile , errno, strerror(errno) ); close (sfd ); return -3; } if (fstat(fd, &sb) < 0) { wLog(LOGERROR,"取[%s]基本信息失败 error [%d, %s]", localFile , errno, strerror(errno) ); chkflg = -3; goto mSend_END; } fsize = sb.st_size; if(blockSize > 1024 * MAXKB ) blockSize = 1024 * MAXKB ; if(blockSize < 1024 ) blockSize = 1024; wLog(LOGINFO, "成功打开本地文件[%s], size[%ld] ", localFile, fsize ); maxBlk = (int ) ( (fsize ) / blockSize) ; // 计算本文件的最大传输次数 if( fsize % blockSize ) maxBlk += 1; memset(&ncb , 0x00 , sizeof (struct tcpBuffer)); ftr = strrchr( remoteFile, '/'); ncb.wlen = snprintf(ncb.buff+25, sizeof(ncb.buff)-1, "%s:%010ld:%010d:%010d", ftr ? ftr +1 : remoteFile, fsize , blockSize , maxBlk ); sprintf(ncb.sndcmd , "%06d%-16s%1s%2s", ncb.wlen+19, "PUT100FILEINFO", "1", "00"); memcpy(ncb.buff, ncb.sndcmd, 25); ncb.wlen += 25; iRet = tcp_writebuf (sfd , ncb.buff, ncb.wlen, timeout); if(iRet != ncb.wlen){ wLog(LOGERROR,"发送[%d] [%s]失败 error [%d, %s]", ncb.wlen, remoteFile , errno, strerror(errno) ); chkflg = -2; goto mSend_END; } wLog(LOGINFO,"发送报文头[%d] [%s]成功", ncb.wlen , ncb.buff ); ncb.rlen = tcp_readbuf(6, sfd, ncb.rcvcmd , 19, timeout); if(ncb.rlen != 19 || memcmp(ncb.rcvcmd+17, "00", 2)){ wLog(LOGERROR,"远程保存[%s]失败 error [%d, %s]", remoteFile , errno, strerror(errno) ); chkflg = -2; goto mSend_END; } wLog(LOGINFO,"接到返回数据 [%s]成功", ncb.rcvcmd ); while ( 1 ){ blknum ++; memset(&ncb , 0x00 , sizeof (struct tcpBuffer )); ncb.rlen = read (fd, ncb.buff+25, blockSize ); if(ncb.rlen < 1 ) break ; // 本地文件已经发送结束 sprintf (ncb.sndcmd , "%06dPUT200BLK%07d%1s%2s", ncb.rlen+19, blknum, "1", "00"); memcpy(ncb.buff, ncb.sndcmd , 25); ncb.rlen += 25; iRet = tcp_writebuf (sfd , ncb.buff, ncb.rlen, timeout); if(iRet != ncb.rlen ){ wLog(LOGERROR,"发送 [%s] 失败 error [%d, %s]", ncb.sndcmd , errno, strerror(errno) ); chkflg = -1; goto mSend_END; } if( blknum == 1 || blknum == maxBlk ) wLog(LOGINFO,"发送数据[%d] [%s]成功", ncb.rlen, ncb.sndcmd ); iRet = tcp_readbuf(6, sfd, ncb.rcvcmd , 19, timeout); if(iRet != 19 || memcmp(ncb.rcvcmd+ 17, "00", 2)){ wLog(LOGERROR,"远程接收 [%s] 失败 error [%d, %s]", ncb.rcvcmd, errno, strerror(errno) ); chkflg = -1; goto mSend_END; } } memset(&ncb , 0x00 , sizeof (struct tcpBuffer )); ncb.wlen = snprintf (ncb.sndcmd , sizeof (ncb.sndcmd )-1, "%-16s%1s%2s", "PUT300FILEOVER", "1", "00" ); sprintf(ncb.buff, "%06d%s", ncb.wlen, ncb.sndcmd ); iRet = tcp_writebuf (sfd , ncb.buff, ncb.wlen+6, timeout); if(iRet != ncb.wlen + 6 ){ wLog(LOGERROR,"发送 FileOver 失败 error [%d, %s]", errno, strerror(errno) ); chkflg = -1; goto mSend_END; } wLog(LOGINFO,"发送数据[%d] [%s]成功", iRet , ncb.sndcmd ); iRet = tcp_readbuf(6, sfd, ncb.rcvcmd , 19, timeout); if(iRet != 19 || memcmp(ncb.rcvcmd+17, "00", 2)){ wLog(LOGERROR,"远程接收 FileOver 失败 error [%d, %s]", errno, strerror(errno) ); chkflg = -1; goto mSend_END; } wLog(LOGINFO, "接到返回数据[%s]成功", ncb.rcvcmd ); wLog(LOGINFO, "传输[%s]-->[%s] [%d]块,共 [%ld]字节, 耗时 %ld 秒\n\n", localFile , remoteFile , blknum, fsize , time(NULL) - start ); mSend_END: if(fd > 0) close(fd); // 关闭本文描述符和通信连接通道描述符 return chkflg; } // ============================================= 多文件处理函数 ====================================== // // // // // // 多文件发送服务程序,本程序对使用 ":" 分隔的文件信息自动进行分解,然后将每一个文件 // // 使用上述函数完成推送工作,在啊全部文件发送完毕后,程序将发出 PUT500 传输结束命令。 // // // ///////////////////////////////////////////////////////////////////////////////////////////////////////// int cli_putm_sendFile(char *netinfo, char *sLocalFile, int blockSize, int timeout) { int sfd; int iRet ; int chkflg = 0 ; struct cli_putm_sendFile{ char lFile[NFILE+1]; // 本地文件名称 char rFile[NFILE+1]; // 远程文件名称 }mSnd; char *qlist , *ptr ; char *ftr; int fnum = 0; long start ; sfd = -1; start = time(NULL); sfd = make_tcpConnect (netinfo); // 使用 connect () 建立到服务器的连接通道 if(sfd < 0 ) { wLog(LOGERROR,"建立到[%s]文件传输通道失败 error [%d, %s]", netinfo , errno, strerror(errno) ); return -4; } wLog(LOGINFO, "成功建立到[%s]发送通道[%d]", netinfo, sfd ); qlist = sLocalFile; ptr = strchr(qlist, ':'); while(qlist != NULL) { memset(&mSnd, 0x00, sizeof(mSnd)); strncpy(mSnd.lFile , qlist, ptr - qlist); ftr = strrchr(mSnd.lFile, '/'); strcpy(mSnd.rFile , ftr ? ftr + 1 : mSnd.lFile ); iRet = cli_mput_sendFile(sfd, mSnd.lFile, mSnd.rFile, blockSize, timeout ); if( iRet == 0 ) fnum ++ ; qlist = ptr + 1; ptr = strchr(qlist, ':'); } // 对输入的文件名称进行分解,调用 cli_mput_sendFile() 函数进行数据发送。 memset(&ncb , 0x00 , sizeof (struct tcpBuffer )); ncb.wlen = snprintf (ncb.sndcmd , sizeof (ncb.sndcmd )-1, "%-16s%1s%2s", "PUT500FTPOVER", "1", "00" ); sprintf(ncb.buff, "%06d%s", ncb.wlen, ncb.sndcmd ); iRet = tcp_writebuf (sfd , ncb.buff, ncb.wlen+6, timeout); if(iRet != ncb.wlen + 6 ) { wLog(LOGERROR,"发送 FileOver 失败 error [%d, %s]", errno, strerror(errno) ); chkflg = -1; goto cli_put_sendFile_END; } wLog(LOGINFO,"发送数据[%d] [%s]成功", iRet , ncb.sndcmd ); iRet = tcp_readbuf(6, sfd, ncb.rcvcmd , 19, timeout); if(iRet != 19 || memcmp(ncb.rcvcmd+17, "00", 2)) { wLog(LOGERROR,"远程接收 FileOver 失败 error [%d, %s]", errno, strerror(errno) ); chkflg = -1; goto cli_put_sendFile_END; } wLog(LOGINFO, "接到返回数据[%s]成功", ncb.rcvcmd ); wLog(LOGINFO, "共 [%d]个文件, 耗时 %ld 秒\n\n", fnum , time(NULL) - start ); cli_put_sendFile_END: if(sfd > 0) close(sfd); // 关闭本文描述符和通信连接通道描述符 return chkflg; } // ============================ 服务端使用 多文件传输程序 =============================== //// // // // socket 文件接收服务程序,对收到的通信报文进行分析。 // // 在收到 PUT100 命令后,将该数据域中的文件名称与本地路径拼写得到完整的文件 // // 路径信息,打开本地文件。 // // 对 PUT200 数据域传输来的数据保存到本地文件中。 // // 收到 PUT300 命令,关闭当前传输文件。 // // 收到 PUT500 命令,TCP 文件传输过程结束,退出程序。 // // // // 文件传输服务端程序: // // sfd 在 accept() 后获取的新的客户端连接通道描述符 // // path 准备保存本地文件的路径信息 // // fileName 根据接收报文中的文件名称与本地路径拼串而得到的返回文件信息 // // timeout 数据传输需要使用的延时参数 // // // // 返回数据: // // 0 ---- 文件接收成功 // // -2 -- 文件无法创建,打开文件名称失败 // // -1 -- 文件内容内容保存失败 // // // // 文件传输过程中,上传的文件名称有客户端提供,但是文件保存路径由服务器控制,以充分保证 // // 服务器文件系统的安全,避免服务器上的文件被客户端上传文件恶意覆盖。 // // // // ====================================================================================== //// int srv_mput_recvFile(int sfd , char *path, int timeout ) { int fd = -1; int blknum = 0 , maxBlk = 0 ; char localFile[NFILE+1]; char tfile[NFILE+1], bfile[NFILE+1]; char *ptr; long fsize = 0; long start = time(NULL); int chkflg = 0; int fnum = 0; memset(tfile, 0x00, sizeof(tfile)); while (1){ memset(&ncb, 0x00, sizeof (struct tcpBuffer)); ncb.rlen = tcp_readbuf (6, sfd, ncb.buff, sizeof(ncb.buff)-1, timeout); if(ncb.rlen < 0 ) break ; memcpy(ncb.sndcmd + 6, ncb.buff, 19); if(memcmp(ncb.buff, "PUT", 3)) { wLog(LOGERROR,"接收命令序列 [%s] 错误 ,程序退出. ", ncb.sndcmd+6 ); memcpy(ncb.rcvcmd+23, "01", 2); chkflg = -3; } switch (Nstr_int(ncb.buff + 3, 3)){ // 获取 PUT 后面的命令控制字,下面分析该控制字进行工作 case 100 : // 开始接收文件,打开本地文件 blknum = maxBlk = 0 ; wLog(LOGINFO, "接收管理报文[%s]成功", ncb.buff ); // 对于接收到的第一条命令,打印该控制命令的全部内容 ptr = strchr(ncb.buff, ':'); if(ptr) memcpy(tfile, ncb.buff+19, ptr - (char *)ncb.buff - 19); else strcpy(tfile, ncb.buff+19); // 获取传输来的文件名称 ptr = strrchr(tfile, '/'); if(ptr) strcpy(bfile, ptr+1); else strcpy(bfile, tfile); // 检查传输来文件名称中的路径信息,得到基本文件名称,将前面的路径 // 信息全部剔除,以保证本地文件的安全。当传输文件名称带有 ".." 标志 // 的时候,将会对服务器文件系统产生影响,需要保证该问题不会出现。 if( (ptr = strrchr(ncb.buff , ':') ) != NULL ) maxBlk = atoi (ptr +1); // 从命令报文中得到文件传输块信息。 if( path ) sprintf(localFile, "%s/%s", path, bfile); else strcpy(localFile, bfile ); // 与本地保存路径拼串,获得本地文件名称 fd = open(localFile, O_CREAT|O_WRONLY|O_TRUNC, 0666 ); if(fd < 0 ) { wLog(LOGERROR,"生成本地文件 [%s] 失败 error [%d, %s]", localFile , errno, strerror(errno) ); memcpy(ncb.rcvcmd+23, "01", 2); chkflg = -2; } fnum ++; // 对需要保存的本地文件,使用清空方式,创建新文件,既是该文件已经存在 // 也可以保证数据处理 wLog(LOGINFO,"创建本地文件[%s]成功", localFile ); break ; case 200 : // 保存文件内容 blknum ++; maxBlk --; if(blknum == 1 || !maxBlk )wLog(LOGINFO, "接收数据[%s]成功", ncb.sndcmd+6 ); ncb.wlen = write(fd, ncb.buff+19, ncb.rlen - 19); if(ncb.wlen != ncb.rlen - 19 ) { memcpy(ncb.sndcmd + 23, "01", 2); chkflg = -1; } else fsize += ncb.wlen ; break ; case 300 : // 文件传输结束 if( !maxBlk ) wLog(LOGINFO, " 文件[%s]成功接收,共 [%d] 传输块 ", localFile, blknum ); else wLog(LOGERROR,"文件[%s]接收结束,差错 [%d] 传输块 ", localFile , maxBlk ); close(fd); break ; case 500 : // 通信处理结束 chkflg = 1; break ; } memcpy(ncb.sndcmd , "000019", 6); ncb.sndcmd [22]='2'; ncb.wlen = tcp_writebuf (sfd , ncb.sndcmd , 25, timeout); if(ncb.wlen != 25 ){ wLog(LOGERROR,"发送返回信息 [%s] 失败 error [%d, %s]", ncb.sndcmd + 6, errno, strerror(errno) ); } if(chkflg ) break ; } if(fd) close (fd); wLog(LOGINFO, "成功接收[%d]文件, 耗时 [%ld] 秒 ", fnum , time(NULL) - start ); return chkflg ; } // ================================================================================================== // ///////////////////////////////////以下为客户端主动下载类程序/////////////////////////////////////////// // ============================= 客户端使用 文件发送程序 ================================= //// // // // socket 方式文件发送程序,根据输入的 netinfo 建立通信通道,然后按照以下要求发送文件 // // GET100 发送文件下载请求,将远程文件名称和分块尺寸上送主机,等主机回应 // // GET200 接收文件内容,根据要求循环执行,...... // // GET300 数据发送结束 // // // // ======================================================================================= //// int cli_get_sendFile(char *netinfo, char *localFile, char *remoteFile, int blockSize, int timeout) { int sfd, fd; struct stat sb; int iRet ; int chkflg = 0 ; int maxBlk, blknum = 0; long start; long fsize; sfd = fd = -1; start = time(NULL); sfd = make_tcpConnect (netinfo); // 申请 socket 描述符,使用 connect () 建立到服务器的连接通道 if(sfd < 0 ) { wLog(LOGERROR,"建立到[%s]连接失败 error [%d, %s]", netinfo , errno, strerror(errno) ); return -4; } wLog(LOGINFO, "成功建立到[%s]发送通道[%d]", netinfo, sfd ); fd = open(localFile, O_CREAT|O_WRONLY|O_TRUNC, 0666 ); if(fd == -1){ wLog(LOGERROR,"本地文件[%s]打开失败 error [%d, %s]", localFile , errno, strerror(errno) ); close (sfd ); return -3; } if (fstat(fd, &sb) < 0) { wLog(LOGERROR,"取[%s]文件信息失败 error [%d, %s]", localFile , errno, strerror(errno) ); chkflg = -3; goto cli_get_sendFile_END; } fsize = sb.st_size; if(blockSize > 1024 * MAXKB ) blockSize = 1024 * MAXKB ; if(blockSize < 1024 ) blockSize = 1024; wLog(LOGINFO, "成功打开本地文件[%s], size[%ld] ", localFile, fsize ); maxBlk = (int ) ( (fsize ) / blockSize) ; // 计算本文件的最大传输次数 if( fsize % blockSize ) maxBlk += 1; // 不足整块的数据,需要按一块进行处理。 memset(&ncb , 0x00 , sizeof (struct tcpBuffer)); ncb.wlen = snprintf(ncb.buff+25, sizeof(ncb.buff)-1, "%s:%010ld:%010d:%010d", remoteFile, 0L , blockSize , 0 ); sprintf(ncb.sndcmd , "%06d%-16s%1s%2s", ncb.wlen+19, "GET100FILEINFO", "1", "00"); memcpy(ncb.buff, ncb.sndcmd, 25); ncb.wlen += 25; iRet = tcp_writebuf (sfd , ncb.buff, ncb.wlen, timeout); if(iRet != ncb.wlen){ wLog(LOGERROR,"发送[%d] [%s]失败 error [%d, %s]", ncb.wlen, remoteFile , errno, strerror(errno) ); chkflg = -2; goto cli_get_sendFile_END; } wLog(LOGINFO, "发送报文头[%d] [%s]成功", ncb.wlen , ncb.buff ); ncb.rlen = tcp_readbuf(6, sfd, ncb.rcvcmd , 19, timeout); if(ncb.rlen != 19 || memcmp(ncb.rcvcmd+17, "00", 2)){ wLog(LOGERROR,"远程保存[%s]失败 error [%d, %s]", remoteFile , errno, strerror(errno) ); chkflg = -2; goto cli_get_sendFile_END; } wLog(LOGINFO,"接到返回数据 [%s]成功", ncb.rcvcmd ); while (1){ blknum ++; memset(&ncb , 0x00 , sizeof (struct tcpBuffer )); ncb.rlen = read (fd, ncb.buff+25, blockSize ); if(ncb.rlen < 1 ) break ; // 本地文件已经发送结束 sprintf (ncb.sndcmd , "%06dGET200BLK%07d%1s%2s", ncb.rlen+19, blknum, "1", "00"); memcpy(ncb.buff, ncb.sndcmd , 25); ncb.rlen += 25; iRet = tcp_writebuf (sfd , ncb.buff, ncb.rlen, timeout); if(iRet != ncb.rlen ){ wLog(LOGERROR,"发送 [%s] 失败 error [%d, %s]", ncb.sndcmd , errno, strerror(errno) ); chkflg = -1; goto cli_get_sendFile_END; } if( blknum == 1 || blknum == maxBlk ) wLog(LOGINFO, "发送数据[%d] [%s]成功", ncb.rlen, ncb.sndcmd ); iRet = tcp_readbuf(6, sfd, ncb.rcvcmd , 19, timeout); if(iRet != 19 || memcmp(ncb.rcvcmd+ 17, "00", 2)){ wLog(LOGERROR,"远程接收 [%s] 失败 error [%d, %s]", ncb.rcvcmd, errno, strerror(errno) ); chkflg = -1; goto cli_get_sendFile_END; } } memset(&ncb , 0x00 , sizeof (struct tcpBuffer )); ncb.wlen = snprintf (ncb.sndcmd , sizeof (ncb.sndcmd )-1, "%-16s%1s%2s", "GET300FILEOVER", "1", "00" ); sprintf(ncb.buff, "%06d%s", ncb.wlen, ncb.sndcmd ); iRet = tcp_writebuf (sfd , ncb.buff, ncb.wlen+6, timeout); if(iRet != ncb.wlen + 6 ){ wLog(LOGERROR,"发送 FileOver 失败 error [%d, %s]", errno, strerror(errno) ); chkflg = -1; goto cli_get_sendFile_END; } wLog(LOGINFO, "发送数据[%d] [%s]成功", iRet , ncb.sndcmd ); iRet = tcp_readbuf(6, sfd, ncb.rcvcmd , 19, timeout); if(iRet != 19 || memcmp(ncb.rcvcmd+17, "00", 2)){ wLog(LOGERROR,"远程接收 FileOver 失败 error [%d, %s]", errno, strerror(errno) ); chkflg = -1; goto cli_get_sendFile_END; } wLog(LOGINFO, "接到返回数据[%s]成功", ncb.rcvcmd ); wLog(LOGINFO, "传输[%s]-->[%s] [%d]块,共 [%ld]字节, 耗时 %ld 秒\n\n", localFile , remoteFile , blknum, fsize , time(NULL) - start ); cli_get_sendFile_END: if(sfd > 0) close(sfd); if(fd > 0) close(fd); // 关闭本文描述符和通信连接通道描述符 return chkflg; } // ============================ 服务端使用 文件传输程序 ================================= //// // // // socket 文件接收服务程序,对收到的通信报文进行分析。 // // 在收到 GET100 命令后,将该数据域中的文件名称与本地路径拼写得到完整的文件 // // 路径信息,打开本地文件。 // // 对 GET200 数据域传输来的数据保存到本地文件中。 // // 收到 GET300 命令,关闭本地文件,传输过程结束。 // // // // 文件传输服务端程序: // // sfd 在 accept() 后获取的新的客户端连接通道描述符 // // path 准备保存本地文件的路径信息 // // fileName 根据接收报文中的文件名称与本地路径拼串而得到的返回文件信息 // // timeout 数据传输需要使用的延时参数 // // // // 返回数据: // // 0 ---- 文件接收成功 // // -2 -- 文件无法创建,打开文件名称失败 // // -1 -- 文件内容内容保存失败 // // ====================================================================================== //// int srv_get_recvFile(int sfd , char *path, char *fileName, int timeout ) { int fd = -1; int blknum = 0, maxBlk=0 ; char tfile[NFILE+1], bfile[NFILE+1]; char *ptr; long fsize = 0; int chkflg = 0; memset(tfile, 0x00, sizeof(tfile)); while (1){ memset(&ncb, 0x00, sizeof (struct tcpBuffer)); ncb.rlen = tcp_readbuf (6, sfd, ncb.buff, sizeof(ncb.buff)-1, timeout); if(ncb.rlen < 0 ) break ; memcpy(ncb.sndcmd + 6, ncb.buff, 19); if(memcmp(ncb.buff, "GET", 3) ) { wLog(LOGERROR,"接收命令序列 [%s] 错误 ,程序退出. ", ncb.sndcmd+6 ); memcpy(ncb.rcvcmd+23, "01", 2); chkflg = -3; } switch (Nstr_int(ncb.buff + 3, 3)){ // 获取 get 后面的命令控制字,下面分析该控制字进行工作 case 100 : // 开始接收文件,打开本地文件 wLog(LOGINFO, "接收管理报文[%s]成功", ncb.buff ); // 对于接收到的第一条命令,打印该控制命令的全部内容 ptr = strchr(ncb.buff, ':'); if(ptr) memcpy(tfile, ncb.buff+19, ptr - (char *)ncb.buff - 19); else strcpy(tfile, ncb.buff+19); // 获取传输来的文件名称 ptr = strrchr(tfile, '/'); if(ptr) strcpy(bfile, ptr+1); else strcpy(bfile, tfile); // 检查传输来文件名称中的路径信息,得到基本文件名称,将前面的路径 // 信息全部剔除,以保证本地文件的安全。 if( (ptr = strrchr(ncb.buff , ':') ) != NULL ) maxBlk = atoi (ptr +1); if( path ) sprintf(fileName, "%s/%s", path, bfile); else strcpy(fileName, bfile ); // 与本地保存路径拼串,获得本地文件名称 fd = open(fileName, O_CREAT|O_WRONLY|O_TRUNC, 0666 ); if(fd < 0 ) { wLog(LOGERROR,"生成本地文件 [%s] 失败 error [%d, %s]", fileName , errno, strerror(errno) ); memcpy(ncb.rcvcmd+23, "01", 2); chkflg = -2; } // 对需要保存的本地文件,使用清空方式,创建新文件,既是该文件已经存在 // 也可以保证数据处理 wLog(LOGINFO, "创建本地文件[%s]成功", fileName ); break ; case 200 : // 保存文件内容 blknum ++; maxBlk --; if(blknum == 1 || !maxBlk )wLog(LOGINFO, "接收数据[%s]成功", ncb.sndcmd+6 ); ncb.wlen = write(fd, ncb.buff+19, ncb.rlen - 19); if(ncb.wlen != ncb.rlen - 19 ) { memcpy(ncb.sndcmd + 23, "01", 2); chkflg = -1; } else fsize += ncb.wlen ; break ; case 300 : // 文件传输结束 if( !maxBlk ) wLog(LOGINFO, " 文件[%s]成功接收,共 [%d] 传输块 ", fileName, blknum ); else wLog(LOGERROR,"文件[%s]接收结束,差错 [%d] 传输块 ", fileName , maxBlk ); close(fd); chkflg = 1; break ; } memcpy(ncb.sndcmd , "000019", 6); ncb.sndcmd [22]='2'; ncb.wlen = tcp_writebuf (sfd , ncb.sndcmd , 25, timeout); if(ncb.wlen != 25 ){ wLog(LOGERROR,"发送返回信息 [%s] 失败 error [%d, %s]", ncb.sndcmd + 6, errno, strerror(errno) ); } if(chkflg ) break ; } if(fd) close (fd); wLog(LOGINFO, "成功接收[%s]文件, 共 [%ld] 字节", fileName , fsize ); return chkflg ; } // ============================= 客户端使用 多文件发送程序 =============================== //// // // // socket 方式文件发送程序,根据输入的 netinfo 建立通信通道,然后按照以下要求发送文件 // // GET100 发送文件基本信息 // // GET200 发送文件内容,根据要求循环执行,...... // // GET300 数据发送结束 // // // // ======================================================================================= //// // // 在建立好的文件传输通道上,将一个文件数据发送到服务端,传输后,不需要关闭传输通道。 // int cli_mget_sendFile(int sfd, char *localFile, char *remoteFile, int blockSize, int timeout ) { int fd = -1; struct stat sb; int iRet ; int chkflg = 0 ; int maxBlk, blknum = 0; char *ftr; long start; long fsize; start = time(NULL); fd = open(localFile, O_RDONLY); if(fd == -1){ wLog(LOGERROR,"本地文件[%s]打开失败 error [%d, %s]", localFile , errno, strerror(errno) ); close (sfd ); return -3; } if (fstat(fd, &sb) < 0) { wLog(LOGERROR,"取[%s]基本信息失败 error [%d, %s]", localFile , errno, strerror(errno) ); chkflg = -3; goto mSend_END; } fsize = sb.st_size; if(blockSize > 1024 * MAXKB ) blockSize = 1024 * MAXKB ; if(blockSize < 1024 ) blockSize = 1024; wLog(LOGINFO,"成功打开本地文件[%s], size[%ld] ", localFile, fsize ); maxBlk = (int ) ( (fsize ) / blockSize) ; // 计算本文件的最大传输次数 if( fsize % blockSize ) maxBlk += 1; memset(&ncb , 0x00 , sizeof (struct tcpBuffer)); ftr = strrchr( remoteFile, '/'); ncb.wlen = snprintf(ncb.buff+25, sizeof(ncb.buff)-1, "%s:%010ld:%010d:%010d", ftr ? ftr +1 : remoteFile, fsize , blockSize , maxBlk ); sprintf(ncb.sndcmd , "%06d%-16s%1s%2s", ncb.wlen+19, "GET100FILEINFO", "1", "00"); memcpy(ncb.buff, ncb.sndcmd, 25); ncb.wlen += 25; iRet = tcp_writebuf (sfd , ncb.buff, ncb.wlen, timeout); if(iRet != ncb.wlen){ wLog(LOGERROR,"发送[%d] [%s]失败 error [%d, %s]", ncb.wlen, remoteFile , errno, strerror(errno) ); chkflg = -2; goto mSend_END; } wLog(LOGINFO,"发送报文头[%d] [%s]成功", ncb.wlen , ncb.buff ); ncb.rlen = tcp_readbuf(6, sfd, ncb.rcvcmd , 19, timeout); if(ncb.rlen != 19 || memcmp(ncb.rcvcmd+17, "00", 2)){ wLog(LOGERROR,"远程保存[%s]失败 error [%d, %s]", remoteFile , errno, strerror(errno) ); chkflg = -2; goto mSend_END; } wLog(LOGINFO,"接到返回数据 [%s]成功", ncb.rcvcmd ); while (1) { blknum ++; memset(&ncb , 0x00 , sizeof (struct tcpBuffer )); ncb.rlen = read (fd, ncb.buff+25, blockSize ); if(ncb.rlen < 1 ) break ; // 本地文件已经发送结束 sprintf (ncb.sndcmd , "%06dGET200BLK%07d%1s%2s", ncb.rlen+19, blknum, "1", "00"); memcpy(ncb.buff, ncb.sndcmd , 25); ncb.rlen += 25; iRet = tcp_writebuf (sfd , ncb.buff, ncb.rlen, timeout); if(iRet != ncb.rlen ){ wLog(LOGERROR,"发送 [%s] 失败 error [%d, %s]", ncb.sndcmd , errno, strerror(errno) ); chkflg = -1; goto mSend_END; } if( blknum == 1 || blknum == maxBlk ) wLog(LOGINFO,"发送数据[%d] [%s]成功", ncb.rlen, ncb.sndcmd ); iRet = tcp_readbuf(6, sfd, ncb.rcvcmd , 19, timeout); if(iRet != 19 || memcmp(ncb.rcvcmd+ 17, "00", 2)){ wLog(LOGERROR,"远程接收 [%s] 失败 error [%d, %s]", ncb.rcvcmd, errno, strerror(errno) ); chkflg = -1; goto mSend_END; } } memset(&ncb , 0x00 , sizeof (struct tcpBuffer )); ncb.wlen = snprintf (ncb.sndcmd , sizeof (ncb.sndcmd )-1, "%-16s%1s%2s", "GET300FILEOVER", "1", "00" ); sprintf(ncb.buff, "%06d%s", ncb.wlen, ncb.sndcmd ); iRet = tcp_writebuf (sfd , ncb.buff, ncb.wlen+6, timeout); if(iRet != ncb.wlen + 6 ){ wLog(LOGERROR,"发送 FileOver 失败 error [%d, %s]", errno, strerror(errno) ); chkflg = -1; goto mSend_END; } wLog(LOGINFO,"发送数据[%d] [%s]成功", iRet , ncb.sndcmd ); iRet = tcp_readbuf(6, sfd, ncb.rcvcmd , 19, timeout); if(iRet != 19 || memcmp(ncb.rcvcmd+17, "00", 2)){ wLog(LOGERROR,"远程接收 FileOver 失败 error [%d, %s]", errno, strerror(errno) ); chkflg = -1; goto mSend_END; } wLog(LOGINFO, "接到返回数据[%s]成功", ncb.rcvcmd ); wLog(LOGINFO, "传输[%s]-->[%s] [%d]块,共 [%ld]字节, 耗时 %ld 秒\n\n", localFile , remoteFile , blknum, fsize , time(NULL) - start ); mSend_END: if(fd > 0) close(fd); // 关闭本文描述符和通信连接通道描述符 return chkflg; } // ============================================= 多文件处理函数 ====================================== // // // // // // 多文件发送服务程序,本程序对使用 ":" 分隔的文件信息自动进行分解,然后将每一个文件 // // 使用上述函数完成推送工作,在啊全部文件发送完毕后,程序将发出 get500 传输结束命令。 // // // ///////////////////////////////////////////////////////////////////////////////////////////////////////// int cli_getm_sendFile(char *netinfo, char *sLocalFile, int blockSize, int timeout) { int sfd = -1; int iRet ; int chkflg = 0 ; struct cli_getm_sendFile{ char lFile[NFILE+1]; // 本地文件名称 char rFile[NFILE+1]; // 远程文件名称 }mSnd; char *qlist , *ptr ; char *ftr; int fnum = 0; long start ; start = time(NULL); sfd = make_tcpConnect (netinfo); // 使用 connect () 建立到服务器的连接通道 if(sfd < 0 ) { wLog(LOGERROR,"建立到[%s]文件传输通道失败 error [%d, %s]", netinfo , errno, strerror(errno) ); return -4; } wLog(LOGINFO, "成功建立到[%s]发送通道[%d]", netinfo, sfd ); qlist = sLocalFile; ptr = strchr(qlist, ':'); while(qlist != NULL){ memset(&mSnd, 0x00, sizeof(mSnd)); strncpy(mSnd.lFile , qlist, ptr - qlist); ftr = strrchr(mSnd.lFile, '/'); strcpy(mSnd.rFile , ftr ? ftr + 1 : mSnd.lFile ); iRet = cli_mget_sendFile(sfd, mSnd.lFile, mSnd.rFile, blockSize, timeout ); if( iRet == 0 ) fnum ++ ; qlist = ptr + 1; ptr = strchr(qlist, ':'); } // 对输入的文件名称进行分解,调用 cli_mget_sendFile() 函数进行数据发送。 memset(&ncb , 0x00 , sizeof (struct tcpBuffer )); ncb.wlen = snprintf (ncb.sndcmd , sizeof (ncb.sndcmd )-1, "%-16s%1s%2s", "get500FTPOVER", "1", "00" ); sprintf(ncb.buff, "%06d%s", ncb.wlen, ncb.sndcmd ); iRet = tcp_writebuf (sfd , ncb.buff, ncb.wlen+6, timeout); if(iRet != ncb.wlen + 6 ){ wLog(LOGERROR,"发送 FileOver 失败 error [%d, %s]", errno, strerror(errno) ); chkflg = -1; goto cli_get_sendFile_END; } wLog(LOGINFO, "发送数据[%d] [%s]成功", iRet , ncb.sndcmd ); iRet = tcp_readbuf(6, sfd, ncb.rcvcmd , 19, timeout); if(iRet != 19 || memcmp(ncb.rcvcmd+17, "00", 2)){ wLog(LOGERROR,"远程接收 FileOver 失败 error [%d, %s]", errno, strerror(errno) ); chkflg = -1; goto cli_get_sendFile_END; } wLog(LOGINFO, "接到返回数据[%s]成功", ncb.rcvcmd ); wLog(LOGINFO, "共 [%d]个文件, 耗时 %ld 秒\n\n", fnum , time(NULL) - start ); cli_get_sendFile_END: if(sfd > 0) close(sfd); // 关闭本文描述符和通信连接通道描述符 return chkflg; } // ============================ 服务端使用 多文件传输程序 =============================== //// // // // socket 文件接收服务程序,对收到的通信报文进行分析。 // // 在收到 GET100 命令后,将该数据域中的文件名称与本地路径拼写得到完整的文件 // // 路径信息,打开本地文件。 // // 对 GET200 数据域传输来的数据保存到本地文件中。 // // 收到 GET300 命令,关闭当前传输文件。 // // 收到 get500 命令,TCP 文件传输过程结束,退出程序。 // // // // 文件传输服务端程序: // // sfd 在 accept() 后获取的新的客户端连接通道描述符 // // path 准备保存本地文件的路径信息 // // fileName 根据接收报文中的文件名称与本地路径拼串而得到的返回文件信息 // // timeout 数据传输需要使用的延时参数 // // // // 返回数据: // // 0 ---- 文件接收成功 // // -2 -- 文件无法创建,打开文件名称失败 // // -1 -- 文件内容内容保存失败 // // // // 文件传输过程中,上传的文件名称有客户端提供,但是文件保存路径由服务器控制,以充分保证 // // 服务器文件系统的安全,避免服务器上的文件被客户端上传文件恶意覆盖。 // // // // ====================================================================================== //// int srv_mget_recvFile(int sfd , char *path, int timeout ) { int fd = -1; int blknum = 0 , maxBlk = 0 ; char localFile[NFILE+1]; char tfile[NFILE+1], bfile[NFILE+1]; char *ptr; long fsize = 0; long start = time(NULL); int chkflg = 0; int fnum = 0; memset(tfile, 0x00, sizeof(tfile)); while (1){ memset(&ncb, 0x00, sizeof (struct tcpBuffer)); ncb.rlen = tcp_readbuf (6, sfd, ncb.buff, sizeof(ncb.buff)-1, timeout); if(ncb.rlen < 0 ) break ; memcpy(ncb.sndcmd + 6, ncb.buff, 19); if( memcmp(ncb.buff, "get", 3) != 0 ) { wLog(LOGERROR,"接收命令序列 [%s] 错误 ,程序退出. ", ncb.sndcmd+6 ); memcpy(ncb.rcvcmd+23, "01", 2); chkflg = -3; } switch ( Nstr_int(ncb.buff + 3, 3) ) { // 获取 get 后面的命令控制字,下面分析该控制字进行工作 case 100 : // 开始接收文件,打开本地文件 blknum = maxBlk = 0 ; wLog(LOGINFO, "接收管理报文[%s]成功", ncb.buff ); // 对于接收到的第一条命令,打印该控制命令的全部内容 ptr = strchr(ncb.buff, ':'); if(ptr) memcpy(tfile, ncb.buff+19, ptr - (char *)ncb.buff - 19); else strcpy(tfile, ncb.buff+19); // 获取传输来的文件名称 ptr = strrchr(tfile, '/'); if(ptr) strcpy(bfile, ptr+1); else strcpy(bfile, tfile); // 检查传输来文件名称中的路径信息,得到基本文件名称,将前面的路径 // 信息全部剔除,以保证本地文件的安全。当传输文件名称带有 ".." 标志 // 的时候,将会对服务器文件系统产生影响,需要保证该问题不会出现。 if( (ptr = strrchr(ncb.buff , ':') ) != NULL ) maxBlk = atoi (ptr +1); // 从命令报文中得到文件传输块信息。 if( path ) sprintf(localFile, "%s/%s", path, bfile); else strcpy(localFile, bfile ); // 与本地保存路径拼串,获得本地文件名称 fd = open(localFile, O_CREAT|O_WRONLY|O_TRUNC, 0666 ); if(fd < 0 ) { wLog(LOGERROR,"生成本地文件 [%s] 失败 error [%d, %s]", localFile , errno, strerror(errno) ); memcpy(ncb.rcvcmd+23, "01", 2); chkflg = -2; } fnum ++; // 对需要保存的本地文件,使用清空方式,创建新文件,既是该文件已经存在 // 也可以保证数据处理 wLog(LOGINFO, "创建本地文件[%s]成功", localFile ); break ; case 200 : // 保存文件内容 blknum ++; maxBlk --; if(blknum == 1 || !maxBlk )wLog(LOGINFO, "接收数据[%s]成功", ncb.sndcmd+6 ); ncb.wlen = write(fd, ncb.buff+19, ncb.rlen - 19); if(ncb.wlen != ncb.rlen - 19 ) { memcpy(ncb.sndcmd + 23, "01", 2); chkflg = -1; } else fsize += ncb.wlen ; break ; case 300 : // 文件传输结束 if( !maxBlk ) wLog(LOGINFO, " 文件[%s]成功接收,共 [%d] 传输块 ", localFile, blknum ); else wLog(LOGERROR,"文件[%s]接收结束,差错 [%d] 传输块 ", localFile , maxBlk ); close(fd); break ; case 500 : // 通信处理结束 chkflg = 1; break ; } memcpy(ncb.sndcmd , "000019", 6); ncb.sndcmd [22]='2'; ncb.wlen = tcp_writebuf (sfd , ncb.sndcmd , 25, timeout); if(ncb.wlen != 25 ){ wLog(LOGERROR,"发送返回信息 [%s] 失败 error [%d, %s]", ncb.sndcmd + 6, errno, strerror(errno) ); } if(chkflg ) break ; } if(fd) close (fd); wLog(LOGINFO, "成功接收[%d]文件, 耗时 [%ld] 秒 ", fnum , time(NULL) - start ); return chkflg ; } #endif
以上就是本文的全部内容,希望对大家的学习有所帮助。
上一篇: C++聚合关系类的构造函数的调用顺序详解
下一篇: C++中的delete不会将操作数置0
相关文章
- 下面小编来给大家演示几个php操作zip文件的实例,我们可以读取zip包中指定文件与删除zip包中指定文件,下面来给大这介绍一下。 从zip压缩文件中提取文件 代...2016-11-25
- vector是表示可以改变大小的数组的序列容器,本文主要介绍了C++STL标准库std::vector的使用详解,文中通过示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下...2022-03-06
Jupyter Notebook读取csv文件出现的问题及解决
这篇文章主要介绍了JupyterNotebook读取csv文件出现的问题及解决,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教...2023-01-06- 这篇文章主要介绍了C++中取余运算的实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-02-23
- 这篇文章主要介绍了C++ string常用截取字符串方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2020-04-25
- 本文通过例子,讲述了C++调用C#的DLL程序的方法,作出了以下总结,下面就让我们一起来学习吧。...2020-06-25
- 本篇文章主要介绍了C++中四种加密算法之AES源代码,小编觉得挺不错的,现在分享给大家,也给大家做个参考。...2020-04-25
- 有时我们接受或下载到的PSD文件打开是空白的,那么我们要如何来解决这个 问题了,下面一聚教程小伙伴就为各位介绍Photoshop打开PSD文件空白解决办法。 1、如我们打开...2016-09-14
- 这篇文章主要介绍了解决python 使用openpyxl读写大文件的坑,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-03-13
- 这篇文章主要介绍了C#实现HTTP下载文件的方法,包括了HTTP通信的创建、本地文件的写入等,非常具有实用价值,需要的朋友可以参考下...2020-06-25
- 这篇文章主要为大家详细介绍了SpringBoot实现excel文件生成和下载,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下...2021-02-09
- C#使用System.IO中的文件操作方法在Windows系统中处理本地文件相当顺手,这里我们还总结了在Oracle中保存文件的方法,嗯,接下来就来看看整理的C#操作本地文件及保存文件到数据库的基本方法总结...2020-06-25
- 整数拆分,指把一个整数分解成若干个整数的和。本文重点给大家介绍C++ 整数拆分方法详解,非常不错,感兴趣的朋友一起学习吧...2020-04-25
- 本文章来人大家介绍一个php文件上传类的使用方法,期望此实例对各位php入门者会有不小帮助哦。 简介 Class.upload.php是用于管理上传文件的php文件上传类, 它可以帮...2016-11-25
php无刷新利用iframe实现页面无刷新上传文件(1/2)
利用form表单的target属性和iframe 一、上传文件的一个php教程方法。 该方法接受一个$file参数,该参数为从客户端获取的$_files变量,返回重新命名后的文件名,如果上传失...2016-11-25- 要替换字符串中的内容我们只要利用php相关函数,如strstr,str_replace,正则表达式了,那么我们要替换目录所有文件的内容就需要先遍历目录再打开文件再利用上面讲的函数替...2016-11-25
- 又码了一个周末的代码,这次在做一些关于文件上传的东西。(PHP UPLOAD)小有收获项目是一个BT种子列表,用户有权限上传自己的种子,然后配合BT TRACK服务器把种子的信息写出来...2016-11-25
- 这篇文章主要介绍了C++中Sort函数详细解析,sort函数是algorithm库下的一个函数,sort函数是不稳定的,即大小相同的元素在排序后相对顺序可能发生改变...2022-08-18
- 本文实例讲述了jQuery实现文件上传进度条效果的代码。分享给大家供大家参考。具体如下: 运行效果截图如下:具体代码如下:<!DOCTYPE html><html><head><meta charset="utf-8"><title>upload</title><link rel="stylesheet...2015-11-24
- 步骤:Window -> PHP -> Editor -> Templates,这里可以设置(增、删、改、导入等)管理你的模板。新建文件注释、函数注释、代码块等模板的实例新建模板,分别输入Name、Description、Patterna)文件注释Name: 3cfileDescriptio...2013-10-04