1 /****************************************************************************
2 **
3 ** Copyright (C) 2016 The Qt Company Ltd.
4 ** Contact: https://www.qt.io/licensing/
5 **
6 ** This file is part of the QtCore module of the Qt Toolkit.
7 **
8 ** $QT_BEGIN_LICENSE:LGPL$
9 ** Commercial License Usage
10 ** Licensees holding valid commercial Qt licenses may use this file in
11 ** accordance with the commercial license agreement provided with the
12 ** Software or, alternatively, in accordance with the terms contained in
13 ** a written agreement between you and The Qt Company. For licensing terms
14 ** and conditions see https://www.qt.io/terms-conditions. For further
15 ** information use the contact form at https://www.qt.io/contact-us.
16 **
17 ** GNU Lesser General Public License Usage
18 ** Alternatively, this file may be used under the terms of the GNU Lesser
19 ** General Public License version 3 as published by the Free Software
20 ** Foundation and appearing in the file LICENSE.LGPL3 included in the
21 ** packaging of this file. Please review the following information to
22 ** ensure the GNU Lesser General Public License version 3 requirements
23 ** will be met: https://www.gnu.org/licenses/lgpl-3.0.html.
24 **
25 ** GNU General Public License Usage
26 ** Alternatively, this file may be used under the terms of the GNU
27 ** General Public License version 2.0 or (at your option) the GNU General
28 ** Public license version 3 or any later version approved by the KDE Free
29 ** Qt Foundation. The licenses are as published by the Free Software
30 ** Foundation and appearing in the file LICENSE.GPL2 and LICENSE.GPL3
31 ** included in the packaging of this file. Please review the following
32 ** information to ensure the GNU General Public License requirements will
33 ** be met: https://www.gnu.org/licenses/gpl-2.0.html and
34 ** https://www.gnu.org/licenses/gpl-3.0.html.
35 **
36 ** $QT_END_LICENSE$
37 **
38 ****************************************************************************/
39 
40 #ifndef QFUNCTIONS_FAKE_ENV_P_H
41 #define QFUNCTIONS_FAKE_ENV_P_H
42 
43 //
44 //  W A R N I N G
45 //  -------------
46 //
47 // This file is not part of the Qt API. It exists purely as an
48 // implementation detail. This header file may change from version to
49 // version without notice, or even be removed.
50 //
51 // We mean it.
52 //
53 
54 #include "qbytearray.h"
55 #include "qvector.h"
56 
57 QT_BEGIN_NAMESPACE
58 
59 // Environment ------------------------------------------------------
60 struct Variable {
VariableVariable61     Variable() { }
62 
VariableVariable63     Variable(const QByteArray &name, const QByteArray &value)
64         : name(name), value(value) { }
65 
66     QByteArray name;
67     QByteArray value;
68 };
69 
70 Q_DECLARE_TYPEINFO(Variable, Q_MOVABLE_TYPE);
71 
72 struct NameEquals {
73     typedef bool result_type;
74     const char *name;
NameEqualsNameEquals75     explicit NameEquals(const char *name) noexcept : name(name) {}
operatorNameEquals76     result_type operator()(const Variable &other) const noexcept
77     { return qstrcmp(other.name, name) == 0; }
78 };
79 
80 #ifndef Q_CLANG_QDOC
Q_GLOBAL_STATIC(QVector<Variable>,qt_app_environment)81 Q_GLOBAL_STATIC(QVector<Variable>, qt_app_environment)
82 #endif
83 
84 errno_t qt_fake_getenv_s(size_t *sizeNeeded, char *buffer, size_t bufferSize, const char *varName)
85 {
86     if (!sizeNeeded)
87         return EINVAL;
88 
89     QVector<Variable>::const_iterator end = qt_app_environment->constEnd();
90     QVector<Variable>::const_iterator iterator = std::find_if(qt_app_environment->constBegin(),
91                                                               end,
92                                                               NameEquals(varName));
93     if (iterator == end) {
94         if (buffer)
95             buffer[0] = '\0';
96         return ENOENT;
97     }
98 
99     const int size = iterator->value.size() + 1;
100     if (bufferSize < size_t(size)) {
101         *sizeNeeded = size;
102         return ERANGE;
103     }
104 
105     qstrcpy(buffer, iterator->value.constData());
106     return 0;
107 }
108 
qt_fake__putenv_s(const char * varName,const char * value)109 errno_t qt_fake__putenv_s(const char *varName, const char *value)
110 {
111     QVector<Variable>::iterator end = qt_app_environment->end();
112     QVector<Variable>::iterator iterator = std::find_if(qt_app_environment->begin(),
113                                                         end,
114                                                         NameEquals(varName));
115     if (!value || !*value) {
116         if (iterator != end)
117             qt_app_environment->erase(iterator);
118     } else {
119         if (iterator == end)
120             qt_app_environment->append(Variable(varName, value));
121         else
122             iterator->value = value;
123     }
124 
125     return 0;
126 }
127 
128 QT_END_NAMESPACE
129 
130 #endif // QFUNCTIONS_FAKE_ENV_P_H
131