static void printList(void)
{
int i, nNum;
printLine();
nNum = g_MountProg.GetMountNumber();
for (i = 0; i < nNum; i++)
printf("%s\n", g_MountProg.GetClientAddr(i));
printCount();
printLine();
}
static void start(char *path, char *pathAlias)
{
int i;
CDatagramSocket DatagramSockets[SOCKET_NUM];
CServerSocket ServerSockets[SOCKET_NUM];
bool bSuccess;
hostent *localHost;
g_PortmapProg.Set(PROG_MOUNT, MOUNT_PORT); //map port for mount
g_PortmapProg.Set(PROG_NFS, NFS_PORT); //map port for nfs
g_NFSProg.SetUserID(g_nUID, g_nGID); //set uid and gid of files
g_MountProg.Export(path, pathAlias); //export path for mount
g_RPCServer.Set(PROG_PORTMAP, &g_PortmapProg); //program for portmap
g_RPCServer.Set(PROG_NFS, &g_NFSProg); //program for nfs
g_RPCServer.Set(PROG_MOUNT, &g_MountProg); //program for mount
g_RPCServer.SetLogOn(g_bLogOn);
for (i = 0; i < SOCKET_NUM; i++)
{
DatagramSockets[i].SetListener(&g_RPCServer);
ServerSockets[i].SetListener(&g_RPCServer);
}
bSuccess = false;
if (ServerSockets[0].Open(PORTMAP_PORT, 3) && DatagramSockets[0].Open(PORTMAP_PORT)) //start portmap daemon
{
printf("Portmap daemon started\n");
if (ServerSockets[1].Open(NFS_PORT, 10) && DatagramSockets[1].Open(NFS_PORT)) //start nfs daemon
{
printf("NFS daemon started\n");
if (ServerSockets[2].Open(MOUNT_PORT, 3) && DatagramSockets[2].Open(MOUNT_PORT)) //start mount daemon
{
printf("Mount daemon started\n");
bSuccess = true; //all daemon started
}
else
printf("Mount daemon starts failed.\n");
}
else
printf("NFS daemon starts failed.\n");
}
else
printf("Portmap daemon starts failed.\n");
if (bSuccess)
{
localHost = gethostbyname("");
printf("Local IP = %s\n", inet_ntoa (*(struct in_addr *)*localHost->h_addr_list)); //local address
inputCommand(); //wait for commands
}
for (i = 0; i < SOCKET_NUM; i++)
{
DatagramSockets[i].Close();
ServerSockets[i].Close();
}
}
static void inputCommand(void)
{
char command[20];
printf("Type 'help' to see help\n\n");
while (true) {
fgets(command, 20, stdin);
if (command[strlen(command) - 1] == '\n') {
command[strlen(command) - 1] = '\0';
}
if (_stricmp(command, "about") == 0) {
printAbout();
} else if (_stricmp(command, "help") == 0) {
printHelp();
} else if (_stricmp(command, "log on") == 0) {
g_RPCServer.SetLogOn(true);
} else if (_stricmp(command, "log off") == 0) {
g_RPCServer.SetLogOn(false);
} else if (_stricmp(command, "list") == 0) {
printList();
} else if (_stricmp(command, "quit") == 0) {
if (g_MountProg.GetMountNumber() == 0) {
break;
} else {
printConfirmQuit();
fgets(command, 20, stdin);
if (command[0] == 'y' || command[0] == 'Y') {
break;
}
}
} else if (_stricmp(command, "refresh") == 0) {
g_MountProg.Refresh();
} else if (_stricmp(command, "reset") == 0) {
g_RPCServer.Set(PROG_NFS, NULL);
} else if (strcmp(command, "") != 0) {
printf("Unknown command: '%s'\n", command);
printf("Type 'help' to see help\n");
}
}
}
static void mountPaths(std::vector<std::vector<std::string>> paths)
{
int i;
int numberOfElements = paths.size();
for (i = 0; i < numberOfElements; i++) {
char *pPath = (char*)paths[i][0].c_str();
char *pPathAlias = (char*)paths[i][1].c_str();
g_MountProg.Export(pPath, pPathAlias); //export path for mount
}
}
static void printCount(void)
{
int nNum;
nNum = g_MountProg.GetMountNumber();
if (nNum == 0)
printf("There is no client mounted.\n");
else if (nNum == 1)
printf("There is 1 client mounted.\n");
else
printf("There are %d clients mounted.\n", nNum);
}
int main(int argc, char *argv[])
{
std::vector<std::vector<std::string>> pPaths;
char *pPath = NULL;
bool pathFile = false;
WSADATA wsaData;
printAbout();
if (argc < 2) {
pPath = strrchr(argv[0], '\\');
pPath = pPath == NULL ? argv[0] : pPath + 1;
printUsage(pPath);
return 1;
}
g_nUID = g_nGID = 0;
g_bLogOn = true;
g_sFileName = NULL;
for (int i = 1; i < argc; i++) {//parse parameters
if (_stricmp(argv[i], "-id") == 0) {
g_nUID = atoi(argv[++i]);
g_nGID = atoi(argv[++i]);
} else if (_stricmp(argv[i], "-log") == 0) {
g_bLogOn = _stricmp(argv[++i], "off") != 0;
} else if (_stricmp(argv[i], "-pathFile") == 0) {
g_sFileName = argv[++i];
if (g_MountProg.SetPathFile(g_sFileName) == false) {
printf("Can't open file %s.\n", g_sFileName);
return 1;
} else {
g_MountProg.Refresh();
pathFile = true;
}
} else if (i == argc - 2) {
pPath = argv[argc - 2]; //path is before the last parameter
char *pCurPathAlias = argv[argc - 1]; //path alias is the last parameter
if (pPath != NULL || pCurPathAlias != NULL) {
std::vector<std::string> pCurPaths;
pCurPaths.push_back(std::string(pPath));
pCurPaths.push_back(std::string(pCurPathAlias));
pPaths.push_back(pCurPaths);
}
break;
} else if (i == argc - 1) {
char *pPath = argv[argc - 1]; //path is the last parameter
if (pPath != NULL) {
char curPathAlias[MAXPATHLEN];
strcpy_s(curPathAlias, pPath);
char *pCurPathAlias = curPathAlias;
std::vector<std::string> pCurPaths;
pCurPaths.push_back(std::string(pPath));
pCurPaths.push_back(std::string(pCurPathAlias));
pPaths.push_back(pCurPaths);
}
break;
}
}
HWND console = GetConsoleWindow();
if (g_bLogOn == false && IsWindow(console)) {
ShowWindow(console, SW_HIDE); // hides the window
}
if (pPaths.size() <= 0 && !pathFile) {
printf("No paths to mount\n");
return 1;
}
WSAStartup(0x0101, &wsaData);
start(pPaths);
WSACleanup();
return 0;
}