ViewVC Help

View File | Revision Log | Show Annotations | Download File | View Changeset | Root Listing

root/repos/Spectre2/Mounter.cpp

Comparing Spectre2/Mounter.cpp (file contents):
Revision 428 by douglas, 2005-03-24T17:38:00-08:00 vs.
Revision 432 by douglas, 2005-03-27T01:56:29-08:00

#	Line 6 | Line 6
6
7		#include "Mounter.hpp"
8
9	<	#include <menes-api/environment.hpp>
10	<	#include <menes-api/socket.hpp>
9	>	#include <grp.h>
10	>	#include <pwd.h>
11
12	<	#include <arpa/inet.h>
13	<
14	<	Mounter::Mounter(const api::Path& config) : Daemon(), config(config) {}
15	<
16	<	Mounter::~Mounter()
12	>	void Mounter::operator()()
13		{
14	<	#ifdef __FreeBSD__
19	<	api::Posix::CheckError(::unlink(secret.NullTerminate()));
20	<	#endif
21	<	}
14	>	// XXX: make sure the directory exists!
15
16		#ifdef __FreeBSD__
17	<	ext::String Mounter::secret(api::TheEnvironment.Get("HOME") + "/.nsmbrc");
18	<	#endif
17	>	{
18	>	bool password(false);
19
20	<	void Mounter::load()
21	<	{
22	<	_synchronized (Spectre2::sharesLock)
30	<	Spectre2::shares.Clear();
20	>	{
21	>	api::FileReader in(Daemon::secret);
22	>	ext::String line;
23
24	<	_synchronized (Share::passwordsLock)
25	<	Share::passwords.Clear();
24	>	while (ios::ReadLine(in, line)) if (line == "[" + share.getHost() + ":" + share.getOwner() + "]")
25	>	{
26	>	password = true;
27
28	<	_H<xml::Document> document(xml::Parse(config.GetPath()));
29	<	_H<xml::Node> spectre(*document/"spectre");
30	<	#ifdef __FreeBSD__
31	<	SecretFileWriter out(secret);
32	<	ios::FormatWriter fout(out);
33	<	#endif
28	>	break;
29	>	}
30	>	}
31	>
32	>	if (!password)
33	>	{
34	>	AppendFileWriter out(Daemon::secret);
35	>	ios::FormatWriter fout(out);
36	>
37	>	fout << "[" << share.getHost() << ":" << share.getOwner() << "]" << ios::NewLine << "password=" << share.getPassword() << ios::NewLine;
38	>	}
39	>	}
40	>	#else
41	>	_S<ios::String> credentials;
42	>
43	>	// XXX: _unsynchronized
44	>	_synchronized (Daemon::secretsLock) if (Daemon::secrets.Contains(share.getOwner() + "@" + share.getHost()))
45	>	credentials = Daemon::secrets[share.getOwner() + "@" + share.getHost()];
46
47	<	_foreach (const xml::NodeSet, host_, *spectre/"host")
47	>	if (credentials.IsEmpty())
48		{
49	<	ext::String host(**host_/"name");
49	>	credentials << "/tmp/.spectre" << api::Uuid::Create();
50
46	–	_foreach (const xml::NodeSet, share, **host_/"share")
51		{
52	<	ext::String name(**share/"name"), owner(**share/"owner"), user(**share/"user"), group(**share/"group");
52	>	SecretFileWriter out(credentials);
53	>	ios::FormatWriter fout(out);
54
55	<	_synchronized (Spectre2::sharesLock)
51	<	Spectre2::shares.Insert(Share(host, name, owner, user, group));
55	>	fout << "username = " << share.getOwner() << ios::NewLine << "password = " << share.getPassword() << ios::NewLine;
56		}
57
58	<	#ifdef __FreeBSD__
59	<	::addrinfo* info;
58	>	_synchronized (Daemon::secretsLock)
59	>	Daemon::secrets[share.getOwner() + "@" + share.getHost()];
60	>	}
61	>	#endif
62
63	<	api::Posix::CheckGaiError(::getaddrinfo(host.NullTerminate(), NULL, NULL, &info));
63	>	_S<ios::String> uid;
64
65	<	::sockaddr_in& sock(*reinterpret_cast< ::sockaddr_in*>(info->ai_addr));
66	<	ext::Buffer buffer(128);
65	>	{
66	>	::passwd pwd, * result;
67	>	ext::Buffer buffer(1024);
68
69	<	fout << "[" << host << "]" << ios::NewLine << "addr=" << ::inet_ntop(sock.sin_family, &sock.sin_addr, buffer.Begin(), buffer.GetSize()) << ios::NewLine;
69	>	api::Posix::CheckError(::getpwnam_r(share.getUser().NullTerminate(), &pwd, buffer.Begin(), buffer.GetSize(), &result));
70
71	<	::freeaddrinfo(info);
65	<	#endif
71	>	uid << pwd.pw_uid;
72		}
73
74	<	_synchronized (Spectre2::sharesLock)
69	<	Spectre2::shares.Output(api::Cout);
70	<
71	<	loaded = true;
72	<	}
73	<
74	<	void Mounter::run()
75	<	{
76	<	ext::ThreadSet<> workers;
74	>	_S<ios::String> gid;
75
78	–	_synchronized (Spectre2::sharesLock) _foreach (const ext::RedBlackSet<Share>, share, Spectre2::shares) if (!share->mounted() && share->mountable())
76		{
77	<	api::Cerr << "STUB: mount " << *share << ios::NewLine;
77	>	::group grp, * result;
78	>	ext::Buffer buffer(1024);
79	>
80	>	api::Posix::CheckError(::getgrnam_r(share.getGroup().NullTerminate(), &grp, buffer.Begin(), buffer.GetSize(), &result));
81	>
82	>	gid << grp.gr_gid;
83		}
84
85	<	workers.Join();
85	>	_L<ext::String> options;
86	>
87	>	#ifdef __FreeBSD__
88	>	options.InsertLast("-N");
89	>	// XXX: hmm?
90	>	//      options.InsertLast("-O=" + share.getOwner() + ":");
91	>	options.InsertLast("-u=" + uid);
92	>	options.InsertLast("-g=" + gid);
93	>	#else
94	>	options.InsertLast("credentials=" + credentials);
95	>	options.InsertLast("uid=" + uid);
96	>	options.InsertLast("gid=" + gid);
97	>	options.InsertLast("rw");
98	>	#endif
99	>
100	>	_L<ext::String> args(1, "-o");
101
102	<	::sleep(5);
102	>	args.InsertLast(ext::JoinAll<ext::String>(options, ","));
103	>	args.InsertLast("-t");
104	>	args.InsertLast("smbfs");
105	>	args.InsertLast(share.getService());
106	>	args.InsertLast(share.getMount());
107	>
108	>	_S<api::Process> mount(Spectre2::mount, args);
109	>
110	>	if (Spectre2::debug)
111	>	ios::ReadToWrite(*mount.GetReader(), api::Cout);
112	>	else
113	>	ios::Discard(*mount.GetReader());
114		}

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines
>	Changed lines