/src/FreeRDP/winpr/libwinpr/path/include/PathAllocCombine.h
Line  | Count  | Source (jump to first uncovered line)  | 
1  |  |  | 
2  |  | /*  | 
3  |  | #define DEFINE_UNICODE    FALSE  | 
4  |  | #define CUR_PATH_SEPARATOR_CHR  '\\'  | 
5  |  | #define CUR_PATH_SEPARATOR_STR  "\\"  | 
6  |  | #define PATH_ALLOC_COMBINE  PathAllocCombineA  | 
7  |  | */  | 
8  |  |  | 
9  |  | /**  | 
10  |  |  * FIXME: These implementations of the PathAllocCombine functions have  | 
11  |  |  * several issues:  | 
12  |  |  * - pszPathIn or pszMore may be NULL (but not both)  | 
13  |  |  * - no check if pszMore is fully qualified (if so, it must be directly  | 
14  |  |  *   copied to the output buffer without being combined with pszPathIn.  | 
15  |  |  * - if pszMore begins with a _single_ backslash it must be combined with  | 
16  |  |  *   only the root of the path pointed to by pszPathIn and there's no code  | 
17  |  |  *   to extract the root of pszPathIn.  | 
18  |  |  * - the function will crash with some short string lengths of the parameters  | 
19  |  |  */  | 
20  |  |  | 
21  |  | #if DEFINE_UNICODE  | 
22  |  |  | 
23  |  | HRESULT PATH_ALLOC_COMBINE(PCWSTR pszPathIn, PCWSTR pszMore, unsigned long dwFlags,  | 
24  |  |                            PWSTR* ppszPathOut)  | 
25  | 0  | { | 
26  | 0  |   PWSTR pszPathOut;  | 
27  | 0  |   BOOL backslashIn;  | 
28  | 0  |   BOOL backslashMore;  | 
29  | 0  |   size_t pszMoreLength;  | 
30  | 0  |   size_t pszPathInLength;  | 
31  | 0  |   size_t pszPathOutLength;  | 
32  | 0  |   WLog_WARN(TAG, "has known bugs and needs fixing.");  | 
33  |  | 
  | 
34  | 0  |   if (!ppszPathOut)  | 
35  | 0  |     return E_INVALIDARG;  | 
36  |  |  | 
37  | 0  |   if (!pszPathIn && !pszMore)  | 
38  | 0  |     return E_INVALIDARG;  | 
39  |  |  | 
40  | 0  |   if (!pszMore)  | 
41  | 0  |     return E_FAIL; /* valid but not implemented, see top comment */  | 
42  |  |  | 
43  | 0  |   if (!pszPathIn)  | 
44  | 0  |     return E_FAIL; /* valid but not implemented, see top comment */  | 
45  |  |  | 
46  | 0  |   pszPathInLength = _wcslen(pszPathIn);  | 
47  | 0  |   pszMoreLength = _wcslen(pszMore);  | 
48  |  |  | 
49  |  |   /* prevent segfaults - the complete implementation below is buggy */  | 
50  | 0  |   if (pszPathInLength < 3)  | 
51  | 0  |     return E_FAIL;  | 
52  |  |  | 
53  | 0  |   backslashIn = (pszPathIn[pszPathInLength - 1] == CUR_PATH_SEPARATOR_CHR) ? TRUE : FALSE;  | 
54  | 0  |   backslashMore = (pszMore[0] == CUR_PATH_SEPARATOR_CHR) ? TRUE : FALSE;  | 
55  |  | 
  | 
56  | 0  |   if (backslashMore)  | 
57  | 0  |   { | 
58  | 0  |     if ((pszPathIn[1] == ':') && (pszPathIn[2] == CUR_PATH_SEPARATOR_CHR))  | 
59  | 0  |     { | 
60  | 0  |       const WCHAR colon[] = { ':', '\0' }; | 
61  | 0  |       size_t sizeOfBuffer;  | 
62  | 0  |       pszPathOutLength = sizeof(WCHAR) + pszMoreLength;  | 
63  | 0  |       sizeOfBuffer = (pszPathOutLength + 1) * sizeof(WCHAR);  | 
64  | 0  |       pszPathOut = (PWSTR)calloc(sizeOfBuffer, sizeof(WCHAR));  | 
65  |  | 
  | 
66  | 0  |       if (!pszPathOut)  | 
67  | 0  |         return E_OUTOFMEMORY;  | 
68  |  |  | 
69  | 0  |       _wcsncat(pszPathOut, &pszPathIn[0], 1);  | 
70  | 0  |       _wcsncat(pszPathOut, colon, ARRAYSIZE(colon));  | 
71  | 0  |       _wcsncat(pszPathOut, pszMore, pszMoreLength);  | 
72  | 0  |       *ppszPathOut = pszPathOut;  | 
73  | 0  |       return S_OK;  | 
74  | 0  |     }  | 
75  | 0  |   }  | 
76  | 0  |   else  | 
77  | 0  |   { | 
78  | 0  |     const WCHAR sep[] = CUR_PATH_SEPARATOR_STR;  | 
79  | 0  |     size_t sizeOfBuffer;  | 
80  | 0  |     pszPathOutLength = pszPathInLength + pszMoreLength;  | 
81  | 0  |     sizeOfBuffer = (pszPathOutLength + 1) * 2;  | 
82  | 0  |     pszPathOut = (PWSTR)calloc(sizeOfBuffer, 2);  | 
83  |  | 
  | 
84  | 0  |     if (!pszPathOut)  | 
85  | 0  |       return E_OUTOFMEMORY;  | 
86  |  |  | 
87  | 0  |     _wcsncat(pszPathOut, pszPathIn, pszPathInLength);  | 
88  | 0  |     if (!backslashIn)  | 
89  | 0  |       _wcsncat(pszPathOut, sep, ARRAYSIZE(sep));  | 
90  | 0  |     _wcsncat(pszPathOut, pszMore, pszMoreLength);  | 
91  |  | 
  | 
92  | 0  |     *ppszPathOut = pszPathOut;  | 
93  | 0  |     return S_OK;  | 
94  | 0  |   }  | 
95  |  |  | 
96  | 0  |   return E_FAIL;  | 
97  | 0  | } Unexecuted instantiation: PathAllocCombineW Unexecuted instantiation: UnixPathAllocCombineW Unexecuted instantiation: NativePathAllocCombineW  | 
98  |  |  | 
99  |  | #else  | 
100  |  |  | 
101  |  | HRESULT PATH_ALLOC_COMBINE(PCSTR pszPathIn, PCSTR pszMore, unsigned long dwFlags, PSTR* ppszPathOut)  | 
102  | 0  | { | 
103  | 0  |   PSTR pszPathOut;  | 
104  | 0  |   BOOL backslashIn;  | 
105  | 0  |   BOOL backslashMore;  | 
106  | 0  |   int pszMoreLength;  | 
107  | 0  |   int pszPathInLength;  | 
108  | 0  |   int pszPathOutLength;  | 
109  | 0  |   WLog_WARN(TAG, "has known bugs and needs fixing.");  | 
110  |  | 
  | 
111  | 0  |   if (!ppszPathOut)  | 
112  | 0  |     return E_INVALIDARG;  | 
113  |  |  | 
114  | 0  |   if (!pszPathIn && !pszMore)  | 
115  | 0  |     return E_INVALIDARG;  | 
116  |  |  | 
117  | 0  |   if (!pszMore)  | 
118  | 0  |     return E_FAIL; /* valid but not implemented, see top comment */  | 
119  |  |  | 
120  | 0  |   if (!pszPathIn)  | 
121  | 0  |     return E_FAIL; /* valid but not implemented, see top comment */  | 
122  |  |  | 
123  | 0  |   pszPathInLength = strlen(pszPathIn);  | 
124  | 0  |   pszMoreLength = strlen(pszMore);  | 
125  |  |  | 
126  |  |   /* prevent segfaults - the complete implementation below is buggy */  | 
127  | 0  |   if (pszPathInLength < 3)  | 
128  | 0  |     return E_FAIL;  | 
129  |  |  | 
130  | 0  |   backslashIn = (pszPathIn[pszPathInLength - 1] == CUR_PATH_SEPARATOR_CHR) ? TRUE : FALSE;  | 
131  | 0  |   backslashMore = (pszMore[0] == CUR_PATH_SEPARATOR_CHR) ? TRUE : FALSE;  | 
132  |  | 
  | 
133  | 0  |   if (backslashMore)  | 
134  | 0  |   { | 
135  | 0  |     if ((pszPathIn[1] == ':') && (pszPathIn[2] == CUR_PATH_SEPARATOR_CHR))  | 
136  | 0  |     { | 
137  | 0  |       size_t sizeOfBuffer;  | 
138  | 0  |       pszPathOutLength = 2 + pszMoreLength;  | 
139  | 0  |       sizeOfBuffer = (pszPathOutLength + 1) * 2;  | 
140  | 0  |       pszPathOut = (PSTR)calloc(sizeOfBuffer, 2);  | 
141  |  | 
  | 
142  | 0  |       if (!pszPathOut)  | 
143  | 0  |         return E_OUTOFMEMORY;  | 
144  |  |  | 
145  | 0  |       sprintf_s(pszPathOut, sizeOfBuffer, "%c:%s", pszPathIn[0], pszMore);  | 
146  | 0  |       *ppszPathOut = pszPathOut;  | 
147  | 0  |       return S_OK;  | 
148  | 0  |     }  | 
149  | 0  |   }  | 
150  | 0  |   else  | 
151  | 0  |   { | 
152  | 0  |     size_t sizeOfBuffer;  | 
153  | 0  |     pszPathOutLength = pszPathInLength + pszMoreLength;  | 
154  | 0  |     sizeOfBuffer = (pszPathOutLength + 1) * 2;  | 
155  | 0  |     pszPathOut = (PSTR)calloc(sizeOfBuffer, 2);  | 
156  |  | 
  | 
157  | 0  |     if (!pszPathOut)  | 
158  | 0  |       return E_OUTOFMEMORY;  | 
159  |  |  | 
160  | 0  |     if (backslashIn)  | 
161  | 0  |       sprintf_s(pszPathOut, sizeOfBuffer, "%s%s", pszPathIn, pszMore);  | 
162  | 0  |     else  | 
163  | 0  |       sprintf_s(pszPathOut, sizeOfBuffer, "%s" CUR_PATH_SEPARATOR_STR "%s", pszPathIn,  | 
164  | 0  |                 pszMore);  | 
165  |  | 
  | 
166  | 0  |     *ppszPathOut = pszPathOut;  | 
167  | 0  |     return S_OK;  | 
168  | 0  |   }  | 
169  |  |  | 
170  | 0  |   return E_FAIL;  | 
171  | 0  | } Unexecuted instantiation: PathAllocCombineA Unexecuted instantiation: UnixPathAllocCombineA Unexecuted instantiation: NativePathAllocCombineA  | 
172  |  |  | 
173  |  | #endif  | 
174  |  |  | 
175  |  | /*  | 
176  |  | #undef DEFINE_UNICODE  | 
177  |  | #undef CUR_PATH_SEPARATOR_CHR  | 
178  |  | #undef CUR_PATH_SEPARATOR_STR  | 
179  |  | #undef PATH_ALLOC_COMBINE  | 
180  |  | */  |