1 //===-- FileSpec.cpp --------------------------------------------*- C++ -*-===//
2 //
3 // The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9
10
11 #include <dirent.h>
12 #include <fcntl.h>
13 #include <libgen.h>
14 #include <sys/stat.h>
15 #include <set>
16 #include <string.h>
17 #include <fstream>
18
19 #include "lldb/Host/Config.h" // Have to include this before we test the define...
20 #ifdef LLDB_CONFIG_TILDE_RESOLVES_TO_USER
21 #include <pwd.h>
22 #endif
23
24 #include "llvm/ADT/StringRef.h"
25 #include "llvm/Support/Path.h"
26 #include "llvm/Support/Program.h"
27
28 #include "lldb/Host/File.h"
29 #include "lldb/Host/FileSpec.h"
30 #include "lldb/Core/DataBufferHeap.h"
31 #include "lldb/Core/DataBufferMemoryMap.h"
32 #include "lldb/Core/RegularExpression.h"
33 #include "lldb/Core/Stream.h"
34 #include "lldb/Host/Host.h"
35 #include "lldb/Utility/CleanUp.h"
36
37 using namespace lldb;
38 using namespace lldb_private;
39
40 static bool
GetFileStats(const FileSpec * file_spec,struct stat * stats_ptr)41 GetFileStats (const FileSpec *file_spec, struct stat *stats_ptr)
42 {
43 char resolved_path[PATH_MAX];
44 if (file_spec->GetPath (resolved_path, sizeof(resolved_path)))
45 return ::stat (resolved_path, stats_ptr) == 0;
46 return false;
47 }
48
49 #ifdef LLDB_CONFIG_TILDE_RESOLVES_TO_USER
50
51 static const char*
GetCachedGlobTildeSlash()52 GetCachedGlobTildeSlash()
53 {
54 static std::string g_tilde;
55 if (g_tilde.empty())
56 {
57 struct passwd *user_entry;
58 user_entry = getpwuid(geteuid());
59 if (user_entry != NULL)
60 g_tilde = user_entry->pw_dir;
61
62 if (g_tilde.empty())
63 return NULL;
64 }
65 return g_tilde.c_str();
66 }
67
68 #endif // #ifdef LLDB_CONFIG_TILDE_RESOLVES_TO_USER
69
70 // Resolves the username part of a path of the form ~user/other/directories, and
71 // writes the result into dst_path.
72 // Returns 0 if there WAS a ~ in the path but the username couldn't be resolved.
73 // Otherwise returns the number of characters copied into dst_path. If the return
74 // is >= dst_len, then the resolved path is too long...
75 size_t
ResolveUsername(const char * src_path,char * dst_path,size_t dst_len)76 FileSpec::ResolveUsername (const char *src_path, char *dst_path, size_t dst_len)
77 {
78 if (src_path == NULL || src_path[0] == '\0')
79 return 0;
80
81 #ifdef LLDB_CONFIG_TILDE_RESOLVES_TO_USER
82
83 char user_home[PATH_MAX];
84 const char *user_name;
85
86
87 // If there's no ~, then just copy src_path straight to dst_path (they may be the same string...)
88 if (src_path[0] != '~')
89 {
90 size_t len = strlen (src_path);
91 if (len >= dst_len)
92 {
93 ::bcopy (src_path, dst_path, dst_len - 1);
94 dst_path[dst_len] = '\0';
95 }
96 else
97 ::bcopy (src_path, dst_path, len + 1);
98
99 return len;
100 }
101
102 const char *first_slash = ::strchr (src_path, '/');
103 char remainder[PATH_MAX];
104
105 if (first_slash == NULL)
106 {
107 // The whole name is the username (minus the ~):
108 user_name = src_path + 1;
109 remainder[0] = '\0';
110 }
111 else
112 {
113 size_t user_name_len = first_slash - src_path - 1;
114 ::memcpy (user_home, src_path + 1, user_name_len);
115 user_home[user_name_len] = '\0';
116 user_name = user_home;
117
118 ::strcpy (remainder, first_slash);
119 }
120
121 if (user_name == NULL)
122 return 0;
123 // User name of "" means the current user...
124
125 struct passwd *user_entry;
126 const char *home_dir = NULL;
127
128 if (user_name[0] == '\0')
129 {
130 home_dir = GetCachedGlobTildeSlash();
131 }
132 else
133 {
134 user_entry = ::getpwnam (user_name);
135 if (user_entry != NULL)
136 home_dir = user_entry->pw_dir;
137 }
138
139 if (home_dir == NULL)
140 return 0;
141 else
142 return ::snprintf (dst_path, dst_len, "%s%s", home_dir, remainder);
143 #else
144 // Resolving home directories is not supported, just copy the path...
145 return ::snprintf (dst_path, dst_len, "%s", src_path);
146 #endif // #ifdef LLDB_CONFIG_TILDE_RESOLVES_TO_USER
147 }
148
149 size_t
ResolvePartialUsername(const char * partial_name,StringList & matches)150 FileSpec::ResolvePartialUsername (const char *partial_name, StringList &matches)
151 {
152 #ifdef LLDB_CONFIG_TILDE_RESOLVES_TO_USER
153 size_t extant_entries = matches.GetSize();
154
155 setpwent();
156 struct passwd *user_entry;
157 const char *name_start = partial_name + 1;
158 std::set<std::string> name_list;
159
160 while ((user_entry = getpwent()) != NULL)
161 {
162 if (strstr(user_entry->pw_name, name_start) == user_entry->pw_name)
163 {
164 std::string tmp_buf("~");
165 tmp_buf.append(user_entry->pw_name);
166 tmp_buf.push_back('/');
167 name_list.insert(tmp_buf);
168 }
169 }
170 std::set<std::string>::iterator pos, end = name_list.end();
171 for (pos = name_list.begin(); pos != end; pos++)
172 {
173 matches.AppendString((*pos).c_str());
174 }
175 return matches.GetSize() - extant_entries;
176 #else
177 // Resolving home directories is not supported, just copy the path...
178 return 0;
179 #endif // #ifdef LLDB_CONFIG_TILDE_RESOLVES_TO_USER
180 }
181
182
183
184 size_t
Resolve(const char * src_path,char * dst_path,size_t dst_len)185 FileSpec::Resolve (const char *src_path, char *dst_path, size_t dst_len)
186 {
187 if (src_path == NULL || src_path[0] == '\0')
188 return 0;
189
190 // Glob if needed for ~/, otherwise copy in case src_path is same as dst_path...
191 char unglobbed_path[PATH_MAX];
192 #ifdef LLDB_CONFIG_TILDE_RESOLVES_TO_USER
193 if (src_path[0] == '~')
194 {
195 size_t return_count = ResolveUsername(src_path, unglobbed_path, sizeof(unglobbed_path));
196
197 // If we couldn't find the user referred to, or the resultant path was too long,
198 // then just copy over the src_path.
199 if (return_count == 0 || return_count >= sizeof(unglobbed_path))
200 ::snprintf (unglobbed_path, sizeof(unglobbed_path), "%s", src_path);
201 }
202 else
203 #endif // #ifdef LLDB_CONFIG_TILDE_RESOLVES_TO_USER
204 {
205 ::snprintf(unglobbed_path, sizeof(unglobbed_path), "%s", src_path);
206 }
207
208 // Now resolve the path if needed
209 char resolved_path[PATH_MAX];
210 if (::realpath (unglobbed_path, resolved_path))
211 {
212 // Success, copy the resolved path
213 return ::snprintf(dst_path, dst_len, "%s", resolved_path);
214 }
215 else
216 {
217 // Failed, just copy the unglobbed path
218 return ::snprintf(dst_path, dst_len, "%s", unglobbed_path);
219 }
220 }
221
FileSpec()222 FileSpec::FileSpec() :
223 m_directory(),
224 m_filename()
225 {
226 }
227
228 //------------------------------------------------------------------
229 // Default constructor that can take an optional full path to a
230 // file on disk.
231 //------------------------------------------------------------------
FileSpec(const char * pathname,bool resolve_path)232 FileSpec::FileSpec(const char *pathname, bool resolve_path) :
233 m_directory(),
234 m_filename(),
235 m_is_resolved(false)
236 {
237 if (pathname && pathname[0])
238 SetFile(pathname, resolve_path);
239 }
240
241 //------------------------------------------------------------------
242 // Copy constructor
243 //------------------------------------------------------------------
FileSpec(const FileSpec & rhs)244 FileSpec::FileSpec(const FileSpec& rhs) :
245 m_directory (rhs.m_directory),
246 m_filename (rhs.m_filename),
247 m_is_resolved (rhs.m_is_resolved)
248 {
249 }
250
251 //------------------------------------------------------------------
252 // Copy constructor
253 //------------------------------------------------------------------
FileSpec(const FileSpec * rhs)254 FileSpec::FileSpec(const FileSpec* rhs) :
255 m_directory(),
256 m_filename()
257 {
258 if (rhs)
259 *this = *rhs;
260 }
261
262 //------------------------------------------------------------------
263 // Virtual destrcuctor in case anyone inherits from this class.
264 //------------------------------------------------------------------
~FileSpec()265 FileSpec::~FileSpec()
266 {
267 }
268
269 //------------------------------------------------------------------
270 // Assignment operator.
271 //------------------------------------------------------------------
272 const FileSpec&
operator =(const FileSpec & rhs)273 FileSpec::operator= (const FileSpec& rhs)
274 {
275 if (this != &rhs)
276 {
277 m_directory = rhs.m_directory;
278 m_filename = rhs.m_filename;
279 m_is_resolved = rhs.m_is_resolved;
280 }
281 return *this;
282 }
283
284 //------------------------------------------------------------------
285 // Update the contents of this object with a new path. The path will
286 // be split up into a directory and filename and stored as uniqued
287 // string values for quick comparison and efficient memory usage.
288 //------------------------------------------------------------------
289 void
SetFile(const char * pathname,bool resolve)290 FileSpec::SetFile (const char *pathname, bool resolve)
291 {
292 m_filename.Clear();
293 m_directory.Clear();
294 m_is_resolved = false;
295 if (pathname == NULL || pathname[0] == '\0')
296 return;
297
298 char resolved_path[PATH_MAX];
299 bool path_fit = true;
300
301 if (resolve)
302 {
303 path_fit = (FileSpec::Resolve (pathname, resolved_path, sizeof(resolved_path)) < sizeof(resolved_path) - 1);
304 m_is_resolved = path_fit;
305 }
306 else
307 {
308 // Copy the path because "basename" and "dirname" want to muck with the
309 // path buffer
310 if (::strlen (pathname) > sizeof(resolved_path) - 1)
311 path_fit = false;
312 else
313 ::strcpy (resolved_path, pathname);
314 }
315
316
317 if (path_fit)
318 {
319 char *filename = ::basename (resolved_path);
320 if (filename)
321 {
322 m_filename.SetCString (filename);
323 // Truncate the basename off the end of the resolved path
324
325 // Only attempt to get the dirname if it looks like we have a path
326 if (strchr(resolved_path, '/'))
327 {
328 char *directory = ::dirname (resolved_path);
329
330 // Make sure we didn't get our directory resolved to "." without having
331 // specified
332 if (directory)
333 m_directory.SetCString(directory);
334 else
335 {
336 char *last_resolved_path_slash = strrchr(resolved_path, '/');
337 if (last_resolved_path_slash)
338 {
339 *last_resolved_path_slash = '\0';
340 m_directory.SetCString(resolved_path);
341 }
342 }
343 }
344 }
345 else
346 m_directory.SetCString(resolved_path);
347 }
348 }
349
350 //----------------------------------------------------------------------
351 // Convert to pointer operator. This allows code to check any FileSpec
352 // objects to see if they contain anything valid using code such as:
353 //
354 // if (file_spec)
355 // {}
356 //----------------------------------------------------------------------
operator bool() const357 FileSpec::operator bool() const
358 {
359 return m_filename || m_directory;
360 }
361
362 //----------------------------------------------------------------------
363 // Logical NOT operator. This allows code to check any FileSpec
364 // objects to see if they are invalid using code such as:
365 //
366 // if (!file_spec)
367 // {}
368 //----------------------------------------------------------------------
369 bool
operator !() const370 FileSpec::operator!() const
371 {
372 return !m_directory && !m_filename;
373 }
374
375 //------------------------------------------------------------------
376 // Equal to operator
377 //------------------------------------------------------------------
378 bool
operator ==(const FileSpec & rhs) const379 FileSpec::operator== (const FileSpec& rhs) const
380 {
381 if (m_filename == rhs.m_filename)
382 {
383 if (m_directory == rhs.m_directory)
384 return true;
385
386 // TODO: determine if we want to keep this code in here.
387 // The code below was added to handle a case where we were
388 // trying to set a file and line breakpoint and one path
389 // was resolved, and the other not and the directory was
390 // in a mount point that resolved to a more complete path:
391 // "/tmp/a.c" == "/private/tmp/a.c". I might end up pulling
392 // this out...
393 if (IsResolved() && rhs.IsResolved())
394 {
395 // Both paths are resolved, no need to look further...
396 return false;
397 }
398
399 FileSpec resolved_lhs(*this);
400
401 // If "this" isn't resolved, resolve it
402 if (!IsResolved())
403 {
404 if (resolved_lhs.ResolvePath())
405 {
406 // This path wasn't resolved but now it is. Check if the resolved
407 // directory is the same as our unresolved directory, and if so,
408 // we can mark this object as resolved to avoid more future resolves
409 m_is_resolved = (m_directory == resolved_lhs.m_directory);
410 }
411 else
412 return false;
413 }
414
415 FileSpec resolved_rhs(rhs);
416 if (!rhs.IsResolved())
417 {
418 if (resolved_rhs.ResolvePath())
419 {
420 // rhs's path wasn't resolved but now it is. Check if the resolved
421 // directory is the same as rhs's unresolved directory, and if so,
422 // we can mark this object as resolved to avoid more future resolves
423 rhs.m_is_resolved = (rhs.m_directory == resolved_rhs.m_directory);
424 }
425 else
426 return false;
427 }
428
429 // If we reach this point in the code we were able to resolve both paths
430 // and since we only resolve the paths if the basenames are equal, then
431 // we can just check if both directories are equal...
432 return resolved_lhs.GetDirectory() == resolved_rhs.GetDirectory();
433 }
434 return false;
435 }
436
437 //------------------------------------------------------------------
438 // Not equal to operator
439 //------------------------------------------------------------------
440 bool
operator !=(const FileSpec & rhs) const441 FileSpec::operator!= (const FileSpec& rhs) const
442 {
443 return !(*this == rhs);
444 }
445
446 //------------------------------------------------------------------
447 // Less than operator
448 //------------------------------------------------------------------
449 bool
operator <(const FileSpec & rhs) const450 FileSpec::operator< (const FileSpec& rhs) const
451 {
452 return FileSpec::Compare(*this, rhs, true) < 0;
453 }
454
455 //------------------------------------------------------------------
456 // Dump a FileSpec object to a stream
457 //------------------------------------------------------------------
458 Stream&
operator <<(Stream & s,const FileSpec & f)459 lldb_private::operator << (Stream &s, const FileSpec& f)
460 {
461 f.Dump(&s);
462 return s;
463 }
464
465 //------------------------------------------------------------------
466 // Clear this object by releasing both the directory and filename
467 // string values and making them both the empty string.
468 //------------------------------------------------------------------
469 void
Clear()470 FileSpec::Clear()
471 {
472 m_directory.Clear();
473 m_filename.Clear();
474 }
475
476 //------------------------------------------------------------------
477 // Compare two FileSpec objects. If "full" is true, then both
478 // the directory and the filename must match. If "full" is false,
479 // then the directory names for "a" and "b" are only compared if
480 // they are both non-empty. This allows a FileSpec object to only
481 // contain a filename and it can match FileSpec objects that have
482 // matching filenames with different paths.
483 //
484 // Return -1 if the "a" is less than "b", 0 if "a" is equal to "b"
485 // and "1" if "a" is greater than "b".
486 //------------------------------------------------------------------
487 int
Compare(const FileSpec & a,const FileSpec & b,bool full)488 FileSpec::Compare(const FileSpec& a, const FileSpec& b, bool full)
489 {
490 int result = 0;
491
492 // If full is true, then we must compare both the directory and filename.
493
494 // If full is false, then if either directory is empty, then we match on
495 // the basename only, and if both directories have valid values, we still
496 // do a full compare. This allows for matching when we just have a filename
497 // in one of the FileSpec objects.
498
499 if (full || (a.m_directory && b.m_directory))
500 {
501 result = ConstString::Compare(a.m_directory, b.m_directory);
502 if (result)
503 return result;
504 }
505 return ConstString::Compare (a.m_filename, b.m_filename);
506 }
507
508 bool
Equal(const FileSpec & a,const FileSpec & b,bool full)509 FileSpec::Equal (const FileSpec& a, const FileSpec& b, bool full)
510 {
511 if (!full && (a.GetDirectory().IsEmpty() || b.GetDirectory().IsEmpty()))
512 return a.m_filename == b.m_filename;
513 else
514 return a == b;
515 }
516
517
518
519 //------------------------------------------------------------------
520 // Dump the object to the supplied stream. If the object contains
521 // a valid directory name, it will be displayed followed by a
522 // directory delimiter, and the filename.
523 //------------------------------------------------------------------
524 void
Dump(Stream * s) const525 FileSpec::Dump(Stream *s) const
526 {
527 static ConstString g_slash_only ("/");
528 if (s)
529 {
530 m_directory.Dump(s);
531 if (m_directory && m_directory != g_slash_only)
532 s->PutChar('/');
533 m_filename.Dump(s);
534 }
535 }
536
537 //------------------------------------------------------------------
538 // Returns true if the file exists.
539 //------------------------------------------------------------------
540 bool
Exists() const541 FileSpec::Exists () const
542 {
543 struct stat file_stats;
544 return GetFileStats (this, &file_stats);
545 }
546
547 bool
ResolveExecutableLocation()548 FileSpec::ResolveExecutableLocation ()
549 {
550 if (!m_directory)
551 {
552 const char *file_cstr = m_filename.GetCString();
553 if (file_cstr)
554 {
555 const std::string file_str (file_cstr);
556 std::string path = llvm::sys::FindProgramByName (file_str);
557 llvm::StringRef dir_ref = llvm::sys::path::parent_path(path);
558 //llvm::StringRef dir_ref = path.getDirname();
559 if (! dir_ref.empty())
560 {
561 // FindProgramByName returns "." if it can't find the file.
562 if (strcmp (".", dir_ref.data()) == 0)
563 return false;
564
565 m_directory.SetCString (dir_ref.data());
566 if (Exists())
567 return true;
568 else
569 {
570 // If FindProgramByName found the file, it returns the directory + filename in its return results.
571 // We need to separate them.
572 FileSpec tmp_file (dir_ref.data(), false);
573 if (tmp_file.Exists())
574 {
575 m_directory = tmp_file.m_directory;
576 return true;
577 }
578 }
579 }
580 }
581 }
582
583 return false;
584 }
585
586 bool
ResolvePath()587 FileSpec::ResolvePath ()
588 {
589 if (m_is_resolved)
590 return true; // We have already resolved this path
591
592 char path_buf[PATH_MAX];
593 if (!GetPath (path_buf, PATH_MAX))
594 return false;
595 // SetFile(...) will set m_is_resolved correctly if it can resolve the path
596 SetFile (path_buf, true);
597 return m_is_resolved;
598 }
599
600 uint64_t
GetByteSize() const601 FileSpec::GetByteSize() const
602 {
603 struct stat file_stats;
604 if (GetFileStats (this, &file_stats))
605 return file_stats.st_size;
606 return 0;
607 }
608
609 FileSpec::FileType
GetFileType() const610 FileSpec::GetFileType () const
611 {
612 struct stat file_stats;
613 if (GetFileStats (this, &file_stats))
614 {
615 mode_t file_type = file_stats.st_mode & S_IFMT;
616 switch (file_type)
617 {
618 case S_IFDIR: return eFileTypeDirectory;
619 case S_IFIFO: return eFileTypePipe;
620 case S_IFREG: return eFileTypeRegular;
621 case S_IFSOCK: return eFileTypeSocket;
622 case S_IFLNK: return eFileTypeSymbolicLink;
623 default:
624 break;
625 }
626 return eFileTypeUnknown;
627 }
628 return eFileTypeInvalid;
629 }
630
631 TimeValue
GetModificationTime() const632 FileSpec::GetModificationTime () const
633 {
634 TimeValue mod_time;
635 struct stat file_stats;
636 if (GetFileStats (this, &file_stats))
637 mod_time.OffsetWithSeconds(file_stats.st_mtime);
638 return mod_time;
639 }
640
641 //------------------------------------------------------------------
642 // Directory string get accessor.
643 //------------------------------------------------------------------
644 ConstString &
GetDirectory()645 FileSpec::GetDirectory()
646 {
647 return m_directory;
648 }
649
650 //------------------------------------------------------------------
651 // Directory string const get accessor.
652 //------------------------------------------------------------------
653 const ConstString &
GetDirectory() const654 FileSpec::GetDirectory() const
655 {
656 return m_directory;
657 }
658
659 //------------------------------------------------------------------
660 // Filename string get accessor.
661 //------------------------------------------------------------------
662 ConstString &
GetFilename()663 FileSpec::GetFilename()
664 {
665 return m_filename;
666 }
667
668 //------------------------------------------------------------------
669 // Filename string const get accessor.
670 //------------------------------------------------------------------
671 const ConstString &
GetFilename() const672 FileSpec::GetFilename() const
673 {
674 return m_filename;
675 }
676
677 //------------------------------------------------------------------
678 // Extract the directory and path into a fixed buffer. This is
679 // needed as the directory and path are stored in separate string
680 // values.
681 //------------------------------------------------------------------
682 size_t
GetPath(char * path,size_t path_max_len) const683 FileSpec::GetPath(char *path, size_t path_max_len) const
684 {
685 if (path_max_len)
686 {
687 const char *dirname = m_directory.GetCString();
688 const char *filename = m_filename.GetCString();
689 if (dirname)
690 {
691 if (filename)
692 return ::snprintf (path, path_max_len, "%s/%s", dirname, filename);
693 else
694 return ::snprintf (path, path_max_len, "%s", dirname);
695 }
696 else if (filename)
697 {
698 return ::snprintf (path, path_max_len, "%s", filename);
699 }
700 }
701 if (path)
702 path[0] = '\0';
703 return 0;
704 }
705
706 std::string
GetPath(void) const707 FileSpec::GetPath (void) const
708 {
709 static ConstString g_slash_only ("/");
710 std::string path;
711 const char *dirname = m_directory.GetCString();
712 const char *filename = m_filename.GetCString();
713 if (dirname)
714 {
715 path.append (dirname);
716 if (filename && m_directory != g_slash_only)
717 path.append ("/");
718 }
719 if (filename)
720 path.append (filename);
721 return path;
722 }
723
724 ConstString
GetFileNameExtension() const725 FileSpec::GetFileNameExtension () const
726 {
727 if (m_filename)
728 {
729 const char *filename = m_filename.GetCString();
730 const char* dot_pos = strrchr(filename, '.');
731 if (dot_pos && dot_pos[1] != '\0')
732 return ConstString(dot_pos+1);
733 }
734 return ConstString();
735 }
736
737 ConstString
GetFileNameStrippingExtension() const738 FileSpec::GetFileNameStrippingExtension () const
739 {
740 const char *filename = m_filename.GetCString();
741 if (filename == NULL)
742 return ConstString();
743
744 const char* dot_pos = strrchr(filename, '.');
745 if (dot_pos == NULL)
746 return m_filename;
747
748 return ConstString(filename, dot_pos-filename);
749 }
750
751 //------------------------------------------------------------------
752 // Returns a shared pointer to a data buffer that contains all or
753 // part of the contents of a file. The data is memory mapped and
754 // will lazily page in data from the file as memory is accessed.
755 // The data that is mappped will start "file_offset" bytes into the
756 // file, and "file_size" bytes will be mapped. If "file_size" is
757 // greater than the number of bytes available in the file starting
758 // at "file_offset", the number of bytes will be appropriately
759 // truncated. The final number of bytes that get mapped can be
760 // verified using the DataBuffer::GetByteSize() function.
761 //------------------------------------------------------------------
762 DataBufferSP
MemoryMapFileContents(off_t file_offset,size_t file_size) const763 FileSpec::MemoryMapFileContents(off_t file_offset, size_t file_size) const
764 {
765 DataBufferSP data_sp;
766 std::unique_ptr<DataBufferMemoryMap> mmap_data(new DataBufferMemoryMap());
767 if (mmap_data.get())
768 {
769 const size_t mapped_length = mmap_data->MemoryMapFromFileSpec (this, file_offset, file_size);
770 if (((file_size == SIZE_MAX) && (mapped_length > 0)) || (mapped_length >= file_size))
771 data_sp.reset(mmap_data.release());
772 }
773 return data_sp;
774 }
775
776
777 //------------------------------------------------------------------
778 // Return the size in bytes that this object takes in memory. This
779 // returns the size in bytes of this object, not any shared string
780 // values it may refer to.
781 //------------------------------------------------------------------
782 size_t
MemorySize() const783 FileSpec::MemorySize() const
784 {
785 return m_filename.MemorySize() + m_directory.MemorySize();
786 }
787
788
789 size_t
ReadFileContents(off_t file_offset,void * dst,size_t dst_len,Error * error_ptr) const790 FileSpec::ReadFileContents (off_t file_offset, void *dst, size_t dst_len, Error *error_ptr) const
791 {
792 Error error;
793 size_t bytes_read = 0;
794 char resolved_path[PATH_MAX];
795 if (GetPath(resolved_path, sizeof(resolved_path)))
796 {
797 File file;
798 error = file.Open(resolved_path, File::eOpenOptionRead);
799 if (error.Success())
800 {
801 off_t file_offset_after_seek = file_offset;
802 bytes_read = dst_len;
803 error = file.Read(dst, bytes_read, file_offset_after_seek);
804 }
805 }
806 else
807 {
808 error.SetErrorString("invalid file specification");
809 }
810 if (error_ptr)
811 *error_ptr = error;
812 return bytes_read;
813 }
814
815 //------------------------------------------------------------------
816 // Returns a shared pointer to a data buffer that contains all or
817 // part of the contents of a file. The data copies into a heap based
818 // buffer that lives in the DataBuffer shared pointer object returned.
819 // The data that is cached will start "file_offset" bytes into the
820 // file, and "file_size" bytes will be mapped. If "file_size" is
821 // greater than the number of bytes available in the file starting
822 // at "file_offset", the number of bytes will be appropriately
823 // truncated. The final number of bytes that get mapped can be
824 // verified using the DataBuffer::GetByteSize() function.
825 //------------------------------------------------------------------
826 DataBufferSP
ReadFileContents(off_t file_offset,size_t file_size,Error * error_ptr) const827 FileSpec::ReadFileContents (off_t file_offset, size_t file_size, Error *error_ptr) const
828 {
829 Error error;
830 DataBufferSP data_sp;
831 char resolved_path[PATH_MAX];
832 if (GetPath(resolved_path, sizeof(resolved_path)))
833 {
834 File file;
835 error = file.Open(resolved_path, File::eOpenOptionRead);
836 if (error.Success())
837 {
838 const bool null_terminate = false;
839 error = file.Read (file_size, file_offset, null_terminate, data_sp);
840 }
841 }
842 else
843 {
844 error.SetErrorString("invalid file specification");
845 }
846 if (error_ptr)
847 *error_ptr = error;
848 return data_sp;
849 }
850
851 DataBufferSP
ReadFileContentsAsCString(Error * error_ptr)852 FileSpec::ReadFileContentsAsCString(Error *error_ptr)
853 {
854 Error error;
855 DataBufferSP data_sp;
856 char resolved_path[PATH_MAX];
857 if (GetPath(resolved_path, sizeof(resolved_path)))
858 {
859 File file;
860 error = file.Open(resolved_path, File::eOpenOptionRead);
861 if (error.Success())
862 {
863 off_t offset = 0;
864 size_t length = SIZE_MAX;
865 const bool null_terminate = true;
866 error = file.Read (length, offset, null_terminate, data_sp);
867 }
868 }
869 else
870 {
871 error.SetErrorString("invalid file specification");
872 }
873 if (error_ptr)
874 *error_ptr = error;
875 return data_sp;
876 }
877
878 size_t
ReadFileLines(STLStringArray & lines)879 FileSpec::ReadFileLines (STLStringArray &lines)
880 {
881 lines.clear();
882 char path[PATH_MAX];
883 if (GetPath(path, sizeof(path)))
884 {
885 std::ifstream file_stream (path);
886
887 if (file_stream)
888 {
889 std::string line;
890 while (getline (file_stream, line))
891 lines.push_back (line);
892 }
893 }
894 return lines.size();
895 }
896
897 FileSpec::EnumerateDirectoryResult
EnumerateDirectory(const char * dir_path,bool find_directories,bool find_files,bool find_other,EnumerateDirectoryCallbackType callback,void * callback_baton)898 FileSpec::EnumerateDirectory
899 (
900 const char *dir_path,
901 bool find_directories,
902 bool find_files,
903 bool find_other,
904 EnumerateDirectoryCallbackType callback,
905 void *callback_baton
906 )
907 {
908 if (dir_path && dir_path[0])
909 {
910 lldb_utility::CleanUp <DIR *, int> dir_path_dir (opendir(dir_path), NULL, closedir);
911 if (dir_path_dir.is_valid())
912 {
913 long path_max = fpathconf (dirfd (dir_path_dir.get()), _PC_NAME_MAX);
914 #if defined (__APPLE_) && defined (__DARWIN_MAXPATHLEN)
915 if (path_max < __DARWIN_MAXPATHLEN)
916 path_max = __DARWIN_MAXPATHLEN;
917 #endif
918 struct dirent *buf, *dp;
919 buf = (struct dirent *) malloc (offsetof (struct dirent, d_name) + path_max + 1);
920
921 while (buf && readdir_r(dir_path_dir.get(), buf, &dp) == 0 && dp)
922 {
923 // Only search directories
924 if (dp->d_type == DT_DIR || dp->d_type == DT_UNKNOWN)
925 {
926 size_t len = strlen(dp->d_name);
927
928 if (len == 1 && dp->d_name[0] == '.')
929 continue;
930
931 if (len == 2 && dp->d_name[0] == '.' && dp->d_name[1] == '.')
932 continue;
933 }
934
935 bool call_callback = false;
936 FileSpec::FileType file_type = eFileTypeUnknown;
937
938 switch (dp->d_type)
939 {
940 default:
941 case DT_UNKNOWN: file_type = eFileTypeUnknown; call_callback = true; break;
942 case DT_FIFO: file_type = eFileTypePipe; call_callback = find_other; break;
943 case DT_CHR: file_type = eFileTypeOther; call_callback = find_other; break;
944 case DT_DIR: file_type = eFileTypeDirectory; call_callback = find_directories; break;
945 case DT_BLK: file_type = eFileTypeOther; call_callback = find_other; break;
946 case DT_REG: file_type = eFileTypeRegular; call_callback = find_files; break;
947 case DT_LNK: file_type = eFileTypeSymbolicLink; call_callback = find_other; break;
948 case DT_SOCK: file_type = eFileTypeSocket; call_callback = find_other; break;
949 #if !defined(__OpenBSD__)
950 case DT_WHT: file_type = eFileTypeOther; call_callback = find_other; break;
951 #endif
952 }
953
954 if (call_callback)
955 {
956 char child_path[PATH_MAX];
957 const int child_path_len = ::snprintf (child_path, sizeof(child_path), "%s/%s", dir_path, dp->d_name);
958 if (child_path_len < (int)(sizeof(child_path) - 1))
959 {
960 // Don't resolve the file type or path
961 FileSpec child_path_spec (child_path, false);
962
963 EnumerateDirectoryResult result = callback (callback_baton, file_type, child_path_spec);
964
965 switch (result)
966 {
967 case eEnumerateDirectoryResultNext:
968 // Enumerate next entry in the current directory. We just
969 // exit this switch and will continue enumerating the
970 // current directory as we currently are...
971 break;
972
973 case eEnumerateDirectoryResultEnter: // Recurse into the current entry if it is a directory or symlink, or next if not
974 if (FileSpec::EnumerateDirectory (child_path,
975 find_directories,
976 find_files,
977 find_other,
978 callback,
979 callback_baton) == eEnumerateDirectoryResultQuit)
980 {
981 // The subdirectory returned Quit, which means to
982 // stop all directory enumerations at all levels.
983 if (buf)
984 free (buf);
985 return eEnumerateDirectoryResultQuit;
986 }
987 break;
988
989 case eEnumerateDirectoryResultExit: // Exit from the current directory at the current level.
990 // Exit from this directory level and tell parent to
991 // keep enumerating.
992 if (buf)
993 free (buf);
994 return eEnumerateDirectoryResultNext;
995
996 case eEnumerateDirectoryResultQuit: // Stop directory enumerations at any level
997 if (buf)
998 free (buf);
999 return eEnumerateDirectoryResultQuit;
1000 }
1001 }
1002 }
1003 }
1004 if (buf)
1005 {
1006 free (buf);
1007 }
1008 }
1009 }
1010 // By default when exiting a directory, we tell the parent enumeration
1011 // to continue enumerating.
1012 return eEnumerateDirectoryResultNext;
1013 }
1014
1015 //------------------------------------------------------------------
1016 /// Returns true if the filespec represents an implementation source
1017 /// file (files with a ".c", ".cpp", ".m", ".mm" (many more)
1018 /// extension).
1019 ///
1020 /// @return
1021 /// \b true if the filespec represents an implementation source
1022 /// file, \b false otherwise.
1023 //------------------------------------------------------------------
1024 bool
IsSourceImplementationFile() const1025 FileSpec::IsSourceImplementationFile () const
1026 {
1027 ConstString extension (GetFileNameExtension());
1028 if (extension)
1029 {
1030 static RegularExpression g_source_file_regex ("^(c|m|mm|cpp|c\\+\\+|cxx|cc|cp|s|asm|f|f77|f90|f95|f03|for|ftn|fpp|ada|adb|ads)$",
1031 REG_EXTENDED | REG_ICASE);
1032 return g_source_file_regex.Execute (extension.GetCString());
1033 }
1034 return false;
1035 }
1036
1037 bool
IsRelativeToCurrentWorkingDirectory() const1038 FileSpec::IsRelativeToCurrentWorkingDirectory () const
1039 {
1040 const char *directory = m_directory.GetCString();
1041 if (directory && directory[0])
1042 {
1043 // If the path doesn't start with '/' or '~', return true
1044 switch (directory[0])
1045 {
1046 case '/':
1047 case '~':
1048 return false;
1049 default:
1050 return true;
1051 }
1052 }
1053 else if (m_filename)
1054 {
1055 // No directory, just a basename, return true
1056 return true;
1057 }
1058 return false;
1059 }
1060