15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// On Linux, when the user tries to launch a second copy of chrome, we check 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// for a socket in the user's profile directory. If the socket file is open we 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// send a message to the first chrome browser process with the current 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// directory and second process command line flags. The second process then 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// exits. 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Because many networked filesystem implementations do not support unix domain 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// sockets, we create the socket in a temporary directory and create a symlink 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// in the profile. This temporary directory is no longer bound to the profile, 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// and may disappear across a reboot or login to a separate session. To bind 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// them, we store a unique cookie in the profile directory, which must also be 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// present in the remote directory to connect. The cookie is checked both before 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// and after the connection. /tmp is sticky, and different Chrome sessions use 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// different cookies. Thus, a matching cookie before and after means the 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// connection was to a directory with a valid cookie. 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// We also have a lock file, which is a symlink to a non-existent destination. 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The destination is a string containing the hostname and process id of 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// chrome's browser process, eg. "SingletonLock -> example.com-9156". When the 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// first copy of chrome exits it will delete the lock file on shutdown, so that 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// a different instance on a different host may then use the profile directory. 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// If writing to the socket fails, the hostname in the lock is checked to see if 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// another instance is running a different host using a shared filesystem (nfs, 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// etc.) If the hostname differs an error is displayed and the second process 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// exits. Otherwise the first process (if any) is killed and the second process 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// starts as normal. 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// When the second process sends the current directory and command line flags to 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// the first process, it waits for an ACK message back from the first process 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// for a certain time. If there is no ACK message back in time, then the first 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// process will be considered as hung for some reason. The second process then 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// retrieves the process id from the symbol link and kills it by sending 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// SIGKILL. Then the second process starts as normal. 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/process_singleton.h" 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <errno.h> 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <fcntl.h> 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(TOOLKIT_GTK) 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <gdk/gdk.h> 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <signal.h> 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <sys/socket.h> 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <sys/stat.h> 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <sys/types.h> 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <sys/un.h> 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <unistd.h> 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <cstring> 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <set> 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <string> 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/base_paths.h" 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/basictypes.h" 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind.h" 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/command_line.h" 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/file_util.h" 632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/files/file_path.h" 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h" 659ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch#include "base/message_loop/message_loop.h" 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/path_service.h" 672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/posix/eintr_wrapper.h" 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/rand_util.h" 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/safe_strerror_posix.h" 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/sequenced_task_runner_helpers.h" 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/stl_util.h" 722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/strings/string_number_conversions.h" 732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/strings/string_split.h" 74868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/stringprintf.h" 75c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "base/strings/sys_string_conversions.h" 76868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/utf_string_conversions.h" 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/threading/platform_thread.h" 78eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch#include "base/time/time.h" 79eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch#include "base/timer/timer.h" 80f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)#include "chrome/browser/ui/process_singleton_dialog_linux.h" 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/common/chrome_constants.h" 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/browser/browser_thread.h" 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "grit/chromium_strings.h" 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "grit/generated_resources.h" 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/net_util.h" 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/base/l10n/l10n_util.h" 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using content::BrowserThread; 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int ProcessSingleton::kTimeoutInSeconds; 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static bool g_disable_prompt; 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kStartToken[] = "START"; 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kACKToken[] = "ACK"; 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kShutdownToken[] = "SHUTDOWN"; 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kTokenDelimiter = '\0'; 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int kMaxMessageLength = 32 * 1024; 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int kMaxACKMessageLength = arraysize(kShutdownToken) - 1; 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kLockDelimiter = '-'; 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Set a file descriptor to be non-blocking. 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Return 0 on success, -1 on failure. 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int SetNonBlocking(int fd) { 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int flags = fcntl(fd, F_GETFL, 0); 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (-1 == flags) 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return flags; 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (flags & O_NONBLOCK) 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0; 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return fcntl(fd, F_SETFL, flags | O_NONBLOCK); 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Set the close-on-exec bit on a file descriptor. 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Returns 0 on success, -1 on failure. 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int SetCloseOnExec(int fd) { 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int flags = fcntl(fd, F_GETFD, 0); 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (-1 == flags) 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return flags; 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (flags & FD_CLOEXEC) 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0; 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return fcntl(fd, F_SETFD, flags | FD_CLOEXEC); 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Close a socket and check return value. 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void CloseSocket(int fd) { 128a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) int rv = IGNORE_EINTR(close(fd)); 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_EQ(0, rv) << "Error closing socket: " << safe_strerror(errno); 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Write a message to a socket fd. 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool WriteToSocket(int fd, const char *message, size_t length) { 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(message); 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(length); 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t bytes_written = 0; 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) do { 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ssize_t rv = HANDLE_EINTR( 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) write(fd, message + bytes_written, length - bytes_written)); 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (rv < 0) { 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (errno == EAGAIN || errno == EWOULDBLOCK) { 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The socket shouldn't block, we're sending so little data. Just give 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // up here, since NotifyOtherProcess() doesn't have an asynchronous api. 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "ProcessSingleton would block on write(), so it gave up."; 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PLOG(ERROR) << "write() failed"; 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bytes_written += rv; 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } while (bytes_written < length); 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Wait a socket for read for a certain timeout in seconds. 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Returns -1 if error occurred, 0 if timeout reached, > 0 if the socket is 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// ready for read. 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int WaitSocketForRead(int fd, int timeout) { 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fd_set read_fds; 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) struct timeval tv; 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FD_ZERO(&read_fds); 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FD_SET(fd, &read_fds); 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tv.tv_sec = timeout; 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tv.tv_usec = 0; 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return HANDLE_EINTR(select(fd + 1, &read_fds, NULL, NULL, &tv)); 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Read a message from a socket fd, with an optional timeout in seconds. 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// If |timeout| <= 0 then read immediately. 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Return number of bytes actually read, or -1 on error. 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ssize_t ReadFromSocket(int fd, char *buf, size_t bufsize, int timeout) { 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (timeout > 0) { 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int rv = WaitSocketForRead(fd, timeout); 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (rv <= 0) 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return rv; 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t bytes_read = 0; 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) do { 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ssize_t rv = HANDLE_EINTR(read(fd, buf + bytes_read, bufsize - bytes_read)); 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (rv < 0) { 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (errno != EAGAIN && errno != EWOULDBLOCK) { 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PLOG(ERROR) << "read() failed"; 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return rv; 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // It would block, so we just return what has been read. 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return bytes_read; 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else if (!rv) { 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // No more data to read. 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return bytes_read; 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bytes_read += rv; 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } while (bytes_read < bufsize); 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return bytes_read; 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Set up a sockaddr appropriate for messaging. 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void SetupSockAddr(const std::string& path, struct sockaddr_un* addr) { 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) addr->sun_family = AF_UNIX; 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CHECK(path.length() < arraysize(addr->sun_path)) 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) << "Socket path too long: " << path; 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::strlcpy(addr->sun_path, path.c_str(), arraysize(addr->sun_path)); 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Set up a socket appropriate for messaging. 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int SetupSocketOnly() { 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int sock = socket(PF_UNIX, SOCK_STREAM, 0); 2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PCHECK(sock >= 0) << "socket() failed"; 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int rv = SetNonBlocking(sock); 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_EQ(0, rv) << "Failed to make non-blocking socket."; 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) rv = SetCloseOnExec(sock); 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_EQ(0, rv) << "Failed to set CLOEXEC on socket."; 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return sock; 2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Set up a socket and sockaddr appropriate for messaging. 2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void SetupSocket(const std::string& path, int* sock, struct sockaddr_un* addr) { 2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *sock = SetupSocketOnly(); 2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetupSockAddr(path, addr); 2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Read a symbolic link, return empty string if given path is not a symbol link. 2312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)base::FilePath ReadLink(const base::FilePath& path) { 2322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath target; 233f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) if (!base::ReadSymbolicLink(path, &target)) { 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The only errno that should occur is ENOENT. 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (errno != 0 && errno != ENOENT) 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PLOG(ERROR) << "readlink(" << path.value() << ") failed"; 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return target; 2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Unlink a path. Return true on success. 2422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool UnlinkPath(const base::FilePath& path) { 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int rv = unlink(path.value().c_str()); 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (rv < 0 && errno != ENOENT) 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PLOG(ERROR) << "Failed to unlink " << path.value(); 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return rv == 0; 2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Create a symlink. Returns true on success. 2512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool SymlinkPath(const base::FilePath& target, const base::FilePath& path) { 252f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) if (!base::CreateSymbolicLink(target, path)) { 2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Double check the value in case symlink suceeded but we got an incorrect 2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // failure due to NFS packet loss & retry. 2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int saved_errno = errno; 2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (ReadLink(path) != target) { 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If we failed to create the lock, most likely another instance won the 2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // startup race. 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) errno = saved_errno; 2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PLOG(ERROR) << "Failed to create " << path.value(); 2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Extract the hostname and pid from the lock symlink. 2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Returns true if the lock existed. 2692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool ParseLockPath(const base::FilePath& path, 2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string* hostname, 2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int* pid) { 2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string real_path = ReadLink(path).value(); 2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (real_path.empty()) 2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string::size_type pos = real_path.rfind(kLockDelimiter); 2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If the path is not a symbolic link, or doesn't contain what we expect, 2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // bail. 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (pos == std::string::npos) { 2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *hostname = ""; 2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *pid = -1; 2835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *hostname = real_path.substr(0, pos); 2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& pid_str = real_path.substr(pos + 1); 2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!base::StringToInt(pid_str, pid)) 2905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *pid = -1; 2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 29558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles)// Returns true if the user opted to unlock the profile. 29658537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles)bool DisplayProfileInUseError(const base::FilePath& lock_path, 2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& hostname, 2985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int pid) { 299a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) base::string16 error = l10n_util::GetStringFUTF16( 3005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) IDS_PROFILE_IN_USE_LINUX, 3015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::IntToString16(pid), 30258537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) ASCIIToUTF16(hostname)); 303a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) base::string16 relaunch_button_text = l10n_util::GetStringUTF16( 30458537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) IDS_PROFILE_IN_USE_LINUX_RELAUNCH); 3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << base::SysWideToNativeMB(UTF16ToWide(error)).c_str(); 306f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) if (!g_disable_prompt) 307f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) return ShowProcessSingletonDialog(error, relaunch_button_text); 30858537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) return false; 3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool IsChromeProcess(pid_t pid) { 3122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath other_chrome_path(base::GetProcessExecutablePath(pid)); 3135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return (!other_chrome_path.empty() && 3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) other_chrome_path.BaseName() == 3152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath(chrome::kBrowserProcessExecutableName)); 3165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// A helper class to hold onto a socket. 3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class ScopedSocket { 3205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 3215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ScopedSocket() : fd_(-1) { Reset(); } 3225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ~ScopedSocket() { Close(); } 3235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int fd() { return fd_; } 3245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void Reset() { 3255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Close(); 3265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fd_ = SetupSocketOnly(); 3275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void Close() { 3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (fd_ >= 0) 3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CloseSocket(fd_); 3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fd_ = -1; 3325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 3345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int fd_; 3355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 3365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Returns a random string for uniquifying profile connections. 3385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)std::string GenerateCookie() { 3395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return base::Uint64ToString(base::RandUint64()); 3405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool CheckCookie(const base::FilePath& path, const base::FilePath& cookie) { 3435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return (cookie == ReadLink(path)); 3445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool ConnectSocket(ScopedSocket* socket, 3472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& socket_path, 3482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& cookie_path) { 3492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath socket_target; 350f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) if (base::ReadSymbolicLink(socket_path, &socket_target)) { 3515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // It's a symlink. Read the cookie. 3522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath cookie = ReadLink(cookie_path); 3535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (cookie.empty()) 3545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 3552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath remote_cookie = socket_target.DirName(). 3565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Append(chrome::kSingletonCookieFilename); 3575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Verify the cookie before connecting. 3585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!CheckCookie(remote_cookie, cookie)) 3595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 3605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Now we know the directory was (at that point) created by the profile 3615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // owner. Try to connect. 3625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sockaddr_un addr; 3635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetupSockAddr(socket_path.value(), &addr); 3645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int ret = HANDLE_EINTR(connect(socket->fd(), 3655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) reinterpret_cast<sockaddr*>(&addr), 3665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sizeof(addr))); 3675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (ret != 0) 3685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 3695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Check the cookie again. We only link in /tmp, which is sticky, so, if the 3705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // directory is still correct, it must have been correct in-between when we 3715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // connected. POSIX, sadly, lacks a connectat(). 3725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!CheckCookie(remote_cookie, cookie)) { 3735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket->Reset(); 3745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 3755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Success! 3775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 3785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else if (errno == EINVAL) { 3795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // It exists, but is not a symlink (or some other error we detect 3805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // later). Just connect to it directly; this is an older version of Chrome. 3815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sockaddr_un addr; 3825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetupSockAddr(socket_path.value(), &addr); 3835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int ret = HANDLE_EINTR(connect(socket->fd(), 3845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) reinterpret_cast<sockaddr*>(&addr), 3855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sizeof(addr))); 3865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return (ret == 0); 3875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 3885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // File is missing, or other error. 3895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (errno != ENOENT) 3905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PLOG(ERROR) << "readlink failed"; 3915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 3925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 3965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)/////////////////////////////////////////////////////////////////////////////// 3985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// ProcessSingleton::LinuxWatcher 3995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// A helper class for a Linux specific implementation of the process singleton. 4005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// This class sets up a listener on the singleton socket and handles parsing 4015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// messages that come in on the singleton socket. 4025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class ProcessSingleton::LinuxWatcher 40390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) : public base::MessageLoopForIO::Watcher, 40490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) public base::MessageLoop::DestructionObserver, 4055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public base::RefCountedThreadSafe<ProcessSingleton::LinuxWatcher, 4065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BrowserThread::DeleteOnIOThread> { 4075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 4085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // A helper class to read message from an established socket. 40990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) class SocketReader : public base::MessageLoopForIO::Watcher { 4105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 4115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketReader(ProcessSingleton::LinuxWatcher* parent, 41290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) base::MessageLoop* ui_message_loop, 4135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int fd) 4145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : parent_(parent), 4155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ui_message_loop_(ui_message_loop), 4165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fd_(fd), 4175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bytes_read_(0) { 4185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 4195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Wait for reads. 42090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) base::MessageLoopForIO::current()->WatchFileDescriptor( 42190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) fd, true, base::MessageLoopForIO::WATCH_READ, &fd_reader_, this); 4225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If we haven't completed in a reasonable amount of time, give up. 4235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timer_.Start(FROM_HERE, base::TimeDelta::FromSeconds(kTimeoutInSeconds), 4245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) this, &SocketReader::CleanupAndDeleteSelf); 4255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~SocketReader() { 4285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CloseSocket(fd_); 4295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // MessageLoopForIO::Watcher impl. 4322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void OnFileCanReadWithoutBlocking(int fd) OVERRIDE; 4332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void OnFileCanWriteWithoutBlocking(int fd) OVERRIDE { 4345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // SocketReader only watches for accept (read) events. 4355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED(); 4365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Finish handling the incoming message by optionally sending back an ACK 4395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // message and removing this SocketReader. 4405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void FinishWithACK(const char *message, size_t length); 4415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 4435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void CleanupAndDeleteSelf() { 4445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 4455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parent_->RemoveSocketReader(this); 4475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We're deleted beyond this point. 4485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 45090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) base::MessageLoopForIO::FileDescriptorWatcher fd_reader_; 4515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The ProcessSingleton::LinuxWatcher that owns us. 4535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ProcessSingleton::LinuxWatcher* const parent_; 4545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // A reference to the UI message loop. 45690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) base::MessageLoop* const ui_message_loop_; 4575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The file descriptor we're reading. 4595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const int fd_; 4605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Store the message in this buffer. 4625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) char buf_[kMaxMessageLength]; 4635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Tracks the number of bytes we've read in case we're getting partial 4655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // reads. 4665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t bytes_read_; 4675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::OneShotTimer<SocketReader> timer_; 4695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(SocketReader); 4715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 4725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We expect to only be constructed on the UI thread. 4745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) explicit LinuxWatcher(ProcessSingleton* parent) 47590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) : ui_message_loop_(base::MessageLoop::current()), 4765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parent_(parent) { 4775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Start listening for connections on the socket. This method should be 4805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // called from the IO thread. 4815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void StartListening(int socket); 4825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This method determines if we should use the same process and if we should, 4845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // opens a new browser tab. This runs on the UI thread. 4855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // |reader| is for sending back ACK message. 4865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void HandleMessage(const std::string& current_dir, 4875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::vector<std::string>& argv, 4885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketReader* reader); 4895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // MessageLoopForIO::Watcher impl. These run on the IO thread. 4912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void OnFileCanReadWithoutBlocking(int fd) OVERRIDE; 4922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void OnFileCanWriteWithoutBlocking(int fd) OVERRIDE { 4935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // ProcessSingleton only watches for accept (read) events. 4945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED(); 4955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // MessageLoop::DestructionObserver 4982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void WillDestroyCurrentMessageLoop() OVERRIDE { 4995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fd_watcher_.StopWatchingFileDescriptor(); 5005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 5015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 5035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) friend struct BrowserThread::DeleteOnThread<BrowserThread::IO>; 5045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) friend class base::DeleteHelper<ProcessSingleton::LinuxWatcher>; 5055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~LinuxWatcher() { 5075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 5085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) STLDeleteElements(&readers_); 5095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 51090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) base::MessageLoopForIO* ml = base::MessageLoopForIO::current(); 5115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ml->RemoveDestructionObserver(this); 5125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 5135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Removes and deletes the SocketReader. 5155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void RemoveSocketReader(SocketReader* reader); 5165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 51790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) base::MessageLoopForIO::FileDescriptorWatcher fd_watcher_; 5185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // A reference to the UI message loop (i.e., the message loop we were 5205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // constructed on). 52190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) base::MessageLoop* ui_message_loop_; 5225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The ProcessSingleton that owns us. 5245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ProcessSingleton* const parent_; 5255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::set<SocketReader*> readers_; 5275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(LinuxWatcher); 5295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 5305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProcessSingleton::LinuxWatcher::OnFileCanReadWithoutBlocking(int fd) { 5325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 5335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Accepting incoming client. 5345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sockaddr_un from; 5355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socklen_t from_len = sizeof(from); 5365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int connection_socket = HANDLE_EINTR(accept( 5375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fd, reinterpret_cast<sockaddr*>(&from), &from_len)); 5385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (-1 == connection_socket) { 5395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PLOG(ERROR) << "accept() failed"; 5405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 5415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 5425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int rv = SetNonBlocking(connection_socket); 5435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_EQ(0, rv) << "Failed to make non-blocking socket."; 5445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketReader* reader = new SocketReader(this, 5455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ui_message_loop_, 5465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) connection_socket); 5475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) readers_.insert(reader); 5485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProcessSingleton::LinuxWatcher::StartListening(int socket) { 5515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 5525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Watch for client connections on this socket. 55390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) base::MessageLoopForIO* ml = base::MessageLoopForIO::current(); 5545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ml->AddDestructionObserver(this); 55590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ml->WatchFileDescriptor(socket, true, base::MessageLoopForIO::WATCH_READ, 5565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &fd_watcher_, this); 5575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProcessSingleton::LinuxWatcher::HandleMessage( 5605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& current_dir, const std::vector<std::string>& argv, 5615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketReader* reader) { 56290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) DCHECK(ui_message_loop_ == base::MessageLoop::current()); 5635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(reader); 5645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (parent_->notification_callback_.Run(CommandLine(argv), 5662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath(current_dir))) { 5675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Send back "ACK" message to prevent the client process from starting up. 5685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) reader->FinishWithACK(kACKToken, arraysize(kACKToken) - 1); 5695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 5705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(WARNING) << "Not handling interprocess notification as browser" 5715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) " is shutting down"; 5725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Send back "SHUTDOWN" message, so that the client process can start up 5735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // without killing this process. 5745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) reader->FinishWithACK(kShutdownToken, arraysize(kShutdownToken) - 1); 5755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 5765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 5775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProcessSingleton::LinuxWatcher::RemoveSocketReader(SocketReader* reader) { 5805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 5815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(reader); 5825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) readers_.erase(reader); 5835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delete reader; 5845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)/////////////////////////////////////////////////////////////////////////////// 5875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// ProcessSingleton::LinuxWatcher::SocketReader 5885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 5895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProcessSingleton::LinuxWatcher::SocketReader::OnFileCanReadWithoutBlocking( 5915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int fd) { 5925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 5935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_EQ(fd, fd_); 5945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while (bytes_read_ < sizeof(buf_)) { 5955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ssize_t rv = HANDLE_EINTR( 5965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) read(fd, buf_ + bytes_read_, sizeof(buf_) - bytes_read_)); 5975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (rv < 0) { 5985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (errno != EAGAIN && errno != EWOULDBLOCK) { 5995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PLOG(ERROR) << "read() failed"; 6005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CloseSocket(fd); 6015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 6025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 6035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // It would block, so we just return and continue to watch for the next 6045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // opportunity to read. 6055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 6065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else if (!rv) { 6085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // No more data to read. It's time to process the message. 6095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 6105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 6115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bytes_read_ += rv; 6125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Validate the message. The shortest message is kStartToken\0x\0x 6165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const size_t kMinMessageLength = arraysize(kStartToken) + 4; 6175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (bytes_read_ < kMinMessageLength) { 6185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) buf_[bytes_read_] = 0; 6195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "Invalid socket message (wrong length):" << buf_; 6205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CleanupAndDeleteSelf(); 6215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 6225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string str(buf_, bytes_read_); 6255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<std::string> tokens; 6265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::SplitString(str, kTokenDelimiter, &tokens); 6275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (tokens.size() < 3 || tokens[0] != kStartToken) { 6295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "Wrong message format: " << str; 6305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CleanupAndDeleteSelf(); 6315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 6325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Stop the expiration timer to prevent this SocketReader object from being 6355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // terminated unexpectly. 6365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timer_.Stop(); 6375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string current_dir = tokens[1]; 6395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Remove the first two tokens. The remaining tokens should be the command 6405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // line argv array. 6415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tokens.erase(tokens.begin()); 6425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tokens.erase(tokens.begin()); 6435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Return to the UI thread to handle opening a new browser tab. 6455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ui_message_loop_->PostTask(FROM_HERE, base::Bind( 6465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &ProcessSingleton::LinuxWatcher::HandleMessage, 6475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parent_, 6485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) current_dir, 6495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tokens, 6505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) this)); 6515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) fd_reader_.StopWatchingFileDescriptor(); 6525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // LinuxWatcher::HandleMessage() is in charge of destroying this SocketReader 6545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // object by invoking SocketReader::FinishWithACK(). 6555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProcessSingleton::LinuxWatcher::SocketReader::FinishWithACK( 6585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const char *message, size_t length) { 6595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (message && length) { 6605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Not necessary to care about the return value. 6615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) WriteToSocket(fd_, message, length); 6625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (shutdown(fd_, SHUT_WR) < 0) 6655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PLOG(ERROR) << "shutdown() failed"; 6665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BrowserThread::PostTask( 6685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BrowserThread::IO, 6695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FROM_HERE, 6705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&ProcessSingleton::LinuxWatcher::RemoveSocketReader, 6715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parent_, 6725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) this)); 6735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We will be deleted once the posted RemoveSocketReader task runs. 6745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)/////////////////////////////////////////////////////////////////////////////// 6775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// ProcessSingleton 6785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 679c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)ProcessSingleton::ProcessSingleton( 680c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::FilePath& user_data_dir, 681c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const NotificationCallback& notification_callback) 682c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) : notification_callback_(notification_callback), 6835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) current_pid_(base::GetCurrentProcId()), 684c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) watcher_(new LinuxWatcher(this)) { 6855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_path_ = user_data_dir.Append(chrome::kSingletonSocketFilename); 6865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) lock_path_ = user_data_dir.Append(chrome::kSingletonLockFilename); 6875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cookie_path_ = user_data_dir.Append(chrome::kSingletonCookieFilename); 6885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kill_callback_ = base::Bind(&ProcessSingleton::KillProcess, 6905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(this)); 6915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ProcessSingleton::~ProcessSingleton() { 6945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ProcessSingleton::NotifyResult ProcessSingleton::NotifyOtherProcess() { 6975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NotifyOtherProcessWithTimeout(*CommandLine::ForCurrentProcess(), 6985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kTimeoutInSeconds, 6995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) true); 7005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 7015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ProcessSingleton::NotifyResult ProcessSingleton::NotifyOtherProcessWithTimeout( 7035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CommandLine& cmd_line, 7045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int timeout_seconds, 7055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool kill_unresponsive) { 7065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK_GE(timeout_seconds, 0); 7075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ScopedSocket socket; 7095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int retries = 0; retries <= timeout_seconds; ++retries) { 7105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Try to connect to the socket. 7115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (ConnectSocket(&socket, socket_path_, cookie_path_)) 7125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 7135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If we're in a race with another process, they may be in Create() and have 7155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // created the lock but not attached to the socket. So we check if the 7165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // process with the pid from the lockfile is currently running and is a 7175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // chrome browser. If so, we loop and try again for |timeout_seconds|. 7185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string hostname; 7205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int pid; 7215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!ParseLockPath(lock_path_, &hostname, &pid)) { 7225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // No lockfile exists. 7235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return PROCESS_NONE; 7245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 7255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (hostname.empty()) { 7275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Invalid lockfile. 7285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UnlinkPath(lock_path_); 7295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return PROCESS_NONE; 7305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 7315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 73258537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) if (hostname != net::GetHostName() && !IsChromeProcess(pid)) { 73358537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) // Locked by process on another host. If the user selected to unlock 73458537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) // the profile, try to continue; otherwise quit. 73558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) if (DisplayProfileInUseError(lock_path_, hostname, pid)) { 73658537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) UnlinkPath(lock_path_); 73758537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) return PROCESS_NONE; 73858537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) } 7395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return PROFILE_IN_USE; 7405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 7415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!IsChromeProcess(pid)) { 7435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Orphaned lockfile (no process with pid, or non-chrome process.) 7445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UnlinkPath(lock_path_); 7455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return PROCESS_NONE; 7465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 7475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (IsSameChromeInstance(pid)) { 7495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Orphaned lockfile (pid is part of same chrome instance we are, even 7505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // though we haven't tried to create a lockfile yet). 7515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UnlinkPath(lock_path_); 7525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return PROCESS_NONE; 7535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 7545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (retries == timeout_seconds) { 7565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Retries failed. Kill the unresponsive chrome process and continue. 7575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!kill_unresponsive || !KillProcessByLockPath()) 7585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return PROFILE_IN_USE; 7595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return PROCESS_NONE; 7605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 7615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::PlatformThread::Sleep(base::TimeDelta::FromSeconds(1)); 7635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 7645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timeval timeout = {timeout_seconds, 0}; 7665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) setsockopt(socket.fd(), SOL_SOCKET, SO_SNDTIMEO, &timeout, sizeof(timeout)); 7675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Found another process, prepare our command line 7695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // format is "START\0<current dir>\0<argv[0]>\0...\0<argv[n]>". 7705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string to_send(kStartToken); 7715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) to_send.push_back(kTokenDelimiter); 7725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath current_dir; 7745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!PathService::Get(base::DIR_CURRENT, ¤t_dir)) 7755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return PROCESS_NONE; 7765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) to_send.append(current_dir.value()); 7775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::vector<std::string>& argv = cmd_line.argv(); 7795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (std::vector<std::string>::const_iterator it = argv.begin(); 7805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) it != argv.end(); ++it) { 7815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) to_send.push_back(kTokenDelimiter); 7825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) to_send.append(*it); 7835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 7845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Send the message 7865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!WriteToSocket(socket.fd(), to_send.data(), to_send.length())) { 7875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Try to kill the other process, because it might have been dead. 7885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!kill_unresponsive || !KillProcessByLockPath()) 7895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return PROFILE_IN_USE; 7905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return PROCESS_NONE; 7915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 7925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (shutdown(socket.fd(), SHUT_WR) < 0) 7945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PLOG(ERROR) << "shutdown() failed"; 7955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Read ACK message from the other process. It might be blocked for a certain 7975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // timeout, to make sure the other process has enough time to return ACK. 7985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) char buf[kMaxACKMessageLength + 1]; 7995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ssize_t len = 8005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ReadFromSocket(socket.fd(), buf, kMaxACKMessageLength, timeout_seconds); 8015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Failed to read ACK, the other process might have been frozen. 8035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (len <= 0) { 8045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!kill_unresponsive || !KillProcessByLockPath()) 8055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return PROFILE_IN_USE; 8065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return PROCESS_NONE; 8075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 8085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) buf[len] = '\0'; 8105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (strncmp(buf, kShutdownToken, arraysize(kShutdownToken) - 1) == 0) { 8115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The other process is shutting down, it's safe to start a new process. 8125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return PROCESS_NONE; 8135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else if (strncmp(buf, kACKToken, arraysize(kACKToken) - 1) == 0) { 8145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(TOOLKIT_GTK) 8155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Notify the window manager that we've started up; if we do not open a 8165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // window, GTK will not automatically call this for us. 8175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) gdk_notify_startup_complete(); 8185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 8195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Assume the other process is handling the request. 8205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return PROCESS_NOTIFIED; 8215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 8225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED() << "The other process returned unknown message: " << buf; 8245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return PROCESS_NOTIFIED; 8255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 8265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 827c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)ProcessSingleton::NotifyResult ProcessSingleton::NotifyOtherProcessOrCreate() { 8285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return NotifyOtherProcessWithTimeoutOrCreate( 8295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *CommandLine::ForCurrentProcess(), 8305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kTimeoutInSeconds); 8315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 8325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ProcessSingleton::NotifyResult 8345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ProcessSingleton::NotifyOtherProcessWithTimeoutOrCreate( 8355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const CommandLine& command_line, 8365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int timeout_seconds) { 8375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NotifyResult result = NotifyOtherProcessWithTimeout(command_line, 8385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) timeout_seconds, true); 8395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (result != PROCESS_NONE) 8405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return result; 841c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (Create()) 8425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return PROCESS_NONE; 8435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If the Create() failed, try again to notify. (It could be that another 8445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // instance was starting at the same time and managed to grab the lock before 8455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // we did.) 8465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This time, we don't want to kill anything if we aren't successful, since we 8475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // aren't going to try to take over the lock ourselves. 8485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) result = NotifyOtherProcessWithTimeout(command_line, timeout_seconds, false); 8495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (result != PROCESS_NONE) 8505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return result; 8515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return LOCK_ERROR; 8535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 8545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProcessSingleton::OverrideCurrentPidForTesting(base::ProcessId pid) { 8565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) current_pid_ = pid; 8575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 8585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProcessSingleton::OverrideKillCallbackForTesting( 8605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const base::Callback<void(int)>& callback) { 8615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kill_callback_ = callback; 8625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 8635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProcessSingleton::DisablePromptForTesting() { 8655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) g_disable_prompt = true; 8665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 8675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 868c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)bool ProcessSingleton::Create() { 8695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int sock; 8705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sockaddr_un addr; 8715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The symlink lock is pointed to the hostname and process id, so other 8735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // processes can find it out. 8742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath symlink_content(base::StringPrintf( 8755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "%s%c%u", 8765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) net::GetHostName().c_str(), 8775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kLockDelimiter, 8785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) current_pid_)); 8795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Create symbol link before binding the socket, to ensure only one instance 8815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // can have the socket open. 8825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!SymlinkPath(symlink_content, lock_path_)) { 8835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If we failed to create the lock, most likely another instance won the 8845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // startup race. 8855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 8865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 8875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Create the socket file somewhere in /tmp which is usually mounted as a 8895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // normal filesystem. Some network filesystems (notably AFS) are screwy and 8905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // do not support Unix domain sockets. 8915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!socket_dir_.CreateUniqueTempDir()) { 8925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "Failed to create socket directory."; 8935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 8945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 8955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Setup the socket symlink and the two cookies. 8962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath socket_target_path = 8975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_dir_.path().Append(chrome::kSingletonSocketFilename); 8982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath cookie(GenerateCookie()); 8992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath remote_cookie_path = 9005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_dir_.path().Append(chrome::kSingletonCookieFilename); 9015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UnlinkPath(socket_path_); 9025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UnlinkPath(cookie_path_); 9035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!SymlinkPath(socket_target_path, socket_path_) || 9045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) !SymlinkPath(cookie, cookie_path_) || 9055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) !SymlinkPath(cookie, remote_cookie_path)) { 9065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We've already locked things, so we can't have lost the startup race, 9075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // but something doesn't like us. 9085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "Failed to create symlinks."; 9095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!socket_dir_.Delete()) 9105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "Encountered a problem when deleting socket directory."; 9115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 9125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 9135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetupSocket(socket_target_path.value(), &sock, &addr); 9155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (bind(sock, reinterpret_cast<sockaddr*>(&addr), sizeof(addr)) < 0) { 9175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PLOG(ERROR) << "Failed to bind() " << socket_target_path.value(); 9185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CloseSocket(sock); 9195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 9205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 9215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (listen(sock, 5) < 0) 9235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED() << "listen failed: " << safe_strerror(errno); 9245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(BrowserThread::IsMessageLoopValid(BrowserThread::IO)); 9265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BrowserThread::PostTask( 9275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BrowserThread::IO, 9285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) FROM_HERE, 9295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&ProcessSingleton::LinuxWatcher::StartListening, 9305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) watcher_.get(), 9315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sock)); 9325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 9345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 9355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProcessSingleton::Cleanup() { 9375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UnlinkPath(socket_path_); 9385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UnlinkPath(cookie_path_); 9395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UnlinkPath(lock_path_); 9405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 9415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool ProcessSingleton::IsSameChromeInstance(pid_t pid) { 9435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pid_t cur_pid = current_pid_; 9445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) while (pid != cur_pid) { 9455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pid = base::GetParentProcessId(pid); 9465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (pid < 0) 9475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 9485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!IsChromeProcess(pid)) 9495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 9505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 9515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 9525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 9535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool ProcessSingleton::KillProcessByLockPath() { 9555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string hostname; 9565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int pid; 9575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ParseLockPath(lock_path_, &hostname, &pid); 9585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!hostname.empty() && hostname != net::GetHostName()) { 96058537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) return DisplayProfileInUseError(lock_path_, hostname, pid); 9615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 9625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UnlinkPath(lock_path_); 9635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (IsSameChromeInstance(pid)) 9655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 9665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (pid > 0) { 9685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kill_callback_.Run(pid); 9695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 9705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 9715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "Failed to extract pid from path: " << lock_path_.value(); 9735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return true; 9745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 9755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProcessSingleton::KillProcess(int pid) { 9775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(james.su@gmail.com): Is SIGKILL ok? 9785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int rv = kill(static_cast<base::ProcessHandle>(pid), SIGKILL); 9795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // ESRCH = No Such Process (can happen if the other process is already in 9805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // progress of shutting down and finishes before we try to kill it). 9815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(rv == 0 || errno == ESRCH) << "Error killing process: " 9825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) << safe_strerror(errno); 9835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 984