cpp实现ftp服务器.zip

#include "srvpi.h" SrvPI::SrvPI(string dbFilename, int connfd): packet(this), readpacket(this), db(DBFILENAME) { this->connfd = connfd; connSockStream.init(connfd); sessionCommandPacketCount = 0; userID ="0"; this->fp = NULL; } bool SrvPI::recvOnePacket() { // clear temporary variables of one tinyFTP transaction // this->filename.clear(); // this->abspath.clear(); int n; packet.reset(NPACKET); if ( (n = connSockStream.readn(packet.getPs(), PACKSIZE)) == 0) { this->saveUserState(); Socket::tcpClose(connfd); Error::quit_pthread("client terminated prematurely"); } else if (n < 0){ this->saveUserState(); Socket::tcpClose(connfd); Error::ret("connSockStream.readn() error"); Error::quit_pthread("socket connection exception"); } else { packet.ntohp(); //packet.print(); } return true; } bool SrvPI::sendOnePacketBlocked(PacketStruct * ps, size_t nbytes) { int m; if ( (m = connSockStream.writen(ps, nbytes)) < 0 || (size_t)m != nbytes ) { this->saveUserState(); Socket::tcpClose(connfd); Error::ret("connSockStream.writen()"); Error::quit_pthread("socket connection exception"); return false; } else { return true; } } bool SrvPI::sendOnePacket(PacketStruct * ps, size_t nbytes) { int n, m; bool sendFlag = false; int maxfdp1; fd_set rset, wset; FD_ZERO(&rset); FD_ZERO(&wset); while(!sendFlag) { FD_SET(connfd, &rset); FD_SET(connfd, &wset); maxfdp1 = connfd + 1; if (select(maxfdp1, &rset, &wset, NULL, NULL) < 0) { this->saveUserState(); Socket::tcpClose(connfd); Error::ret("select error"); Error::quit_pthread("socket connection exception"); } if (FD_ISSET(connfd, &rset)) /* socket is readable */ { readpacket.reset(NPACKET); if ( (n = connSockStream.readn(readpacket.getPs(), PACKSIZE)) == 0) { this->saveUserState(); Socket::tcpClose(connfd); Error::quit_pthread("client terminated prematurely"); } else if (n < 0){ this->saveUserState(); Socket::tcpClose(connfd); Error::ret("connSockStream.readn() error"); Error::quit_pthread("socket connection exception"); } else { if (n == PACKSIZE) { readpacket.ntohp(); printf("sendOnePacket method recive one packet: %s\n", readpacket.getSBody().c_str()); //readpacket.print(); } else { printf("ERROR: sendOnePacket method recive one packet: n != PACKSIZE"); } } } if (FD_ISSET(connfd, &wset)) /* socket is writable */ { if ( (m = connSockStream.writen(ps, nbytes)) < 0 || (size_t)m != nbytes ) { this->saveUserState(); Socket::tcpClose(connfd); Error::ret("connSockStream.writen()"); Error::quit_pthread("socket connection exception"); } else { sendFlag = true; } } } return true; } void SrvPI::run() { recvOnePacket(); std::cout<< "\n\n\033[32mNewCMD connfd: " << connfd << " [" << userRootDir <<" " << userRCWD << "]\033[0m" << std::endl; sessionCommandPacketCount++; // if (std::stoul(userID) != packet.getSesid()) // { // Error::quit_pthread("session failed"); // } // vector<string> testVector; // split("home", "///!", testVector); // for (vector<string>::iterator iter=testVector.begin(); iter!=testVector.end(); ++iter) // { // std::cout << "test[" << *iter << "]" << '\n'; // } if (packet.getTagid() == TAG_CMD) { switch(packet.getCmdid()) { case USER: cmdUSER(); break; case PASS: cmdPASS(); break; case USERADD: cmdUSERADD(); break; case USERDEL: cmdUSERDEL(); break; case GET: cmdGET(); break; case RGET: cmdRGET(); break; case PUT: cmdPUT(); break; case RPUT: cmdRPUT(); break; case LS: cmdLS(); break; case CD: cmdCD(); break; case RM: cmdRM(); break; case PWD: cmdPWD(); break; case MKDIR: cmdMKDIR(); break; case RMDIR: cmdRMDIR(); break; case SHELL: cmdSHELL(); break; default: Error::msg("Server: Sorry! this command function not finished yet.\n"); break; } } else { Error::msg("Error: received packet is not a command.\n"); packet.print(); Error::quit_pthread("*********socket connection exception*********"); } } void SrvPI::split(std::string src, std::string token, vector<string>& vect) { int nbegin=0; int nend=0; while(nend != -1 && (unsigned int)nbegin < src.length() ) { nend = src.find_first_of(token, nbegin); if(nend == -1) { vect.push_back(src.substr(nbegin, src.length()-nbegin)); } else { if (nend != nbegin ) { vect.push_back(src.substr(nbegin, nend-nbegin)); } } nbegin = nend + 1; } } // static void Split(const char* content, const char* token, vector<std::string>& vect) // { // if(content == NULL) // return; // int len = strlen(content); // if(len <= 0) // return; // char* pBuf =(char*)malloc(len+1); // strcpy(pBuf , content); // // not thread safe // char* str = strtok(pBuf , token); // while(str != NULL) // { // vect.push_back(str); // str = strtok(NULL, token); // } // free(pBuf); // } void SrvPI::cmdUSER() { printf("USER request\n"); vector<string> paramVector; split(packet.getSBody(), "\t", paramVector); std::map<string, string> selectParamMap = { {"username", paramVector[0]} }; if (db.select("user", selectParamMap)) { vector< map<string ,string> > resultMapVector = db.getResult(); if (!resultMapVector.empty()) { packet.sendSTAT_OK("this username exists"); } else { packet.sendSTAT_ERR("no such username"); } }else { packet.sendSTAT_ERR("Database select error"); } } void SrvPI::cmdPASS() { printf("PASS request\n"); vector<string> paramVector; split(packet.getSBody(), DELIMITER, paramVector); // for (vector<string>::iterator iter=paramVector.begin(); iter!=paramVector.end(); ++iter) // { // std::cout << "paramVector[" << *iter << "]" << '\n'; // } std::map<string, string> selectParamMap = { {"username", paramVector[0]}, {"password", paramVector[1]} }; if (db.select("user", selectParamMap)) { vector< map<string ,string> > resultMapVector = db.getResult(); if (!resultMapVector.empty()) { // init userID, userRootDir, and userRCWD userID = resultMapVector[0]["ID"]; userRootDir = ROOTDIR + resultMapVector[0]["USERNAME"]; userRCWD = resultMapVector[0]["RCWD"]; // set session ID: important packet.setSessionID(std::stoul(userID)); //packet.print(); packet.sendSTAT_OK("Welcome to tinyFTP, written by Charles Wenchy <wenchy.zwz@gmail.com>\n" \ + resultMapVector[0]["USERNAME"] + ", your last working directory is: ~" + userRCWD); } else { packet.sendSTAT_ERR("error: username mismatch password"); } } else { packet.sendSTAT_ERR("Database select error"); } } void SrvPI::cmdUSERADD() { printf("USERADD request\n"); if (userID != "1") { packet.sendSTAT_ERR("Permission denied, admin required"); return; } vector<string> paramVector; split(packet.getSBody(), DELIMITER, paramVector); std::map<string, string> selectParamMap = { {"username", paramVector[0]} }; if (db.select("user", selectParamMap)) { vector< map<string ,string> > resultMapVector = db.getResult(); if (!resultMapVector.empty()) { packet.sendSTAT_ERR("User '" + paramVector[0] + "' already exists"); return; } }else { packet.sendSTAT_ERR("Database select error"); } std::map<string, string> insertParamMap = { {"username", paramVector[0]}, {"password", paramVector[1]} }; if (db.insert("user", insertParamMap)) { string path = ROOTDIR + paramVector[0]; if(mkdir(path.c_str(), 0777) == -1)