code.vuplus.com Git - vuplus_webkit/blob - Source/WebKit2/Platform/unix/EnvironmentUtilities.cpp

   1 /*
   2  * Copyright (C) 2011 Apple Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
  14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
  17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  23  * THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #include "config.h"
  27 #include "EnvironmentUtilities.h"
  28
  29 #include <wtf/text/CString.h>
  30
  31 namespace WebKit {
  32
  33 namespace EnvironmentUtilities {
  34
  35 void stripValuesEndingWithString(const char* environmentVariable, const char* searchValue)
  36 {
  37     ASSERT(environmentVariable);
  38     ASSERT(searchValue);
  39
  40     // Grab the current value of the environment variable.
  41     char* environmentValue = getenv(environmentVariable);
  42
  43     if (!environmentValue || environmentValue[0] == '\0')
  44         return;
  45
  46     // Set up the strings we'll be searching for.
  47     size_t searchLength = strlen(searchValue);
  48     if (!searchLength)
  49         return;
  50
  51     Vector<char> searchValueWithColonVector;
  52     searchValueWithColonVector.grow(searchLength + 2);
  53     char* searchValueWithColon = searchValueWithColonVector.data();
  54     size_t searchLengthWithColon = searchLength + 1;
  55
  56     memcpy(searchValueWithColon, searchValue, searchLength);
  57     searchValueWithColon[searchLength] = ':';
  58     searchValueWithColon[searchLengthWithColon] = '\0';
  59
  60     // Loop over environmentValueBuffer, removing any components that match the search value ending with a colon.
  61     char* componentStart = environmentValue;
  62     char* match = strstr(componentStart, searchValueWithColon);
  63     bool foundAnyMatches = match != NULL;
  64     while (match != NULL) {
  65         // Update componentStart to point to the colon immediately preceding the match.
  66         char* nextColon = strstr(componentStart, ":");
  67         while (nextColon && nextColon < match) {
  68             componentStart = nextColon;
  69             nextColon = strstr(componentStart, ":");
  70         }
  71
  72         // Copy over everything right of the match to the current component start, and search from there again.
  73         if (componentStart[0] == ':') {
  74             // If componentStart points to a colon, go ahead and copy the colon over.
  75             strcpy(componentStart, match + searchLength);
  76         } else {
  77             // Otherwise, componentStart still points to the beginning of environmentValueBuffer, so don't copy over the colon.
  78             // The edge case is if the colon is the last character in the string, so "match + searchLengthWithoutColon + 1" is the
  79             // null terminator of the original input, in which case this is still safe.
  80             strcpy(componentStart, match + searchLengthWithColon);
  81         }
  82
  83         match = strstr(componentStart, searchValueWithColon);
  84     }
  85
  86     // Search for the value without a trailing colon, seeing if the original input ends with it.
  87     match = strstr(componentStart, searchValue);
  88     while (match != NULL) {
  89         if (match[searchLength + 1] == '\0')
  90             break;
  91         match = strstr(match + 1, searchValue);
  92     }
  93
  94     // Since the original input ends with the search, strip out the last component.
  95     if (match) {
  96         // Update componentStart to point to the colon immediately preceding the match.
  97         char* nextColon = strstr(componentStart, ":");
  98         while (nextColon && nextColon < match) {
  99             componentStart = nextColon;
 100             nextColon = strstr(componentStart, ":");
 101         }
 102
 103         // Whether componentStart points to the original string or the last colon, putting the null terminator there will get us the desired result.
 104         componentStart[0] = '\0';
 105
 106         foundAnyMatches = true;
 107     }
 108
 109     // If we found no matches, don't change anything.
 110     if (!foundAnyMatches)
 111         return;
 112
 113     // If we have nothing left, just unset the variable
 114     if (environmentValue[0] == '\0') {
 115         unsetenv(environmentVariable);
 116         return;
 117     }
 118
 119     setenv(environmentVariable, environmentValue, 1);
 120 }
 121
 122 } // namespace EnvironmentUtilities
 123
 124 } // namespace WebKit