| Index: chrome/browser/process_singleton_linux.cc
|
| diff --git a/chrome/browser/process_singleton_linux.cc b/chrome/browser/process_singleton_linux.cc
|
| deleted file mode 100644
|
| index 54d7db296f329d29016c8f5ba35d64bbefadd778..0000000000000000000000000000000000000000
|
| --- a/chrome/browser/process_singleton_linux.cc
|
| +++ /dev/null
|
| @@ -1,986 +0,0 @@
|
| -// Copyright (c) 2012 The Chromium Authors. All rights reserved.
|
| -// Use of this source code is governed by a BSD-style license that can be
|
| -// found in the LICENSE file.
|
| -
|
| -// On Linux, when the user tries to launch a second copy of chrome, we check
|
| -// for a socket in the user's profile directory. If the socket file is open we
|
| -// send a message to the first chrome browser process with the current
|
| -// directory and second process command line flags. The second process then
|
| -// exits.
|
| -//
|
| -// Because many networked filesystem implementations do not support unix domain
|
| -// sockets, we create the socket in a temporary directory and create a symlink
|
| -// in the profile. This temporary directory is no longer bound to the profile,
|
| -// and may disappear across a reboot or login to a separate session. To bind
|
| -// them, we store a unique cookie in the profile directory, which must also be
|
| -// present in the remote directory to connect. The cookie is checked both before
|
| -// and after the connection. /tmp is sticky, and different Chrome sessions use
|
| -// different cookies. Thus, a matching cookie before and after means the
|
| -// connection was to a directory with a valid cookie.
|
| -//
|
| -// We also have a lock file, which is a symlink to a non-existent destination.
|
| -// The destination is a string containing the hostname and process id of
|
| -// chrome's browser process, eg. "SingletonLock -> example.com-9156". When the
|
| -// first copy of chrome exits it will delete the lock file on shutdown, so that
|
| -// a different instance on a different host may then use the profile directory.
|
| -//
|
| -// If writing to the socket fails, the hostname in the lock is checked to see if
|
| -// another instance is running a different host using a shared filesystem (nfs,
|
| -// etc.) If the hostname differs an error is displayed and the second process
|
| -// exits. Otherwise the first process (if any) is killed and the second process
|
| -// starts as normal.
|
| -//
|
| -// When the second process sends the current directory and command line flags to
|
| -// the first process, it waits for an ACK message back from the first process
|
| -// for a certain time. If there is no ACK message back in time, then the first
|
| -// process will be considered as hung for some reason. The second process then
|
| -// retrieves the process id from the symbol link and kills it by sending
|
| -// SIGKILL. Then the second process starts as normal.
|
| -
|
| -#include "chrome/browser/process_singleton.h"
|
| -
|
| -#include <errno.h>
|
| -#include <fcntl.h>
|
| -#include <signal.h>
|
| -#include <sys/socket.h>
|
| -#include <sys/stat.h>
|
| -#include <sys/types.h>
|
| -#include <sys/un.h>
|
| -#include <unistd.h>
|
| -
|
| -#include <cstring>
|
| -#include <set>
|
| -#include <string>
|
| -
|
| -#include "base/base_paths.h"
|
| -#include "base/basictypes.h"
|
| -#include "base/bind.h"
|
| -#include "base/command_line.h"
|
| -#include "base/file_util.h"
|
| -#include "base/files/file_path.h"
|
| -#include "base/logging.h"
|
| -#include "base/message_loop/message_loop.h"
|
| -#include "base/path_service.h"
|
| -#include "base/posix/eintr_wrapper.h"
|
| -#include "base/rand_util.h"
|
| -#include "base/safe_strerror_posix.h"
|
| -#include "base/sequenced_task_runner_helpers.h"
|
| -#include "base/stl_util.h"
|
| -#include "base/strings/string_number_conversions.h"
|
| -#include "base/strings/string_split.h"
|
| -#include "base/strings/stringprintf.h"
|
| -#include "base/strings/sys_string_conversions.h"
|
| -#include "base/strings/utf_string_conversions.h"
|
| -#include "base/threading/platform_thread.h"
|
| -#include "base/time/time.h"
|
| -#include "base/timer/timer.h"
|
| -#include "chrome/browser/ui/process_singleton_dialog_linux.h"
|
| -#include "chrome/common/chrome_constants.h"
|
| -#include "content/public/browser/browser_thread.h"
|
| -#include "grit/chromium_strings.h"
|
| -#include "grit/generated_resources.h"
|
| -#include "net/base/net_util.h"
|
| -#include "ui/base/l10n/l10n_util.h"
|
| -
|
| -#if defined(TOOLKIT_VIEWS) && !defined(OS_CHROMEOS)
|
| -#include "ui/views/linux_ui/linux_ui.h"
|
| -#endif
|
| -
|
| -using content::BrowserThread;
|
| -
|
| -const int ProcessSingleton::kTimeoutInSeconds;
|
| -
|
| -namespace {
|
| -
|
| -static bool g_disable_prompt;
|
| -const char kStartToken[] = "START";
|
| -const char kACKToken[] = "ACK";
|
| -const char kShutdownToken[] = "SHUTDOWN";
|
| -const char kTokenDelimiter = '\0';
|
| -const int kMaxMessageLength = 32 * 1024;
|
| -const int kMaxACKMessageLength = arraysize(kShutdownToken) - 1;
|
| -
|
| -const char kLockDelimiter = '-';
|
| -
|
| -// Set a file descriptor to be non-blocking.
|
| -// Return 0 on success, -1 on failure.
|
| -int SetNonBlocking(int fd) {
|
| - int flags = fcntl(fd, F_GETFL, 0);
|
| - if (-1 == flags)
|
| - return flags;
|
| - if (flags & O_NONBLOCK)
|
| - return 0;
|
| - return fcntl(fd, F_SETFL, flags | O_NONBLOCK);
|
| -}
|
| -
|
| -// Set the close-on-exec bit on a file descriptor.
|
| -// Returns 0 on success, -1 on failure.
|
| -int SetCloseOnExec(int fd) {
|
| - int flags = fcntl(fd, F_GETFD, 0);
|
| - if (-1 == flags)
|
| - return flags;
|
| - if (flags & FD_CLOEXEC)
|
| - return 0;
|
| - return fcntl(fd, F_SETFD, flags | FD_CLOEXEC);
|
| -}
|
| -
|
| -// Close a socket and check return value.
|
| -void CloseSocket(int fd) {
|
| - int rv = IGNORE_EINTR(close(fd));
|
| - DCHECK_EQ(0, rv) << "Error closing socket: " << safe_strerror(errno);
|
| -}
|
| -
|
| -// Write a message to a socket fd.
|
| -bool WriteToSocket(int fd, const char *message, size_t length) {
|
| - DCHECK(message);
|
| - DCHECK(length);
|
| - size_t bytes_written = 0;
|
| - do {
|
| - ssize_t rv = HANDLE_EINTR(
|
| - write(fd, message + bytes_written, length - bytes_written));
|
| - if (rv < 0) {
|
| - if (errno == EAGAIN || errno == EWOULDBLOCK) {
|
| - // The socket shouldn't block, we're sending so little data. Just give
|
| - // up here, since NotifyOtherProcess() doesn't have an asynchronous api.
|
| - LOG(ERROR) << "ProcessSingleton would block on write(), so it gave up.";
|
| - return false;
|
| - }
|
| - PLOG(ERROR) << "write() failed";
|
| - return false;
|
| - }
|
| - bytes_written += rv;
|
| - } while (bytes_written < length);
|
| -
|
| - return true;
|
| -}
|
| -
|
| -// Wait a socket for read for a certain timeout in seconds.
|
| -// Returns -1 if error occurred, 0 if timeout reached, > 0 if the socket is
|
| -// ready for read.
|
| -int WaitSocketForRead(int fd, int timeout) {
|
| - fd_set read_fds;
|
| - struct timeval tv;
|
| -
|
| - FD_ZERO(&read_fds);
|
| - FD_SET(fd, &read_fds);
|
| - tv.tv_sec = timeout;
|
| - tv.tv_usec = 0;
|
| -
|
| - return HANDLE_EINTR(select(fd + 1, &read_fds, NULL, NULL, &tv));
|
| -}
|
| -
|
| -// Read a message from a socket fd, with an optional timeout in seconds.
|
| -// If |timeout| <= 0 then read immediately.
|
| -// Return number of bytes actually read, or -1 on error.
|
| -ssize_t ReadFromSocket(int fd, char *buf, size_t bufsize, int timeout) {
|
| - if (timeout > 0) {
|
| - int rv = WaitSocketForRead(fd, timeout);
|
| - if (rv <= 0)
|
| - return rv;
|
| - }
|
| -
|
| - size_t bytes_read = 0;
|
| - do {
|
| - ssize_t rv = HANDLE_EINTR(read(fd, buf + bytes_read, bufsize - bytes_read));
|
| - if (rv < 0) {
|
| - if (errno != EAGAIN && errno != EWOULDBLOCK) {
|
| - PLOG(ERROR) << "read() failed";
|
| - return rv;
|
| - } else {
|
| - // It would block, so we just return what has been read.
|
| - return bytes_read;
|
| - }
|
| - } else if (!rv) {
|
| - // No more data to read.
|
| - return bytes_read;
|
| - } else {
|
| - bytes_read += rv;
|
| - }
|
| - } while (bytes_read < bufsize);
|
| -
|
| - return bytes_read;
|
| -}
|
| -
|
| -// Set up a sockaddr appropriate for messaging.
|
| -void SetupSockAddr(const std::string& path, struct sockaddr_un* addr) {
|
| - addr->sun_family = AF_UNIX;
|
| - CHECK(path.length() < arraysize(addr->sun_path))
|
| - << "Socket path too long: " << path;
|
| - base::strlcpy(addr->sun_path, path.c_str(), arraysize(addr->sun_path));
|
| -}
|
| -
|
| -// Set up a socket appropriate for messaging.
|
| -int SetupSocketOnly() {
|
| - int sock = socket(PF_UNIX, SOCK_STREAM, 0);
|
| - PCHECK(sock >= 0) << "socket() failed";
|
| -
|
| - int rv = SetNonBlocking(sock);
|
| - DCHECK_EQ(0, rv) << "Failed to make non-blocking socket.";
|
| - rv = SetCloseOnExec(sock);
|
| - DCHECK_EQ(0, rv) << "Failed to set CLOEXEC on socket.";
|
| -
|
| - return sock;
|
| -}
|
| -
|
| -// Set up a socket and sockaddr appropriate for messaging.
|
| -void SetupSocket(const std::string& path, int* sock, struct sockaddr_un* addr) {
|
| - *sock = SetupSocketOnly();
|
| - SetupSockAddr(path, addr);
|
| -}
|
| -
|
| -// Read a symbolic link, return empty string if given path is not a symbol link.
|
| -base::FilePath ReadLink(const base::FilePath& path) {
|
| - base::FilePath target;
|
| - if (!base::ReadSymbolicLink(path, &target)) {
|
| - // The only errno that should occur is ENOENT.
|
| - if (errno != 0 && errno != ENOENT)
|
| - PLOG(ERROR) << "readlink(" << path.value() << ") failed";
|
| - }
|
| - return target;
|
| -}
|
| -
|
| -// Unlink a path. Return true on success.
|
| -bool UnlinkPath(const base::FilePath& path) {
|
| - int rv = unlink(path.value().c_str());
|
| - if (rv < 0 && errno != ENOENT)
|
| - PLOG(ERROR) << "Failed to unlink " << path.value();
|
| -
|
| - return rv == 0;
|
| -}
|
| -
|
| -// Create a symlink. Returns true on success.
|
| -bool SymlinkPath(const base::FilePath& target, const base::FilePath& path) {
|
| - if (!base::CreateSymbolicLink(target, path)) {
|
| - // Double check the value in case symlink suceeded but we got an incorrect
|
| - // failure due to NFS packet loss & retry.
|
| - int saved_errno = errno;
|
| - if (ReadLink(path) != target) {
|
| - // If we failed to create the lock, most likely another instance won the
|
| - // startup race.
|
| - errno = saved_errno;
|
| - PLOG(ERROR) << "Failed to create " << path.value();
|
| - return false;
|
| - }
|
| - }
|
| - return true;
|
| -}
|
| -
|
| -// Extract the hostname and pid from the lock symlink.
|
| -// Returns true if the lock existed.
|
| -bool ParseLockPath(const base::FilePath& path,
|
| - std::string* hostname,
|
| - int* pid) {
|
| - std::string real_path = ReadLink(path).value();
|
| - if (real_path.empty())
|
| - return false;
|
| -
|
| - std::string::size_type pos = real_path.rfind(kLockDelimiter);
|
| -
|
| - // If the path is not a symbolic link, or doesn't contain what we expect,
|
| - // bail.
|
| - if (pos == std::string::npos) {
|
| - *hostname = "";
|
| - *pid = -1;
|
| - return true;
|
| - }
|
| -
|
| - *hostname = real_path.substr(0, pos);
|
| -
|
| - const std::string& pid_str = real_path.substr(pos + 1);
|
| - if (!base::StringToInt(pid_str, pid))
|
| - *pid = -1;
|
| -
|
| - return true;
|
| -}
|
| -
|
| -// Returns true if the user opted to unlock the profile.
|
| -bool DisplayProfileInUseError(const base::FilePath& lock_path,
|
| - const std::string& hostname,
|
| - int pid) {
|
| - base::string16 error = l10n_util::GetStringFUTF16(
|
| - IDS_PROFILE_IN_USE_LINUX,
|
| - base::IntToString16(pid),
|
| - base::ASCIIToUTF16(hostname));
|
| - base::string16 relaunch_button_text = l10n_util::GetStringUTF16(
|
| - IDS_PROFILE_IN_USE_LINUX_RELAUNCH);
|
| - LOG(ERROR) << base::SysWideToNativeMB(base::UTF16ToWide(error)).c_str();
|
| - if (!g_disable_prompt)
|
| - return ShowProcessSingletonDialog(error, relaunch_button_text);
|
| - return false;
|
| -}
|
| -
|
| -bool IsChromeProcess(pid_t pid) {
|
| - base::FilePath other_chrome_path(base::GetProcessExecutablePath(pid));
|
| - return (!other_chrome_path.empty() &&
|
| - other_chrome_path.BaseName() ==
|
| - base::FilePath(chrome::kBrowserProcessExecutableName));
|
| -}
|
| -
|
| -// A helper class to hold onto a socket.
|
| -class ScopedSocket {
|
| - public:
|
| - ScopedSocket() : fd_(-1) { Reset(); }
|
| - ~ScopedSocket() { Close(); }
|
| - int fd() { return fd_; }
|
| - void Reset() {
|
| - Close();
|
| - fd_ = SetupSocketOnly();
|
| - }
|
| - void Close() {
|
| - if (fd_ >= 0)
|
| - CloseSocket(fd_);
|
| - fd_ = -1;
|
| - }
|
| - private:
|
| - int fd_;
|
| -};
|
| -
|
| -// Returns a random string for uniquifying profile connections.
|
| -std::string GenerateCookie() {
|
| - return base::Uint64ToString(base::RandUint64());
|
| -}
|
| -
|
| -bool CheckCookie(const base::FilePath& path, const base::FilePath& cookie) {
|
| - return (cookie == ReadLink(path));
|
| -}
|
| -
|
| -bool ConnectSocket(ScopedSocket* socket,
|
| - const base::FilePath& socket_path,
|
| - const base::FilePath& cookie_path) {
|
| - base::FilePath socket_target;
|
| - if (base::ReadSymbolicLink(socket_path, &socket_target)) {
|
| - // It's a symlink. Read the cookie.
|
| - base::FilePath cookie = ReadLink(cookie_path);
|
| - if (cookie.empty())
|
| - return false;
|
| - base::FilePath remote_cookie = socket_target.DirName().
|
| - Append(chrome::kSingletonCookieFilename);
|
| - // Verify the cookie before connecting.
|
| - if (!CheckCookie(remote_cookie, cookie))
|
| - return false;
|
| - // Now we know the directory was (at that point) created by the profile
|
| - // owner. Try to connect.
|
| - sockaddr_un addr;
|
| - SetupSockAddr(socket_path.value(), &addr);
|
| - int ret = HANDLE_EINTR(connect(socket->fd(),
|
| - reinterpret_cast<sockaddr*>(&addr),
|
| - sizeof(addr)));
|
| - if (ret != 0)
|
| - return false;
|
| - // Check the cookie again. We only link in /tmp, which is sticky, so, if the
|
| - // directory is still correct, it must have been correct in-between when we
|
| - // connected. POSIX, sadly, lacks a connectat().
|
| - if (!CheckCookie(remote_cookie, cookie)) {
|
| - socket->Reset();
|
| - return false;
|
| - }
|
| - // Success!
|
| - return true;
|
| - } else if (errno == EINVAL) {
|
| - // It exists, but is not a symlink (or some other error we detect
|
| - // later). Just connect to it directly; this is an older version of Chrome.
|
| - sockaddr_un addr;
|
| - SetupSockAddr(socket_path.value(), &addr);
|
| - int ret = HANDLE_EINTR(connect(socket->fd(),
|
| - reinterpret_cast<sockaddr*>(&addr),
|
| - sizeof(addr)));
|
| - return (ret == 0);
|
| - } else {
|
| - // File is missing, or other error.
|
| - if (errno != ENOENT)
|
| - PLOG(ERROR) << "readlink failed";
|
| - return false;
|
| - }
|
| -}
|
| -
|
| -} // namespace
|
| -
|
| -///////////////////////////////////////////////////////////////////////////////
|
| -// ProcessSingleton::LinuxWatcher
|
| -// A helper class for a Linux specific implementation of the process singleton.
|
| -// This class sets up a listener on the singleton socket and handles parsing
|
| -// messages that come in on the singleton socket.
|
| -class ProcessSingleton::LinuxWatcher
|
| - : public base::MessageLoopForIO::Watcher,
|
| - public base::MessageLoop::DestructionObserver,
|
| - public base::RefCountedThreadSafe<ProcessSingleton::LinuxWatcher,
|
| - BrowserThread::DeleteOnIOThread> {
|
| - public:
|
| - // A helper class to read message from an established socket.
|
| - class SocketReader : public base::MessageLoopForIO::Watcher {
|
| - public:
|
| - SocketReader(ProcessSingleton::LinuxWatcher* parent,
|
| - base::MessageLoop* ui_message_loop,
|
| - int fd)
|
| - : parent_(parent),
|
| - ui_message_loop_(ui_message_loop),
|
| - fd_(fd),
|
| - bytes_read_(0) {
|
| - DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
|
| - // Wait for reads.
|
| - base::MessageLoopForIO::current()->WatchFileDescriptor(
|
| - fd, true, base::MessageLoopForIO::WATCH_READ, &fd_reader_, this);
|
| - // If we haven't completed in a reasonable amount of time, give up.
|
| - timer_.Start(FROM_HERE, base::TimeDelta::FromSeconds(kTimeoutInSeconds),
|
| - this, &SocketReader::CleanupAndDeleteSelf);
|
| - }
|
| -
|
| - virtual ~SocketReader() {
|
| - CloseSocket(fd_);
|
| - }
|
| -
|
| - // MessageLoopForIO::Watcher impl.
|
| - virtual void OnFileCanReadWithoutBlocking(int fd) OVERRIDE;
|
| - virtual void OnFileCanWriteWithoutBlocking(int fd) OVERRIDE {
|
| - // SocketReader only watches for accept (read) events.
|
| - NOTREACHED();
|
| - }
|
| -
|
| - // Finish handling the incoming message by optionally sending back an ACK
|
| - // message and removing this SocketReader.
|
| - void FinishWithACK(const char *message, size_t length);
|
| -
|
| - private:
|
| - void CleanupAndDeleteSelf() {
|
| - DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
|
| -
|
| - parent_->RemoveSocketReader(this);
|
| - // We're deleted beyond this point.
|
| - }
|
| -
|
| - base::MessageLoopForIO::FileDescriptorWatcher fd_reader_;
|
| -
|
| - // The ProcessSingleton::LinuxWatcher that owns us.
|
| - ProcessSingleton::LinuxWatcher* const parent_;
|
| -
|
| - // A reference to the UI message loop.
|
| - base::MessageLoop* const ui_message_loop_;
|
| -
|
| - // The file descriptor we're reading.
|
| - const int fd_;
|
| -
|
| - // Store the message in this buffer.
|
| - char buf_[kMaxMessageLength];
|
| -
|
| - // Tracks the number of bytes we've read in case we're getting partial
|
| - // reads.
|
| - size_t bytes_read_;
|
| -
|
| - base::OneShotTimer<SocketReader> timer_;
|
| -
|
| - DISALLOW_COPY_AND_ASSIGN(SocketReader);
|
| - };
|
| -
|
| - // We expect to only be constructed on the UI thread.
|
| - explicit LinuxWatcher(ProcessSingleton* parent)
|
| - : ui_message_loop_(base::MessageLoop::current()),
|
| - parent_(parent) {
|
| - }
|
| -
|
| - // Start listening for connections on the socket. This method should be
|
| - // called from the IO thread.
|
| - void StartListening(int socket);
|
| -
|
| - // This method determines if we should use the same process and if we should,
|
| - // opens a new browser tab. This runs on the UI thread.
|
| - // |reader| is for sending back ACK message.
|
| - void HandleMessage(const std::string& current_dir,
|
| - const std::vector<std::string>& argv,
|
| - SocketReader* reader);
|
| -
|
| - // MessageLoopForIO::Watcher impl. These run on the IO thread.
|
| - virtual void OnFileCanReadWithoutBlocking(int fd) OVERRIDE;
|
| - virtual void OnFileCanWriteWithoutBlocking(int fd) OVERRIDE {
|
| - // ProcessSingleton only watches for accept (read) events.
|
| - NOTREACHED();
|
| - }
|
| -
|
| - // MessageLoop::DestructionObserver
|
| - virtual void WillDestroyCurrentMessageLoop() OVERRIDE {
|
| - fd_watcher_.StopWatchingFileDescriptor();
|
| - }
|
| -
|
| - private:
|
| - friend struct BrowserThread::DeleteOnThread<BrowserThread::IO>;
|
| - friend class base::DeleteHelper<ProcessSingleton::LinuxWatcher>;
|
| -
|
| - virtual ~LinuxWatcher() {
|
| - DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
|
| - STLDeleteElements(&readers_);
|
| -
|
| - base::MessageLoopForIO* ml = base::MessageLoopForIO::current();
|
| - ml->RemoveDestructionObserver(this);
|
| - }
|
| -
|
| - // Removes and deletes the SocketReader.
|
| - void RemoveSocketReader(SocketReader* reader);
|
| -
|
| - base::MessageLoopForIO::FileDescriptorWatcher fd_watcher_;
|
| -
|
| - // A reference to the UI message loop (i.e., the message loop we were
|
| - // constructed on).
|
| - base::MessageLoop* ui_message_loop_;
|
| -
|
| - // The ProcessSingleton that owns us.
|
| - ProcessSingleton* const parent_;
|
| -
|
| - std::set<SocketReader*> readers_;
|
| -
|
| - DISALLOW_COPY_AND_ASSIGN(LinuxWatcher);
|
| -};
|
| -
|
| -void ProcessSingleton::LinuxWatcher::OnFileCanReadWithoutBlocking(int fd) {
|
| - DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
|
| - // Accepting incoming client.
|
| - sockaddr_un from;
|
| - socklen_t from_len = sizeof(from);
|
| - int connection_socket = HANDLE_EINTR(accept(
|
| - fd, reinterpret_cast<sockaddr*>(&from), &from_len));
|
| - if (-1 == connection_socket) {
|
| - PLOG(ERROR) << "accept() failed";
|
| - return;
|
| - }
|
| - int rv = SetNonBlocking(connection_socket);
|
| - DCHECK_EQ(0, rv) << "Failed to make non-blocking socket.";
|
| - SocketReader* reader = new SocketReader(this,
|
| - ui_message_loop_,
|
| - connection_socket);
|
| - readers_.insert(reader);
|
| -}
|
| -
|
| -void ProcessSingleton::LinuxWatcher::StartListening(int socket) {
|
| - DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
|
| - // Watch for client connections on this socket.
|
| - base::MessageLoopForIO* ml = base::MessageLoopForIO::current();
|
| - ml->AddDestructionObserver(this);
|
| - ml->WatchFileDescriptor(socket, true, base::MessageLoopForIO::WATCH_READ,
|
| - &fd_watcher_, this);
|
| -}
|
| -
|
| -void ProcessSingleton::LinuxWatcher::HandleMessage(
|
| - const std::string& current_dir, const std::vector<std::string>& argv,
|
| - SocketReader* reader) {
|
| - DCHECK(ui_message_loop_ == base::MessageLoop::current());
|
| - DCHECK(reader);
|
| -
|
| - if (parent_->notification_callback_.Run(CommandLine(argv),
|
| - base::FilePath(current_dir))) {
|
| - // Send back "ACK" message to prevent the client process from starting up.
|
| - reader->FinishWithACK(kACKToken, arraysize(kACKToken) - 1);
|
| - } else {
|
| - LOG(WARNING) << "Not handling interprocess notification as browser"
|
| - " is shutting down";
|
| - // Send back "SHUTDOWN" message, so that the client process can start up
|
| - // without killing this process.
|
| - reader->FinishWithACK(kShutdownToken, arraysize(kShutdownToken) - 1);
|
| - return;
|
| - }
|
| -}
|
| -
|
| -void ProcessSingleton::LinuxWatcher::RemoveSocketReader(SocketReader* reader) {
|
| - DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
|
| - DCHECK(reader);
|
| - readers_.erase(reader);
|
| - delete reader;
|
| -}
|
| -
|
| -///////////////////////////////////////////////////////////////////////////////
|
| -// ProcessSingleton::LinuxWatcher::SocketReader
|
| -//
|
| -
|
| -void ProcessSingleton::LinuxWatcher::SocketReader::OnFileCanReadWithoutBlocking(
|
| - int fd) {
|
| - DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
|
| - DCHECK_EQ(fd, fd_);
|
| - while (bytes_read_ < sizeof(buf_)) {
|
| - ssize_t rv = HANDLE_EINTR(
|
| - read(fd, buf_ + bytes_read_, sizeof(buf_) - bytes_read_));
|
| - if (rv < 0) {
|
| - if (errno != EAGAIN && errno != EWOULDBLOCK) {
|
| - PLOG(ERROR) << "read() failed";
|
| - CloseSocket(fd);
|
| - return;
|
| - } else {
|
| - // It would block, so we just return and continue to watch for the next
|
| - // opportunity to read.
|
| - return;
|
| - }
|
| - } else if (!rv) {
|
| - // No more data to read. It's time to process the message.
|
| - break;
|
| - } else {
|
| - bytes_read_ += rv;
|
| - }
|
| - }
|
| -
|
| - // Validate the message. The shortest message is kStartToken\0x\0x
|
| - const size_t kMinMessageLength = arraysize(kStartToken) + 4;
|
| - if (bytes_read_ < kMinMessageLength) {
|
| - buf_[bytes_read_] = 0;
|
| - LOG(ERROR) << "Invalid socket message (wrong length):" << buf_;
|
| - CleanupAndDeleteSelf();
|
| - return;
|
| - }
|
| -
|
| - std::string str(buf_, bytes_read_);
|
| - std::vector<std::string> tokens;
|
| - base::SplitString(str, kTokenDelimiter, &tokens);
|
| -
|
| - if (tokens.size() < 3 || tokens[0] != kStartToken) {
|
| - LOG(ERROR) << "Wrong message format: " << str;
|
| - CleanupAndDeleteSelf();
|
| - return;
|
| - }
|
| -
|
| - // Stop the expiration timer to prevent this SocketReader object from being
|
| - // terminated unexpectly.
|
| - timer_.Stop();
|
| -
|
| - std::string current_dir = tokens[1];
|
| - // Remove the first two tokens. The remaining tokens should be the command
|
| - // line argv array.
|
| - tokens.erase(tokens.begin());
|
| - tokens.erase(tokens.begin());
|
| -
|
| - // Return to the UI thread to handle opening a new browser tab.
|
| - ui_message_loop_->PostTask(FROM_HERE, base::Bind(
|
| - &ProcessSingleton::LinuxWatcher::HandleMessage,
|
| - parent_,
|
| - current_dir,
|
| - tokens,
|
| - this));
|
| - fd_reader_.StopWatchingFileDescriptor();
|
| -
|
| - // LinuxWatcher::HandleMessage() is in charge of destroying this SocketReader
|
| - // object by invoking SocketReader::FinishWithACK().
|
| -}
|
| -
|
| -void ProcessSingleton::LinuxWatcher::SocketReader::FinishWithACK(
|
| - const char *message, size_t length) {
|
| - if (message && length) {
|
| - // Not necessary to care about the return value.
|
| - WriteToSocket(fd_, message, length);
|
| - }
|
| -
|
| - if (shutdown(fd_, SHUT_WR) < 0)
|
| - PLOG(ERROR) << "shutdown() failed";
|
| -
|
| - BrowserThread::PostTask(
|
| - BrowserThread::IO,
|
| - FROM_HERE,
|
| - base::Bind(&ProcessSingleton::LinuxWatcher::RemoveSocketReader,
|
| - parent_,
|
| - this));
|
| - // We will be deleted once the posted RemoveSocketReader task runs.
|
| -}
|
| -
|
| -///////////////////////////////////////////////////////////////////////////////
|
| -// ProcessSingleton
|
| -//
|
| -ProcessSingleton::ProcessSingleton(
|
| - const base::FilePath& user_data_dir,
|
| - const NotificationCallback& notification_callback)
|
| - : notification_callback_(notification_callback),
|
| - current_pid_(base::GetCurrentProcId()),
|
| - watcher_(new LinuxWatcher(this)) {
|
| - socket_path_ = user_data_dir.Append(chrome::kSingletonSocketFilename);
|
| - lock_path_ = user_data_dir.Append(chrome::kSingletonLockFilename);
|
| - cookie_path_ = user_data_dir.Append(chrome::kSingletonCookieFilename);
|
| -
|
| - kill_callback_ = base::Bind(&ProcessSingleton::KillProcess,
|
| - base::Unretained(this));
|
| -}
|
| -
|
| -ProcessSingleton::~ProcessSingleton() {
|
| -}
|
| -
|
| -ProcessSingleton::NotifyResult ProcessSingleton::NotifyOtherProcess() {
|
| - return NotifyOtherProcessWithTimeout(*CommandLine::ForCurrentProcess(),
|
| - kTimeoutInSeconds,
|
| - true);
|
| -}
|
| -
|
| -ProcessSingleton::NotifyResult ProcessSingleton::NotifyOtherProcessWithTimeout(
|
| - const CommandLine& cmd_line,
|
| - int timeout_seconds,
|
| - bool kill_unresponsive) {
|
| - DCHECK_GE(timeout_seconds, 0);
|
| -
|
| - ScopedSocket socket;
|
| - for (int retries = 0; retries <= timeout_seconds; ++retries) {
|
| - // Try to connect to the socket.
|
| - if (ConnectSocket(&socket, socket_path_, cookie_path_))
|
| - break;
|
| -
|
| - // If we're in a race with another process, they may be in Create() and have
|
| - // created the lock but not attached to the socket. So we check if the
|
| - // process with the pid from the lockfile is currently running and is a
|
| - // chrome browser. If so, we loop and try again for |timeout_seconds|.
|
| -
|
| - std::string hostname;
|
| - int pid;
|
| - if (!ParseLockPath(lock_path_, &hostname, &pid)) {
|
| - // No lockfile exists.
|
| - return PROCESS_NONE;
|
| - }
|
| -
|
| - if (hostname.empty()) {
|
| - // Invalid lockfile.
|
| - UnlinkPath(lock_path_);
|
| - return PROCESS_NONE;
|
| - }
|
| -
|
| - if (hostname != net::GetHostName() && !IsChromeProcess(pid)) {
|
| - // Locked by process on another host. If the user selected to unlock
|
| - // the profile, try to continue; otherwise quit.
|
| - if (DisplayProfileInUseError(lock_path_, hostname, pid)) {
|
| - UnlinkPath(lock_path_);
|
| - return PROCESS_NONE;
|
| - }
|
| - return PROFILE_IN_USE;
|
| - }
|
| -
|
| - if (!IsChromeProcess(pid)) {
|
| - // Orphaned lockfile (no process with pid, or non-chrome process.)
|
| - UnlinkPath(lock_path_);
|
| - return PROCESS_NONE;
|
| - }
|
| -
|
| - if (IsSameChromeInstance(pid)) {
|
| - // Orphaned lockfile (pid is part of same chrome instance we are, even
|
| - // though we haven't tried to create a lockfile yet).
|
| - UnlinkPath(lock_path_);
|
| - return PROCESS_NONE;
|
| - }
|
| -
|
| - if (retries == timeout_seconds) {
|
| - // Retries failed. Kill the unresponsive chrome process and continue.
|
| - if (!kill_unresponsive || !KillProcessByLockPath())
|
| - return PROFILE_IN_USE;
|
| - return PROCESS_NONE;
|
| - }
|
| -
|
| - base::PlatformThread::Sleep(base::TimeDelta::FromSeconds(1));
|
| - }
|
| -
|
| - timeval timeout = {timeout_seconds, 0};
|
| - setsockopt(socket.fd(), SOL_SOCKET, SO_SNDTIMEO, &timeout, sizeof(timeout));
|
| -
|
| - // Found another process, prepare our command line
|
| - // format is "START\0<current dir>\0<argv[0]>\0...\0<argv[n]>".
|
| - std::string to_send(kStartToken);
|
| - to_send.push_back(kTokenDelimiter);
|
| -
|
| - base::FilePath current_dir;
|
| - if (!PathService::Get(base::DIR_CURRENT, ¤t_dir))
|
| - return PROCESS_NONE;
|
| - to_send.append(current_dir.value());
|
| -
|
| - const std::vector<std::string>& argv = cmd_line.argv();
|
| - for (std::vector<std::string>::const_iterator it = argv.begin();
|
| - it != argv.end(); ++it) {
|
| - to_send.push_back(kTokenDelimiter);
|
| - to_send.append(*it);
|
| - }
|
| -
|
| - // Send the message
|
| - if (!WriteToSocket(socket.fd(), to_send.data(), to_send.length())) {
|
| - // Try to kill the other process, because it might have been dead.
|
| - if (!kill_unresponsive || !KillProcessByLockPath())
|
| - return PROFILE_IN_USE;
|
| - return PROCESS_NONE;
|
| - }
|
| -
|
| - if (shutdown(socket.fd(), SHUT_WR) < 0)
|
| - PLOG(ERROR) << "shutdown() failed";
|
| -
|
| - // Read ACK message from the other process. It might be blocked for a certain
|
| - // timeout, to make sure the other process has enough time to return ACK.
|
| - char buf[kMaxACKMessageLength + 1];
|
| - ssize_t len =
|
| - ReadFromSocket(socket.fd(), buf, kMaxACKMessageLength, timeout_seconds);
|
| -
|
| - // Failed to read ACK, the other process might have been frozen.
|
| - if (len <= 0) {
|
| - if (!kill_unresponsive || !KillProcessByLockPath())
|
| - return PROFILE_IN_USE;
|
| - return PROCESS_NONE;
|
| - }
|
| -
|
| - buf[len] = '\0';
|
| - if (strncmp(buf, kShutdownToken, arraysize(kShutdownToken) - 1) == 0) {
|
| - // The other process is shutting down, it's safe to start a new process.
|
| - return PROCESS_NONE;
|
| - } else if (strncmp(buf, kACKToken, arraysize(kACKToken) - 1) == 0) {
|
| -#if defined(TOOLKIT_VIEWS) && !defined(OS_CHROMEOS)
|
| - // Likely NULL in unit tests.
|
| - views::LinuxUI* linux_ui = views::LinuxUI::instance();
|
| - if (linux_ui)
|
| - linux_ui->NotifyWindowManagerStartupComplete();
|
| -#endif
|
| -
|
| - // Assume the other process is handling the request.
|
| - return PROCESS_NOTIFIED;
|
| - }
|
| -
|
| - NOTREACHED() << "The other process returned unknown message: " << buf;
|
| - return PROCESS_NOTIFIED;
|
| -}
|
| -
|
| -ProcessSingleton::NotifyResult ProcessSingleton::NotifyOtherProcessOrCreate() {
|
| - return NotifyOtherProcessWithTimeoutOrCreate(
|
| - *CommandLine::ForCurrentProcess(),
|
| - kTimeoutInSeconds);
|
| -}
|
| -
|
| -ProcessSingleton::NotifyResult
|
| -ProcessSingleton::NotifyOtherProcessWithTimeoutOrCreate(
|
| - const CommandLine& command_line,
|
| - int timeout_seconds) {
|
| - NotifyResult result = NotifyOtherProcessWithTimeout(command_line,
|
| - timeout_seconds, true);
|
| - if (result != PROCESS_NONE)
|
| - return result;
|
| - if (Create())
|
| - return PROCESS_NONE;
|
| - // If the Create() failed, try again to notify. (It could be that another
|
| - // instance was starting at the same time and managed to grab the lock before
|
| - // we did.)
|
| - // This time, we don't want to kill anything if we aren't successful, since we
|
| - // aren't going to try to take over the lock ourselves.
|
| - result = NotifyOtherProcessWithTimeout(command_line, timeout_seconds, false);
|
| - if (result != PROCESS_NONE)
|
| - return result;
|
| -
|
| - return LOCK_ERROR;
|
| -}
|
| -
|
| -void ProcessSingleton::OverrideCurrentPidForTesting(base::ProcessId pid) {
|
| - current_pid_ = pid;
|
| -}
|
| -
|
| -void ProcessSingleton::OverrideKillCallbackForTesting(
|
| - const base::Callback<void(int)>& callback) {
|
| - kill_callback_ = callback;
|
| -}
|
| -
|
| -void ProcessSingleton::DisablePromptForTesting() {
|
| - g_disable_prompt = true;
|
| -}
|
| -
|
| -bool ProcessSingleton::Create() {
|
| - int sock;
|
| - sockaddr_un addr;
|
| -
|
| - // The symlink lock is pointed to the hostname and process id, so other
|
| - // processes can find it out.
|
| - base::FilePath symlink_content(base::StringPrintf(
|
| - "%s%c%u",
|
| - net::GetHostName().c_str(),
|
| - kLockDelimiter,
|
| - current_pid_));
|
| -
|
| - // Create symbol link before binding the socket, to ensure only one instance
|
| - // can have the socket open.
|
| - if (!SymlinkPath(symlink_content, lock_path_)) {
|
| - // If we failed to create the lock, most likely another instance won the
|
| - // startup race.
|
| - return false;
|
| - }
|
| -
|
| - // Create the socket file somewhere in /tmp which is usually mounted as a
|
| - // normal filesystem. Some network filesystems (notably AFS) are screwy and
|
| - // do not support Unix domain sockets.
|
| - if (!socket_dir_.CreateUniqueTempDir()) {
|
| - LOG(ERROR) << "Failed to create socket directory.";
|
| - return false;
|
| - }
|
| - // Setup the socket symlink and the two cookies.
|
| - base::FilePath socket_target_path =
|
| - socket_dir_.path().Append(chrome::kSingletonSocketFilename);
|
| - base::FilePath cookie(GenerateCookie());
|
| - base::FilePath remote_cookie_path =
|
| - socket_dir_.path().Append(chrome::kSingletonCookieFilename);
|
| - UnlinkPath(socket_path_);
|
| - UnlinkPath(cookie_path_);
|
| - if (!SymlinkPath(socket_target_path, socket_path_) ||
|
| - !SymlinkPath(cookie, cookie_path_) ||
|
| - !SymlinkPath(cookie, remote_cookie_path)) {
|
| - // We've already locked things, so we can't have lost the startup race,
|
| - // but something doesn't like us.
|
| - LOG(ERROR) << "Failed to create symlinks.";
|
| - if (!socket_dir_.Delete())
|
| - LOG(ERROR) << "Encountered a problem when deleting socket directory.";
|
| - return false;
|
| - }
|
| -
|
| - SetupSocket(socket_target_path.value(), &sock, &addr);
|
| -
|
| - if (bind(sock, reinterpret_cast<sockaddr*>(&addr), sizeof(addr)) < 0) {
|
| - PLOG(ERROR) << "Failed to bind() " << socket_target_path.value();
|
| - CloseSocket(sock);
|
| - return false;
|
| - }
|
| -
|
| - if (listen(sock, 5) < 0)
|
| - NOTREACHED() << "listen failed: " << safe_strerror(errno);
|
| -
|
| - DCHECK(BrowserThread::IsMessageLoopValid(BrowserThread::IO));
|
| - BrowserThread::PostTask(
|
| - BrowserThread::IO,
|
| - FROM_HERE,
|
| - base::Bind(&ProcessSingleton::LinuxWatcher::StartListening,
|
| - watcher_.get(),
|
| - sock));
|
| -
|
| - return true;
|
| -}
|
| -
|
| -void ProcessSingleton::Cleanup() {
|
| - UnlinkPath(socket_path_);
|
| - UnlinkPath(cookie_path_);
|
| - UnlinkPath(lock_path_);
|
| -}
|
| -
|
| -bool ProcessSingleton::IsSameChromeInstance(pid_t pid) {
|
| - pid_t cur_pid = current_pid_;
|
| - while (pid != cur_pid) {
|
| - pid = base::GetParentProcessId(pid);
|
| - if (pid < 0)
|
| - return false;
|
| - if (!IsChromeProcess(pid))
|
| - return false;
|
| - }
|
| - return true;
|
| -}
|
| -
|
| -bool ProcessSingleton::KillProcessByLockPath() {
|
| - std::string hostname;
|
| - int pid;
|
| - ParseLockPath(lock_path_, &hostname, &pid);
|
| -
|
| - if (!hostname.empty() && hostname != net::GetHostName()) {
|
| - return DisplayProfileInUseError(lock_path_, hostname, pid);
|
| - }
|
| - UnlinkPath(lock_path_);
|
| -
|
| - if (IsSameChromeInstance(pid))
|
| - return true;
|
| -
|
| - if (pid > 0) {
|
| - kill_callback_.Run(pid);
|
| - return true;
|
| - }
|
| -
|
| - LOG(ERROR) << "Failed to extract pid from path: " << lock_path_.value();
|
| - return true;
|
| -}
|
| -
|
| -void ProcessSingleton::KillProcess(int pid) {
|
| - // TODO(james.su@gmail.com): Is SIGKILL ok?
|
| - int rv = kill(static_cast<base::ProcessHandle>(pid), SIGKILL);
|
| - // ESRCH = No Such Process (can happen if the other process is already in
|
| - // progress of shutting down and finishes before we try to kill it).
|
| - DCHECK(rv == 0 || errno == ESRCH) << "Error killing process: "
|
| - << safe_strerror(errno);
|
| -}
|
|
|