Prevent loading libffmpegsumo.so to python (autotests) process.

content/test/content_test_suite_base.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "content/public/test/content_test_suite_base.h"
 
+#include <fcntl.h>
+#include <stdio.h>
+
 #include "base/basictypes.h"
 #include "base/compiler_specific.h"
+#include "base/eintr_wrapper.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/test/test_suite.h"
 #include "content/common/url_schemes.h"
 #include "content/public/common/content_client.h"
 #include "content/public/common/content_paths.h"
 #include "media/base/media.h"
 #include "ui/base/ui_base_paths.h"
 #include "ui/compositor/compositor_setup.h"
 
 namespace content {
 
 ContentTestSuiteBase::ContentTestSuiteBase(int argc, char** argv)
     : base::TestSuite(argc, argv) {
 }
 
+static bool IsAutotestProcess() {
+#ifdef OS_CHROMEOS

[scherkus (not reviewing), 2012/07/18 17:40:02]

#if defined(OS_CHROMEOS)

[glotov, 2012/07/19 20:12:40]

Done.

+  int status_fd = open("/proc/self/cmdline", O_RDONLY);
+  if (status_fd == -1)
+    return false;
+  const char kPythonPrefix[] = "/usr/bin/python";

[scherkus (not reviewing), 2012/07/18 17:40:02]

would this catch processes started with /usr/bin/env python?

overall this seems a bit brittle but I'm not a linux guy so I'm not sure what's
the best way to figure out whether we've been invoked via python or not

Is there a stronger signal of some sort (i.e., command line flag) that we can
use as opposed to snooping /proc/self/cmdline? 

[glotov, 2012/07/19 20:12:40]

Done.

+  const size_t kPythonPrefixLength = sizeof(kPythonPrefix) - 1;
+  char cmdline_prefix[kPythonPrefixLength];
+  ssize_t num_read =
+      HANDLE_EINTR(read(status_fd, cmdline_prefix, sizeof(cmdline_prefix)));
+  if (HANDLE_EINTR(close(status_fd)) < 0)
+    return false;
+  if (num_read == sizeof(cmdline_prefix) &&
+      !memcmp(kPythonPrefix, cmdline_prefix, sizeof(cmdline_prefix)))
+    return true;
+#endif
+  return false;
+}
+
 void ContentTestSuiteBase::Initialize() {
   base::TestSuite::Initialize();
-  media::InitializeMediaLibraryForTesting();
+
+  // Initialize media library for unit tests. If we are auto test
+  // (python process), media library will be loaded to chrome directly
+  // so don't load it here.
+  if (!IsAutotestProcess())
+    media::InitializeMediaLibraryForTesting();
 
   scoped_ptr<ContentClient> client_for_init(CreateClientForInitialization());
   SetContentClient(client_for_init.get());
   RegisterContentSchemes(false);
   SetContentClient(NULL);
 
   RegisterPathProvider();
   ui::RegisterPathProvider();
 
   // Mock out the compositor on platforms that use it.
   ui::SetupTestCompositor();
 }
 
 }  // namespace content