Unreviewed, rolling out r241559 and r241566.
[WebKit-https.git] / Source / WebKit / 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     const size_t environmentValueLength = strlen(environmentValue);
47     const size_t environmentValueBufferLength = environmentValueLength + 1;
48
49     // Set up the strings we'll be searching for.
50     size_t searchLength = strlen(searchValue);
51     if (!searchLength)
52         return;
53
54     Vector<char> searchValueWithColonVector;
55     searchValueWithColonVector.grow(searchLength + 2);
56     char* searchValueWithColon = searchValueWithColonVector.data();
57     size_t searchLengthWithColon = searchLength + 1;
58
59     memcpy(searchValueWithColon, searchValue, searchLength);
60     searchValueWithColon[searchLength] = ':';
61     searchValueWithColon[searchLengthWithColon] = '\0';
62     
63     // Loop over environmentValueBuffer, removing any components that match the search value ending with a colon.
64     char* componentStart = environmentValue;
65     char* match = strnstr(componentStart, searchValueWithColon, environmentValueLength - static_cast<size_t>(componentStart - environmentValue));
66     bool foundAnyMatches = match != NULL;
67     while (match != NULL) {
68         // Update componentStart to point to the colon immediately preceding the match.
69         char* nextColon = strnstr(componentStart, ":", environmentValueLength - static_cast<size_t>(componentStart - environmentValue));
70         while (nextColon && nextColon < match) {
71             componentStart = nextColon;
72             nextColon = strnstr(componentStart + 1, ":", environmentValueLength - static_cast<size_t>(componentStart + 1 - environmentValue));
73         }
74
75         RELEASE_ASSERT(componentStart >= environmentValue);
76         size_t environmentValueOffset = static_cast<size_t>(componentStart - environmentValue);
77         RELEASE_ASSERT(environmentValueOffset < environmentValueBufferLength);
78
79         // Copy over everything right of the match to the current component start, and search from there again.
80         if (componentStart[0] == ':') {
81             // If componentStart points to a colon, copy the colon over.
82             strlcpy(componentStart, match + searchLength, environmentValueBufferLength - environmentValueOffset);
83         } else {
84             // Otherwise, componentStart still points to the beginning of environmentValueBuffer, so don't copy over the colon.
85             // The edge case is if the colon is the last character in the string, so "match + searchLengthWithoutColon + 1" is the
86             // null terminator of the original input, in which case this is still safe.
87             strlcpy(componentStart, match + searchLengthWithColon, environmentValueBufferLength - environmentValueOffset);
88         }
89         
90         match = strnstr(componentStart, searchValueWithColon, environmentValueLength - static_cast<size_t>(componentStart - environmentValue));
91     }
92     
93     // Search for the value without a trailing colon, seeing if the original input ends with it.
94     match = strnstr(componentStart, searchValue, environmentValueLength - static_cast<size_t>(componentStart - environmentValue));
95     while (match != NULL) {
96         if (match[searchLength] == '\0')
97             break;
98         match = strnstr(match + 1, searchValue, environmentValueLength - static_cast<size_t>(match + 1 - environmentValue));
99     }
100     
101     // Since the original input ends with the search, strip out the last component.
102     if (match) {
103         // Update componentStart to point to the colon immediately preceding the match.
104         char* nextColon = strnstr(componentStart, ":", environmentValueLength - static_cast<size_t>(componentStart - environmentValue));
105         while (nextColon && nextColon < match) {
106             componentStart = nextColon;
107             nextColon = strnstr(componentStart + 1, ":", environmentValueLength - static_cast<size_t>(componentStart + 1 - environmentValue));
108         }
109         
110         // Whether componentStart points to the original string or the last colon, putting the null terminator there will get us the desired result.
111         componentStart[0] = '\0';
112
113         foundAnyMatches = true;
114     }
115
116     // If we found no matches, don't change anything.
117     if (!foundAnyMatches)
118         return;
119
120     // If we have nothing left, just unset the variable
121     if (environmentValue[0] == '\0') {
122         unsetenv(environmentVariable);
123         return;
124     }
125     
126     setenv(environmentVariable, environmentValue, 1);
127 }
128
129 } // namespace EnvironmentUtilities
130
131 } // namespace WebKit