Line data Source code
1 : /******************************************************************************
2 : *
3 : * Project: GDAL Core
4 : * Purpose: Implementation of GDALOpenInfo class.
5 : * Author: Frank Warmerdam, warmerdam@pobox.com
6 : *
7 : **********************************************************************
8 : * Copyright (c) 2002, Frank Warmerdam
9 : * Copyright (c) 2008-2012, Even Rouault <even dot rouault at spatialys.com>
10 : *
11 : * SPDX-License-Identifier: MIT
12 : ****************************************************************************/
13 :
14 : #include "gdal_priv.h" // Must be included first for mingw VSIStatBufL.
15 : #include "cpl_port.h"
16 :
17 : #include <cstdlib>
18 : #include <cstring>
19 : #ifdef HAVE_UNISTD_H
20 : #include <unistd.h>
21 : #endif
22 :
23 : #include <algorithm>
24 : #include <map>
25 : #include <mutex>
26 : #include <vector>
27 :
28 : #include "cpl_config.h"
29 : #include "cpl_conv.h"
30 : #include "cpl_error.h"
31 : #include "cpl_string.h"
32 : #include "cpl_vsi.h"
33 : #include "gdal.h"
34 :
35 : // Keep in sync prototype of those 2 functions between gdalopeninfo.cpp,
36 : // ogrsqlitedatasource.cpp and ogrgeopackagedatasource.cpp
37 : void GDALOpenInfoDeclareFileNotToOpen(const char *pszFilename,
38 : const GByte *pabyHeader,
39 : int nHeaderBytes);
40 : void GDALOpenInfoUnDeclareFileNotToOpen(const char *pszFilename);
41 :
42 : /************************************************************************/
43 :
44 : /* This whole section helps for SQLite/GPKG, especially with write-ahead
45 : * log enabled. The issue is that sqlite3 relies on POSIX advisory locks to
46 : * properly work and decide when to create/delete the wal related files.
47 : * One issue with POSIX advisory locks is that if within the same process
48 : * you do
49 : * f1 = open('somefile')
50 : * set locks on f1
51 : * f2 = open('somefile')
52 : * close(f2)
53 : * The close(f2) will cancel the locks set on f1. The work on f1 is done by
54 : * libsqlite3 whereas the work on f2 is done by GDALOpenInfo.
55 : * So as soon as sqlite3 has opened a file we should make sure not to re-open
56 : * it (actually close it) ourselves.
57 : */
58 :
59 : namespace
60 : {
61 : struct FileNotToOpen
62 : {
63 : CPLString osFilename{};
64 : int nRefCount{};
65 : GByte *pabyHeader{nullptr};
66 : int nHeaderBytes{0};
67 : };
68 : } // namespace
69 :
70 : static std::mutex sFNTOMutex;
71 : static std::map<CPLString, FileNotToOpen> *pMapFNTO = nullptr;
72 :
73 1717 : void GDALOpenInfoDeclareFileNotToOpen(const char *pszFilename,
74 : const GByte *pabyHeader, int nHeaderBytes)
75 : {
76 3434 : std::lock_guard<std::mutex> oLock(sFNTOMutex);
77 1717 : if (pMapFNTO == nullptr)
78 1408 : pMapFNTO = new std::map<CPLString, FileNotToOpen>();
79 1717 : auto oIter = pMapFNTO->find(pszFilename);
80 1717 : if (oIter != pMapFNTO->end())
81 : {
82 120 : oIter->second.nRefCount++;
83 : }
84 : else
85 : {
86 1597 : FileNotToOpen fnto;
87 1597 : fnto.osFilename = pszFilename;
88 1597 : fnto.nRefCount = 1;
89 1597 : fnto.pabyHeader = static_cast<GByte *>(CPLMalloc(nHeaderBytes + 1));
90 1597 : memcpy(fnto.pabyHeader, pabyHeader, nHeaderBytes);
91 1597 : fnto.pabyHeader[nHeaderBytes] = 0;
92 1597 : fnto.nHeaderBytes = nHeaderBytes;
93 1597 : (*pMapFNTO)[pszFilename] = std::move(fnto);
94 : }
95 1717 : }
96 :
97 1716 : void GDALOpenInfoUnDeclareFileNotToOpen(const char *pszFilename)
98 : {
99 3432 : std::lock_guard<std::mutex> oLock(sFNTOMutex);
100 1716 : CPLAssert(pMapFNTO);
101 1716 : auto oIter = pMapFNTO->find(pszFilename);
102 1716 : CPLAssert(oIter != pMapFNTO->end());
103 1716 : oIter->second.nRefCount--;
104 1716 : if (oIter->second.nRefCount == 0)
105 : {
106 1596 : CPLFree(oIter->second.pabyHeader);
107 1596 : pMapFNTO->erase(oIter);
108 : }
109 1716 : if (pMapFNTO->empty())
110 : {
111 1407 : delete pMapFNTO;
112 1407 : pMapFNTO = nullptr;
113 : }
114 1716 : }
115 :
116 98443 : static GByte *GDALOpenInfoGetFileNotToOpen(const char *pszFilename,
117 : int *pnHeaderBytes)
118 : {
119 196886 : std::lock_guard<std::mutex> oLock(sFNTOMutex);
120 98443 : *pnHeaderBytes = 0;
121 98443 : if (pMapFNTO == nullptr)
122 : {
123 92456 : return nullptr;
124 : }
125 5987 : auto oIter = pMapFNTO->find(pszFilename);
126 5987 : if (oIter == pMapFNTO->end())
127 : {
128 4548 : return nullptr;
129 : }
130 1439 : *pnHeaderBytes = oIter->second.nHeaderBytes;
131 1439 : GByte *pabyHeader = static_cast<GByte *>(CPLMalloc(*pnHeaderBytes + 1));
132 1439 : memcpy(pabyHeader, oIter->second.pabyHeader, *pnHeaderBytes);
133 1439 : pabyHeader[*pnHeaderBytes] = 0;
134 1439 : return pabyHeader;
135 : }
136 :
137 : /************************************************************************/
138 : /* ==================================================================== */
139 : /* GDALOpenInfo */
140 : /* ==================================================================== */
141 : /************************************************************************/
142 :
143 : /************************************************************************/
144 : /* GDALOpenInfo() */
145 : /************************************************************************/
146 :
147 : /** Constructor/
148 : * @param pszFilenameIn filename
149 : * @param nOpenFlagsIn open flags
150 : * @param papszSiblingsIn list of sibling files, or NULL.
151 : */
152 99291 : GDALOpenInfo::GDALOpenInfo(const char *pszFilenameIn, int nOpenFlagsIn,
153 99291 : const char *const *papszSiblingsIn)
154 198582 : : pszFilename(CPLStrdup(pszFilenameIn)),
155 : osExtension(CPLGetExtensionSafe(pszFilenameIn)),
156 99289 : eAccess(nOpenFlagsIn & GDAL_OF_UPDATE ? GA_Update : GA_ReadOnly),
157 99291 : nOpenFlags(nOpenFlagsIn)
158 : {
159 99289 : if (STARTS_WITH(pszFilename, "MVT:/vsi"))
160 850 : return;
161 :
162 : /* -------------------------------------------------------------------- */
163 : /* Ensure that C: is treated as C:\ so we can stat it on */
164 : /* Windows. Similar to what is done in CPLStat(). */
165 : /* -------------------------------------------------------------------- */
166 : #ifdef _WIN32
167 : if (strlen(pszFilenameIn) == 2 && pszFilenameIn[1] == ':')
168 : {
169 : char szAltPath[10];
170 :
171 : strcpy(szAltPath, pszFilenameIn);
172 : strcat(szAltPath, "\\");
173 : CPLFree(pszFilename);
174 : pszFilename = CPLStrdup(szAltPath);
175 : }
176 : #endif // WIN32
177 :
178 : /* -------------------------------------------------------------------- */
179 : /* Collect information about the file. */
180 : /* -------------------------------------------------------------------- */
181 :
182 : #ifdef HAVE_READLINK
183 98439 : bool bHasRetried = false;
184 :
185 98442 : retry: // TODO(schwehr): Stop using goto.
186 :
187 : #endif // HAVE_READLINK
188 :
189 : #if !(defined(_WIN32) || defined(__linux__) || defined(__ANDROID__) || \
190 : (defined(__MACH__) && defined(__APPLE__)))
191 : /* On BSDs, fread() on a directory returns non zero, so we have to */
192 : /* do a stat() before to check the nature of pszFilename. */
193 : bool bPotentialDirectory = (eAccess == GA_ReadOnly);
194 : #else
195 98442 : bool bPotentialDirectory = false;
196 : #endif
197 :
198 : /* Check if the filename might be a directory of a special virtual file
199 : * system */
200 98442 : if (STARTS_WITH(pszFilename, "/vsizip/") ||
201 98269 : STARTS_WITH(pszFilename, "/vsitar/") ||
202 98250 : STARTS_WITH(pszFilename, "/vsi7z/") ||
203 98250 : STARTS_WITH(pszFilename, "/vsirar/"))
204 : {
205 191 : const char *pszExt = osExtension.c_str();
206 165 : if (EQUAL(pszExt, "zip") || EQUAL(pszExt, "tar") ||
207 163 : EQUAL(pszExt, "gz") || EQUAL(pszExt, "7z") ||
208 162 : EQUAL(pszExt, "rar") ||
209 162 : pszFilename[strlen(pszFilename) - 1] == '}'
210 : #ifdef DEBUG
211 : // For AFL, so that .cur_input is detected as the archive filename.
212 356 : || EQUAL(CPLGetFilename(pszFilename), ".cur_input")
213 : #endif // DEBUG
214 : )
215 : {
216 40 : bPotentialDirectory = true;
217 191 : }
218 : }
219 98251 : else if (STARTS_WITH(pszFilename, "/vsicurl/"))
220 : {
221 43 : bPotentialDirectory = true;
222 : }
223 :
224 98442 : if (bPotentialDirectory)
225 : {
226 83 : int nStatFlags = VSI_STAT_EXISTS_FLAG | VSI_STAT_NATURE_FLAG;
227 83 : if (nOpenFlagsIn & GDAL_OF_VERBOSE_ERROR)
228 51 : nStatFlags |= VSI_STAT_SET_ERROR_FLAG;
229 :
230 : // For those special files, opening them with VSIFOpenL() might result
231 : // in content, even if they should be considered as directories, so
232 : // use stat.
233 : VSIStatBufL sStat;
234 :
235 83 : if (VSIStatExL(pszFilename, &sStat, nStatFlags) == 0)
236 : {
237 75 : bStatOK = TRUE;
238 75 : if (VSI_ISDIR(sStat.st_mode))
239 22 : bIsDirectory = TRUE;
240 : }
241 : }
242 :
243 98442 : pabyHeader = GDALOpenInfoGetFileNotToOpen(pszFilename, &nHeaderBytes);
244 :
245 98443 : if (!bIsDirectory && pabyHeader == nullptr)
246 : {
247 96982 : fpL = VSIFOpenExL(pszFilename, (eAccess == GA_Update) ? "r+b" : "rb",
248 96982 : (nOpenFlagsIn & GDAL_OF_VERBOSE_ERROR) > 0);
249 : }
250 98442 : if (pabyHeader)
251 : {
252 1439 : bStatOK = TRUE;
253 1439 : nHeaderBytesTried = nHeaderBytes;
254 : }
255 97003 : else if (fpL != nullptr)
256 : {
257 54192 : bStatOK = TRUE;
258 : int nBufSize =
259 54192 : atoi(CPLGetConfigOption("GDAL_INGESTED_BYTES_AT_OPEN", "1024"));
260 54191 : if (nBufSize < 1024)
261 0 : nBufSize = 1024;
262 54191 : else if (nBufSize > 10 * 1024 * 1024)
263 0 : nBufSize = 10 * 1024 * 1024;
264 54191 : pabyHeader = static_cast<GByte *>(CPLCalloc(nBufSize + 1, 1));
265 54190 : nHeaderBytesTried = nBufSize;
266 54188 : nHeaderBytes =
267 54190 : static_cast<int>(VSIFReadL(pabyHeader, 1, nHeaderBytesTried, fpL));
268 54188 : VSIRewindL(fpL);
269 :
270 : /* If we cannot read anything, check if it is not a directory instead */
271 : VSIStatBufL sStat;
272 109400 : if (nHeaderBytes == 0 &&
273 1030 : VSIStatExL(pszFilename, &sStat,
274 55215 : VSI_STAT_EXISTS_FLAG | VSI_STAT_NATURE_FLAG) == 0 &&
275 1030 : VSI_ISDIR(sStat.st_mode))
276 : {
277 914 : CPL_IGNORE_RET_VAL(VSIFCloseL(fpL));
278 914 : fpL = nullptr;
279 914 : CPLFree(pabyHeader);
280 914 : pabyHeader = nullptr;
281 914 : bIsDirectory = TRUE;
282 : }
283 : }
284 42811 : else if (!bStatOK)
285 : {
286 : VSIStatBufL sStat;
287 85572 : if (!bPotentialDirectory &&
288 42781 : VSIStatExL(pszFilename, &sStat,
289 : VSI_STAT_EXISTS_FLAG | VSI_STAT_NATURE_FLAG) == 0)
290 : {
291 1234 : bStatOK = TRUE;
292 1234 : if (VSI_ISDIR(sStat.st_mode))
293 1229 : bIsDirectory = TRUE;
294 : }
295 : #ifdef HAVE_READLINK
296 41557 : else if (!bHasRetried && !STARTS_WITH(pszFilename, "/vsi"))
297 : {
298 : // If someone creates a file with "ln -sf
299 : // /vsicurl/http://download.osgeo.org/gdal/data/gtiff/utm.tif
300 : // my_remote_utm.tif" we will be able to open it by passing
301 : // my_remote_utm.tif. This helps a lot for GDAL based readers that
302 : // only provide file explorers to open datasets.
303 21230 : const int nBufSize = 2048;
304 21230 : std::vector<char> oFilename(nBufSize);
305 21230 : char *szPointerFilename = &oFilename[0];
306 : int nBytes = static_cast<int>(
307 21229 : readlink(pszFilename, szPointerFilename, nBufSize));
308 21230 : if (nBytes != -1)
309 : {
310 2 : szPointerFilename[std::min(nBytes, nBufSize - 1)] = 0;
311 2 : CPLFree(pszFilename);
312 2 : pszFilename = CPLStrdup(szPointerFilename);
313 2 : osExtension = CPLGetExtensionSafe(pszFilename);
314 2 : papszSiblingsIn = nullptr;
315 2 : bHasRetried = true;
316 2 : goto retry;
317 : }
318 : }
319 : #endif // HAVE_READLINK
320 : }
321 :
322 : /* -------------------------------------------------------------------- */
323 : /* Capture sibling list either from passed in values, or by */
324 : /* scanning for them only if requested through GetSiblingFiles(). */
325 : /* -------------------------------------------------------------------- */
326 98432 : if (papszSiblingsIn != nullptr)
327 : {
328 252 : papszSiblingFiles = CSLDuplicate(papszSiblingsIn);
329 252 : bHasGotSiblingFiles = true;
330 : }
331 98180 : else if (bStatOK && !bIsDirectory)
332 : {
333 54460 : papszSiblingFiles = VSISiblingFiles(pszFilename);
334 54470 : if (papszSiblingFiles != nullptr)
335 : {
336 6 : bHasGotSiblingFiles = true;
337 : }
338 : else
339 : {
340 108928 : const char *pszOptionVal = VSIGetPathSpecificOption(
341 54464 : pszFilename, "GDAL_DISABLE_READDIR_ON_OPEN", "NO");
342 54464 : if (EQUAL(pszOptionVal, "EMPTY_DIR"))
343 : {
344 65 : papszSiblingFiles =
345 65 : CSLAddString(nullptr, CPLGetFilename(pszFilename));
346 65 : bHasGotSiblingFiles = true;
347 : }
348 54399 : else if (CPLTestBool(pszOptionVal))
349 : {
350 : /* skip reading the directory */
351 18 : papszSiblingFiles = nullptr;
352 18 : bHasGotSiblingFiles = true;
353 : }
354 : else
355 : {
356 : /* will be lazy loaded */
357 54381 : papszSiblingFiles = nullptr;
358 54381 : bHasGotSiblingFiles = false;
359 : }
360 54470 : }
361 : }
362 : else
363 : {
364 43720 : papszSiblingFiles = nullptr;
365 43720 : bHasGotSiblingFiles = true;
366 : }
367 : }
368 :
369 : /************************************************************************/
370 : /* ~GDALOpenInfo() */
371 : /************************************************************************/
372 :
373 99197 : GDALOpenInfo::~GDALOpenInfo()
374 :
375 : {
376 99268 : VSIFree(pabyHeader);
377 99189 : CPLFree(pszFilename);
378 :
379 99246 : if (fpL != nullptr)
380 12385 : CPL_IGNORE_RET_VAL(VSIFCloseL(fpL));
381 99246 : CSLDestroy(papszSiblingFiles);
382 99160 : }
383 :
384 : /************************************************************************/
385 : /* GetSiblingFiles() */
386 : /************************************************************************/
387 :
388 : /** Return sibling files.
389 : *
390 : * If the list of sibling files has not already been established, it will be,
391 : * unless the GDAL_DISABLE_READDIR_ON_OPEN configuration option has been set to
392 : * YES or EMPTY_DIR when this instance was constructed.
393 : *
394 : * @return sibling files. Ownership belongs to "this".
395 : */
396 39663 : char **GDALOpenInfo::GetSiblingFiles()
397 : {
398 39663 : if (bHasGotSiblingFiles)
399 22737 : return papszSiblingFiles;
400 16926 : bHasGotSiblingFiles = true;
401 :
402 16926 : papszSiblingFiles = VSISiblingFiles(pszFilename);
403 16926 : if (papszSiblingFiles != nullptr)
404 : {
405 0 : return papszSiblingFiles;
406 : }
407 :
408 16926 : const CPLString osDir = CPLGetDirnameSafe(pszFilename);
409 33851 : const int nMaxFiles = atoi(VSIGetPathSpecificOption(
410 16925 : pszFilename, "GDAL_READDIR_LIMIT_ON_OPEN", "1000"));
411 16926 : papszSiblingFiles = VSIReadDirEx(osDir, nMaxFiles);
412 16926 : if (nMaxFiles > 0 && CSLCount(papszSiblingFiles) > nMaxFiles)
413 : {
414 1 : CPLDebug("GDAL", "GDAL_READDIR_LIMIT_ON_OPEN reached on %s",
415 : osDir.c_str());
416 1 : CSLDestroy(papszSiblingFiles);
417 1 : papszSiblingFiles = nullptr;
418 : }
419 :
420 16926 : return papszSiblingFiles;
421 : }
422 :
423 : /************************************************************************/
424 : /* StealSiblingFiles() */
425 : /* */
426 : /* Same as GetSiblingFiles() except that the list is stealed */
427 : /* (ie ownership transferred to the caller) and the associated */
428 : /* member variable is set to NULL. */
429 : /************************************************************************/
430 :
431 : /** Return sibling files and steal reference
432 : * @return sibling files. Ownership below to the caller (must be freed with
433 : * CSLDestroy)
434 : */
435 8466 : char **GDALOpenInfo::StealSiblingFiles()
436 : {
437 8466 : char **papszRet = GetSiblingFiles();
438 8466 : papszSiblingFiles = nullptr;
439 8466 : return papszRet;
440 : }
441 :
442 : /************************************************************************/
443 : /* AreSiblingFilesLoaded() */
444 : /************************************************************************/
445 :
446 : /** Return whether sibling files have been loaded.
447 : * @return true or false.
448 : */
449 49536 : bool GDALOpenInfo::AreSiblingFilesLoaded() const
450 : {
451 49536 : return bHasGotSiblingFiles;
452 : }
453 :
454 : /************************************************************************/
455 : /* TryToIngest() */
456 : /************************************************************************/
457 :
458 : /** Ingest bytes from the file.
459 : * @param nBytes number of bytes to ingest.
460 : * @return TRUE if successful
461 : */
462 22756 : int GDALOpenInfo::TryToIngest(int nBytes)
463 : {
464 22756 : if (fpL == nullptr)
465 54 : return FALSE;
466 22702 : if (nHeaderBytes < nHeaderBytesTried)
467 13281 : return TRUE;
468 9421 : pabyHeader = static_cast<GByte *>(CPLRealloc(pabyHeader, nBytes + 1));
469 9421 : memset(pabyHeader, 0, nBytes + 1);
470 9421 : VSIRewindL(fpL);
471 9421 : nHeaderBytesTried = nBytes;
472 9421 : nHeaderBytes = static_cast<int>(VSIFReadL(pabyHeader, 1, nBytes, fpL));
473 9421 : VSIRewindL(fpL);
474 :
475 9421 : return TRUE;
476 : }
477 :
478 : /************************************************************************/
479 : /* IsSingleAllowedDriver() */
480 : /************************************************************************/
481 :
482 : /** Returns true if the driver name is the single in the list of allowed
483 : * drivers.
484 : *
485 : * @param pszDriverName Driver name to test.
486 : * @return true if the driver name is the single in the list of allowed
487 : * drivers.
488 : * @since GDAL 3.10
489 : */
490 587915 : bool GDALOpenInfo::IsSingleAllowedDriver(const char *pszDriverName) const
491 : {
492 2849 : return papszAllowedDrivers && papszAllowedDrivers[0] &&
493 591460 : !papszAllowedDrivers[1] &&
494 588611 : EQUAL(papszAllowedDrivers[0], pszDriverName);
495 : }
|
