trunk/Spectre/Mounter.cpp

// Spectre Samba Mounter
//
// Douglas Thrift
//
// $Id: Mounter.cpp,v 1.10 2003/07/31 00:56:29 douglas Exp $

#include "Mounter.h"

#ifdef _FreeBSD_
#include <fcntl.h>
#endif

void Mounter::mount()
{
        if (!good) return;

        samba();

        if (debug)
        {
                cerr << "mounted = {\n";

                for (set<string>::iterator itor = mounted.begin();
                        itor != mounted.end(); itor++)
                {
                        cerr << "   " << *itor << "\n";
                }

                cerr << "}\n";
        }

        if (debug) cerr << "folders = {\n";

        for (map<string, Folder>::iterator itor = folders.begin(); itor !=
                folders.end(); itor++)
        {
                if (debug) cerr << "   " << itor->first << " = {\n"
                        << "      local = " << itor->second.local << "\n"
                        << "      remote = " << itor->second.remote << "\n"
                        << "   }\n";

                mount(itor->first, itor->second.local, itor->second.remote);
        }

        if (debug) cerr << "}\n";
}

void Mounter::samba()
{
        string path = config.root + "/" + host;
        DIR* dir = opendir(path.c_str());
        dirent* file;
        
        while ((file = readdir(dir)) != NULL)
        {
                struct stat dir;
                string folder = path + "/" + file->d_name;

                stat(folder.c_str(), &dir);

                if (S_ISDIR(dir.st_mode))
                {
                        string folder = file->d_name;

                        if (folder == "." || folder == "..") continue;
                        
                        string share = folder.find('.') == 0 ? folder.erase(0, 1) + "$" :
                                folder;

                        if (folders.find(share) == folders.end())
                        {
                                folders.insert(pair<string, Folder>(share, regular));
                        }
                }
        }

        closedir(dir);

        vector<string> args;

        args.push_back("spectre");
        args.push_back("-t");
        args.push_back("smbfs");

        ipstream mount(config.mount, args);

        if (debug) cerr << "mount = {\n";

        samba(mount);

        if (debug) cerr << "}\n";

        mount.close();
}

void Mounter::samba(ipstream& pin)
{
        while (pin.good())
        {
                string line;

                getline(pin, line);

                if (debug) cerr << line << "\n";

                if (line == "") continue;

                unsigned begin = line.find(" on ") + 4;
#if defined _Linux_
                unsigned end = line.rfind(" type smbfs");
#elif defined _FreeBSD_
                unsigned end = line.rfind(" (smbfs)");
#endif

                string path = line.substr(begin, end - begin);

                if (mounted.find(path) == mounted.end())
                {
                        mounted.insert(path);
                }
        }
}

void Mounter::mount(const string& folder, const string& user, const string&
        owner)
{
        string path = config.root + "/" + host + "/" + (folder[folder.length() -
                1] == '$' ? "." + folder.substr(0, folder.length() - 1) : folder);
        
        if (mounted.find(path) != mounted.end()) return;

        ostringstream options;

        struct passwd* item = getpwnam(user.c_str());

        if (item == NULL)
        {
                cerr << program << ": " << user << ": Unknown user\n";
                return;
        }
        
#if defined _Linux_
        char credentials[17];

        sprintf(credentials, "%s", "/tmp/spectre.XXXX");
        mkstemp(credentials);
        
        ofstream fout(credentials);

        fout << "username = " << owner << "\n"
                << "password = " << password(owner) << "\n";

        fout.close();

        options << "credentials=" << credentials << ","
                << "uid=" << item->pw_uid << ","
                << "gid=" << item->pw_gid;
        
        string smb = "//" + host + "/" + folder;
#elif defined _FreeBSD_
        char* credentials = new char[strlen(getenv("HOME")) + 9];
        
        sprintf(credentials, "%s/.nsmbrc", getenv("HOME"));
        int dot = open(credentials, O_CREAT | O_TRUNC, S_IRUSR | S_IWUSR);
        close(dot);

        ofstream fout(credentials);

        fout << "[" << toupper(host) << ":" << toupper(owner) << "]\n"
                << "password=" << password(owner) << "\n";

        fout.close();

        options << "-N,"
                << "-u=" << item->pw_uid << ","
                << "-g=" << item->pw_gid;

        string smb = "//" + owner + "@" + host + "/" + folder;
#endif

        vector<string> args;

        args.push_back("spectre");
#ifdef _FreeBSD_
        args.push_back("-o");
        args.push_back(options.str());
#endif
        args.push_back("-t");
        args.push_back("smbfs");
#ifdef _Linux_
        args.push_back("-o");
        args.push_back(options.str());
#endif
        args.push_back(smb);
        args.push_back(path);

        ipstream mount(config.mount, args, pstreambuf::pstderr);

        while (mount.good())
        {
                string line;

                getline(mount, line);

                if (line != "") cerr << line << "\n";
        }

        mount.close();

        unlink(credentials);
#ifdef _FreeBSD_
        delete [] credentials;
#endif
}
Revision:	227
Committed:	2003-07-30T17:56:29-07:00 (21 years, 10 months ago) by douglas
File size:	4086 byte(s)
Log Message:	Thought about the meaning of life, added $Id$ tags.
#	User	Rev	Content
1	douglas	161	// Spectre Samba Mounter
2			//
3			// Douglas Thrift
4			//
5	douglas	227	// $Id: Mounter.cpp,v 1.10 2003/07/31 00:56:29 douglas Exp $
6	douglas	161
7	douglas	174	#include "Mounter.h"
8
9	douglas	172	#ifdef _FreeBSD_
10			#include <fcntl.h>
11			#endif
12
13	douglas	161	void Mounter::mount()
14			{
15	douglas	176	if (!good) return;
16
17	douglas	171	samba();
18
19	douglas	172	if (debug)
20			{
21			cerr << "mounted = {\n";
22
23			for (set<string>::iterator itor = mounted.begin();
24			itor != mounted.end(); itor++)
25			{
26			cerr << " " << *itor << "\n";
27			}
28
29			cerr << "}\n";
30			}
31
32	douglas	171	if (debug) cerr << "folders = {\n";
33
34			for (map<string, Folder>::iterator itor = folders.begin(); itor !=
35			folders.end(); itor++)
36			{
37			if (debug) cerr << " " << itor->first << " = {\n"
38			<< " local = " << itor->second.local << "\n"
39			<< " remote = " << itor->second.remote << "\n"
40			<< " }\n";
41
42	douglas	172	mount(itor->first, itor->second.local, itor->second.remote);
43	douglas	171	}
44
45			if (debug) cerr << "}\n";
46	douglas	161	}
47	douglas	171
48			void Mounter::samba()
49			{
50			string path = config.root + "/" + host;
51			DIR* dir = opendir(path.c_str());
52			dirent* file;
53
54			while ((file = readdir(dir)) != NULL)
55			{
56			struct stat dir;
57			string folder = path + "/" + file->d_name;
58
59			stat(folder.c_str(), &dir);
60
61			if (S_ISDIR(dir.st_mode))
62			{
63			string folder = file->d_name;
64
65			if (folder == "." \|\| folder == "..") continue;
66
67			string share = folder.find('.') == 0 ? folder.erase(0, 1) + "$" :
68			folder;
69
70			if (folders.find(share) == folders.end())
71			{
72			folders.insert(pair<string, Folder>(share, regular));
73			}
74			}
75			}
76
77			closedir(dir);
78	douglas	172
79			vector<string> args;
80
81			args.push_back("spectre");
82			args.push_back("-t");
83			args.push_back("smbfs");
84
85			ipstream mount(config.mount, args);
86
87			if (debug) cerr << "mount = {\n";
88
89			samba(mount);
90
91			if (debug) cerr << "}\n";
92
93			mount.close();
94	douglas	171	}
95
96			void Mounter::samba(ipstream& pin)
97			{
98	douglas	172	while (pin.good())
99			{
100			string line;
101
102			getline(pin, line);
103
104			if (debug) cerr << line << "\n";
105
106			if (line == "") continue;
107
108			unsigned begin = line.find(" on ") + 4;
109			#if defined _Linux_
110			unsigned end = line.rfind(" type smbfs");
111			#elif defined _FreeBSD_
112			unsigned end = line.rfind(" (smbfs)");
113			#endif
114
115			string path = line.substr(begin, end - begin);
116
117			if (mounted.find(path) == mounted.end())
118			{
119			mounted.insert(path);
120			}
121			}
122	douglas	171	}
123
124	douglas	172	void Mounter::mount(const string& folder, const string& user, const string&
125			owner)
126	douglas	171	{
127	douglas	173	string path = config.root + "/" + host + "/" + (folder[folder.length() -
128			1] == '$' ? "." + folder.substr(0, folder.length() - 1) : folder);
129	douglas	172
130			if (mounted.find(path) != mounted.end()) return;
131
132			ostringstream options;
133
134			struct passwd* item = getpwnam(user.c_str());
135
136			if (item == NULL)
137			{
138			cerr << program << ": " << user << ": Unknown user\n";
139			return;
140			}
141
142			#if defined _Linux_
143			char credentials[17];
144
145			sprintf(credentials, "%s", "/tmp/spectre.XXXX");
146			mkstemp(credentials);
147
148			ofstream fout(credentials);
149
150			fout << "username = " << owner << "\n"
151			<< "password = " << password(owner) << "\n";
152
153			fout.close();
154
155			options << "credentials=" << credentials << ","
156			<< "uid=" << item->pw_uid << ","
157			<< "gid=" << item->pw_gid;
158
159			string smb = "//" + host + "/" + folder;
160			#elif defined _FreeBSD_
161	douglas	174	char* credentials = new char[strlen(getenv("HOME")) + 9];
162
163			sprintf(credentials, "%s/.nsmbrc", getenv("HOME"));
164	douglas	172	int dot = open(credentials, O_CREAT \| O_TRUNC, S_IRUSR \| S_IWUSR);
165			close(dot);
166
167			ofstream fout(credentials);
168
169	douglas	174	fout << "[" << toupper(host) << ":" << toupper(owner) << "]\n"
170	douglas	172	<< "password=" << password(owner) << "\n";
171
172			fout.close();
173
174			options << "-N,"
175			<< "-u=" << item->pw_uid << ","
176			<< "-g=" << item->pw_gid;
177
178			string smb = "//" + owner + "@" + host + "/" + folder;
179			#endif
180
181			vector<string> args;
182
183			args.push_back("spectre");
184			#ifdef _FreeBSD_
185			args.push_back("-o");
186			args.push_back(options.str());
187			#endif
188			args.push_back("-t");
189			args.push_back("smbfs");
190			#ifdef _Linux_
191			args.push_back("-o");
192			args.push_back(options.str());
193			#endif
194			args.push_back(smb);
195			args.push_back(path);
196
197			ipstream mount(config.mount, args, pstreambuf::pstderr);
198
199			while (mount.good())
200			{
201			string line;
202
203			getline(mount, line);
204
205			if (line != "") cerr << line << "\n";
206			}
207
208			mount.close();
209
210			unlink(credentials);
211	douglas	174	#ifdef _FreeBSD_
212			delete [] credentials;
213			#endif
214	douglas	171	}