contrib/llvm/tools/lldb/source/Plugins/Process/Darwin/CFString.cpp

   1 //===-- CFString.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 //  Created by Greg Clayton on 1/16/08.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #include "CFString.h"
  15 #include <glob.h>
  16 #include <string>
  17
  18 //----------------------------------------------------------------------
  19 // CFString constructor
  20 //----------------------------------------------------------------------
  21 CFString::CFString(CFStringRef s) : CFReleaser<CFStringRef>(s) {}
  22
  23 //----------------------------------------------------------------------
  24 // CFString copy constructor
  25 //----------------------------------------------------------------------
  26 CFString::CFString(const CFString &rhs) : CFReleaser<CFStringRef>(rhs) {}
  27
  28 //----------------------------------------------------------------------
  29 // CFString copy constructor
  30 //----------------------------------------------------------------------
  31 CFString &CFString::operator=(const CFString &rhs) {
  32   if (this != &rhs)
  33     *this = rhs;
  34   return *this;
  35 }
  36
  37 CFString::CFString(const char *cstr, CFStringEncoding cstr_encoding)
  38     : CFReleaser<CFStringRef>() {
  39   if (cstr && cstr[0]) {
  40     reset(
  41         ::CFStringCreateWithCString(kCFAllocatorDefault, cstr, cstr_encoding));
  42   }
  43 }
  44
  45 //----------------------------------------------------------------------
  46 // Destructor
  47 //----------------------------------------------------------------------
  48 CFString::~CFString() {}
  49
  50 const char *CFString::GetFileSystemRepresentation(std::string &s) {
  51   return CFString::FileSystemRepresentation(get(), s);
  52 }
  53
  54 CFStringRef CFString::SetFileSystemRepresentation(const char *path) {
  55   CFStringRef new_value = NULL;
  56   if (path && path[0])
  57     new_value =
  58         ::CFStringCreateWithFileSystemRepresentation(kCFAllocatorDefault, path);
  59   reset(new_value);
  60   return get();
  61 }
  62
  63 CFStringRef CFString::SetFileSystemRepresentationFromCFType(CFTypeRef cf_type) {
  64   CFStringRef new_value = NULL;
  65   if (cf_type != NULL) {
  66     CFTypeID cf_type_id = ::CFGetTypeID(cf_type);
  67
  68     if (cf_type_id == ::CFStringGetTypeID()) {
  69       // Retain since we are using the existing object
  70       new_value = (CFStringRef)::CFRetain(cf_type);
  71     } else if (cf_type_id == ::CFURLGetTypeID()) {
  72       new_value =
  73           ::CFURLCopyFileSystemPath((CFURLRef)cf_type, kCFURLPOSIXPathStyle);
  74     }
  75   }
  76   reset(new_value);
  77   return get();
  78 }
  79
  80 CFStringRef
  81 CFString::SetFileSystemRepresentationAndExpandTilde(const char *path) {
  82   std::string expanded_path;
  83   if (CFString::GlobPath(path, expanded_path))
  84     SetFileSystemRepresentation(expanded_path.c_str());
  85   else
  86     reset();
  87   return get();
  88 }
  89
  90 const char *CFString::UTF8(std::string &str) {
  91   return CFString::UTF8(get(), str);
  92 }
  93
  94 // Static function that puts a copy of the UTF8 contents of CF_STR into STR and
  95 // returns the C string pointer that is contained in STR when successful, else
  96 // NULL is returned. This allows the std::string parameter to own the extracted
  97 // string,
  98 // and also allows that string to be returned as a C string pointer that can be
  99 // used.
 100
 101 const char *CFString::UTF8(CFStringRef cf_str, std::string &str) {
 102   if (cf_str) {
 103     const CFStringEncoding encoding = kCFStringEncodingUTF8;
 104     CFIndex max_utf8_str_len = CFStringGetLength(cf_str);
 105     max_utf8_str_len =
 106         CFStringGetMaximumSizeForEncoding(max_utf8_str_len, encoding);
 107     if (max_utf8_str_len > 0) {
 108       str.resize(max_utf8_str_len);
 109       if (!str.empty()) {
 110         if (CFStringGetCString(cf_str, &str[0], str.size(), encoding)) {
 111           str.resize(strlen(str.c_str()));
 112           return str.c_str();
 113         }
 114       }
 115     }
 116   }
 117   return NULL;
 118 }
 119
 120 // Static function that puts a copy of the file system representation of CF_STR
 121 // into STR and returns the C string pointer that is contained in STR when
 122 // successful, else NULL is returned. This allows the std::string parameter to
 123 // own the extracted string, and also allows that string to be returned as a C
 124 // string pointer that can be used.
 125
 126 const char *CFString::FileSystemRepresentation(CFStringRef cf_str,
 127                                                std::string &str) {
 128   if (cf_str) {
 129     CFIndex max_length =
 130         ::CFStringGetMaximumSizeOfFileSystemRepresentation(cf_str);
 131     if (max_length > 0) {
 132       str.resize(max_length);
 133       if (!str.empty()) {
 134         if (::CFStringGetFileSystemRepresentation(cf_str, &str[0],
 135                                                   str.size())) {
 136           str.erase(::strlen(str.c_str()));
 137           return str.c_str();
 138         }
 139       }
 140     }
 141   }
 142   str.erase();
 143   return NULL;
 144 }
 145
 146 CFIndex CFString::GetLength() const {
 147   CFStringRef str = get();
 148   if (str)
 149     return CFStringGetLength(str);
 150   return 0;
 151 }
 152
 153 const char *CFString::GlobPath(const char *path, std::string &expanded_path) {
 154   glob_t globbuf;
 155   if (::glob(path, GLOB_TILDE, NULL, &globbuf) == 0) {
 156     expanded_path = globbuf.gl_pathv[0];
 157     ::globfree(&globbuf);
 158   } else
 159     expanded_path.clear();
 160
 161   return expanded_path.c_str();
 162 }