Merge 264d006dd2 into 04c2b05340

format.cc

@@ -410,233 +410,240 @@ void fmt::BasicFormatter<Char>::CheckSign(const Char *&s, const Arg &arg) {
 template <typename Char>
 void fmt::BasicFormatter<Char>::DoFormat() {
   const Char *start = format_;
+  const Char *original = format_; // capture the format string before it is reset
   format_ = 0;
   next_arg_index_ = 0;
   const Char *s = start;
   BasicWriter<Char> &writer = *writer_;
-  while (*s) {
+  try {
-    Char c = *s++;
+    while (*s) {
-    if (c != '{' && c != '}') continue;
+      Char c = *s++;
-    if (*s == c) {
+      if (c != '{' && c != '}') continue;
-      writer.buffer_.append(start, s);
+      if (*s == c) {
-      start = ++s;
+        writer.buffer_.append(start, s);
-      continue;
+        start = ++s;
-    }
+        continue;
-    if (c == '}')
-      throw FormatError("unmatched '}' in format");
-    num_open_braces_= 1;
-    writer.buffer_.append(start, s - 1);
-
-    const Arg &arg = ParseArgIndex(s);
-
-    FormatSpec spec;
-    int precision = -1;
-    if (*s == ':') {
-      ++s;
-
-      // Parse fill and alignment.
-      if (Char c = *s) {
-        const Char *p = s + 1;
-        spec.align_ = ALIGN_DEFAULT;
-        do {
-          switch (*p) {
-          case '<':
-            spec.align_ = ALIGN_LEFT;
-            break;
-          case '>':
-            spec.align_ = ALIGN_RIGHT;
-            break;
-          case '=':
-            spec.align_ = ALIGN_NUMERIC;
-            break;
-          case '^':
-            spec.align_ = ALIGN_CENTER;
-            break;
-          }
-          if (spec.align_ != ALIGN_DEFAULT) {
-            if (p != s) {
-              if (c == '}') break;
-              if (c == '{')
-                ReportError(s, "invalid fill character '{'");
-              s += 2;
-              spec.fill_ = c;
-            } else ++s;
-            if (spec.align_ == ALIGN_NUMERIC && arg.type > LAST_NUMERIC_TYPE)
-              ReportError(s, "format specifier '=' requires numeric argument");
-            break;
-          }
-        } while (--p >= s);
       }
+      if (c == '}')
+        throw FormatError("unmatched '}' in format");
+      num_open_braces_= 1;
+      writer.buffer_.append(start, s - 1);
 
-      // Parse sign.
+      const Arg &arg = ParseArgIndex(s);
-      switch (*s) {
-      case '+':
-        CheckSign(s, arg);
-        spec.flags_ |= SIGN_FLAG | PLUS_FLAG;
-        break;
-      case '-':
-        CheckSign(s, arg);
-        break;
-      case ' ':
-        CheckSign(s, arg);
-        spec.flags_ |= SIGN_FLAG;
-        break;
-      }
 
-      if (*s == '#') {
+      FormatSpec spec;
-        if (arg.type > LAST_NUMERIC_TYPE)
+      int precision = -1;
-          ReportError(s, "format specifier '#' requires numeric argument");
+      if (*s == ':') {
-        spec.flags_ |= HASH_FLAG;
         ++s;
-      }
 
-      // Parse width and zero flag.
+        // Parse fill and alignment.
-      if ('0' <= *s && *s <= '9') {
+        if (Char c = *s) {
-        if (*s == '0') {
+          const Char *p = s + 1;
-          if (arg.type > LAST_NUMERIC_TYPE)
+          spec.align_ = ALIGN_DEFAULT;
-            ReportError(s, "format specifier '0' requires numeric argument");
+          do {
-          spec.align_ = ALIGN_NUMERIC;
+            switch (*p) {
-          spec.fill_ = '0';
+            case '<':
+              spec.align_ = ALIGN_LEFT;
+              break;
+            case '>':
+              spec.align_ = ALIGN_RIGHT;
+              break;
+            case '=':
+              spec.align_ = ALIGN_NUMERIC;
+              break;
+            case '^':
+              spec.align_ = ALIGN_CENTER;
+              break;
+            }
+            if (spec.align_ != ALIGN_DEFAULT) {
+              if (p != s) {
+                if (c == '}') break;
+                if (c == '{')
+                  ReportError(s, "invalid fill character '{'");
+                s += 2;
+                spec.fill_ = c;
+              } else ++s;
+              if (spec.align_ == ALIGN_NUMERIC && arg.type > LAST_NUMERIC_TYPE)
+                ReportError(s, "format specifier '=' requires numeric argument");
+              break;
+            }
+          } while (--p >= s);
         }
-        // Zero may be parsed again as a part of the width, but it is simpler
-        // and more efficient than checking if the next char is a digit.
-        unsigned value = ParseUInt(s);
-        if (value > INT_MAX)
-          ReportError(s, "number is too big in format");
-        spec.width_ = value;
-      }
 
-      // Parse precision.
+        // Parse sign.
-      if (*s == '.') {
+        switch (*s) {
-        ++s;
+        case '+':
-        precision = 0;
+          CheckSign(s, arg);
+          spec.flags_ |= SIGN_FLAG | PLUS_FLAG;
+          break;
+        case '-':
+          CheckSign(s, arg);
+          break;
+        case ' ':
+          CheckSign(s, arg);
+          spec.flags_ |= SIGN_FLAG;
+          break;
+        }
+
+        if (*s == '#') {
+          if (arg.type > LAST_NUMERIC_TYPE)
+            ReportError(s, "format specifier '#' requires numeric argument");
+          spec.flags_ |= HASH_FLAG;
+          ++s;
+        }
+
+        // Parse width and zero flag.
         if ('0' <= *s && *s <= '9') {
+          if (*s == '0') {
+            if (arg.type > LAST_NUMERIC_TYPE)
+              ReportError(s, "format specifier '0' requires numeric argument");
+            spec.align_ = ALIGN_NUMERIC;
+            spec.fill_ = '0';
+          }
+          // Zero may be parsed again as a part of the width, but it is simpler
+          // and more efficient than checking if the next char is a digit.
           unsigned value = ParseUInt(s);
           if (value > INT_MAX)
             ReportError(s, "number is too big in format");
-          precision = value;
+          spec.width_ = value;
-        } else if (*s == '{') {
+        }
+
+        // Parse precision.
+        if (*s == '.') {
           ++s;
-          ++num_open_braces_;
+          precision = 0;
-          const Arg &precision_arg = ParseArgIndex(s);
+          if ('0' <= *s && *s <= '9') {
-          unsigned long value = 0;
+            unsigned value = ParseUInt(s);
-          switch (precision_arg.type) {
+            if (value > INT_MAX)
-          case INT:
+              ReportError(s, "number is too big in format");
-            if (precision_arg.int_value < 0)
+            precision = value;
-              ReportError(s, "negative precision in format");
+          } else if (*s == '{') {
-            value = precision_arg.int_value;
+            ++s;
-            break;
+            ++num_open_braces_;
-          case UINT:
+            const Arg &precision_arg = ParseArgIndex(s);
-            value = precision_arg.uint_value;
+            unsigned long value = 0;
-            break;
+            switch (precision_arg.type) {
-          case LONG:
+            case INT:
-            if (precision_arg.long_value < 0)
+              if (precision_arg.int_value < 0)
-              ReportError(s, "negative precision in format");
+                ReportError(s, "negative precision in format");
-            value = precision_arg.long_value;
+              value = precision_arg.int_value;
-            break;
+              break;
-          case ULONG:
+            case UINT:
-            value = precision_arg.ulong_value;
+              value = precision_arg.uint_value;
-            break;
+              break;
-          default:
+            case LONG:
-            ReportError(s, "precision is not integer");
+              if (precision_arg.long_value < 0)
+                ReportError(s, "negative precision in format");
+              value = precision_arg.long_value;
+              break;
+            case ULONG:
+              value = precision_arg.ulong_value;
+              break;
+            default:
+              ReportError(s, "precision is not integer");
+            }
+            if (value > INT_MAX)
+              ReportError(s, "number is too big in format");
+            precision = static_cast<int>(value);
+            if (*s++ != '}')
+              throw FormatError("unmatched '{' in format");
+            --num_open_braces_;
+          } else {
+            ReportError(s, "missing precision in format");
+          }
+          if (arg.type != DOUBLE && arg.type != LONG_DOUBLE) {
+            ReportError(s,
+                "precision specifier requires floating-point argument");
+          }
+        }
+
+        // Parse type.
+        if (*s != '}' && *s)
+          spec.type_ = static_cast<char>(*s++);
+      }
+
+      if (*s++ != '}')
+        throw FormatError("unmatched '{' in format");
+      start = s;
+
+      // Format argument.
+      switch (arg.type) {
+      case INT:
+        writer.FormatInt(arg.int_value, spec);
+        break;
+      case UINT:
+        writer.FormatInt(arg.uint_value, spec);
+        break;
+      case LONG:
+        writer.FormatInt(arg.long_value, spec);
+        break;
+      case ULONG:
+        writer.FormatInt(arg.ulong_value, spec);
+        break;
+      case DOUBLE:
+        writer.FormatDouble(arg.double_value, spec, precision);
+        break;
+      case LONG_DOUBLE:
+        writer.FormatDouble(arg.long_double_value, spec, precision);
+        break;
+      case CHAR: {
+        if (spec.type_ && spec.type_ != 'c')
+          internal::ReportUnknownType(spec.type_, "char");
+        typedef typename BasicWriter<Char>::CharPtr CharPtr;
+        CharPtr out = CharPtr();
+        if (spec.width_ > 1) {
+          Char fill = static_cast<Char>(spec.fill());
+          out = writer.GrowBuffer(spec.width_);
+          if (spec.align_ == ALIGN_RIGHT) {
+            std::fill_n(out, spec.width_ - 1, fill);
+            out += spec.width_ - 1;
+          } else if (spec.align_ == ALIGN_CENTER) {
+            out = writer.FillPadding(out, spec.width_, 1, fill);
+          } else {
+            std::fill_n(out + 1, spec.width_ - 1, fill);
           }
-          if (value > INT_MAX)
-            ReportError(s, "number is too big in format");
-          precision = value;
-          if (*s++ != '}')
-            throw FormatError("unmatched '{' in format");
-          --num_open_braces_;
         } else {
-          ReportError(s, "missing precision in format");
+          out = writer.GrowBuffer(1);
         }
-        if (arg.type != DOUBLE && arg.type != LONG_DOUBLE) {
+        *out = arg.int_value;
-          ReportError(s,
+        break;
-              "precision specifier requires floating-point argument");
+      }
+      case STRING: {
+        if (spec.type_ && spec.type_ != 's')
+          internal::ReportUnknownType(spec.type_, "string");
+        const Char *str = arg.string.value;
+        std::size_t size = arg.string.size;
+        if (size == 0) {
+          if (!str)
+            throw FormatError("string pointer is null");
+          if (*str)
+            size = std::char_traits<Char>::length(str);
         }
+        writer.FormatString(str, size, spec);
+        break;
       }
-
+      case POINTER:
-      // Parse type.
+        if (spec.type_ && spec.type_ != 'p')
-      if (*s != '}' && *s)
+          internal::ReportUnknownType(spec.type_, "pointer");
-        spec.type_ = static_cast<char>(*s++);
+        spec.flags_= HASH_FLAG;
-    }
+        spec.type_ = 'x';
-
+        writer.FormatInt(reinterpret_cast<uintptr_t>(arg.pointer_value), spec);
-    if (*s++ != '}')
+        break;
-      throw FormatError("unmatched '{' in format");
+      case CUSTOM:
-    start = s;
+        if (spec.type_)
-
+          internal::ReportUnknownType(spec.type_, "object");
-    // Format argument.
+        arg.custom.format(writer, arg.custom.value, spec);
-    switch (arg.type) {
+        break;
-    case INT:
+      default:
-      writer.FormatInt(arg.int_value, spec);
+        assert(false);
-      break;
+        break;
-    case UINT:
-      writer.FormatInt(arg.uint_value, spec);
-      break;
-    case LONG:
-      writer.FormatInt(arg.long_value, spec);
-      break;
-    case ULONG:
-      writer.FormatInt(arg.ulong_value, spec);
-      break;
-    case DOUBLE:
-      writer.FormatDouble(arg.double_value, spec, precision);
-      break;
-    case LONG_DOUBLE:
-      writer.FormatDouble(arg.long_double_value, spec, precision);
-      break;
-    case CHAR: {
-      if (spec.type_ && spec.type_ != 'c')
-        internal::ReportUnknownType(spec.type_, "char");
-      typedef typename BasicWriter<Char>::CharPtr CharPtr;
-      CharPtr out = CharPtr();
-      if (spec.width_ > 1) {
-        Char fill = static_cast<Char>(spec.fill());
-        out = writer.GrowBuffer(spec.width_);
-        if (spec.align_ == ALIGN_RIGHT) {
-          std::fill_n(out, spec.width_ - 1, fill);
-          out += spec.width_ - 1;
-        } else if (spec.align_ == ALIGN_CENTER) {
-          out = writer.FillPadding(out, spec.width_, 1, fill);
-        } else {
-          std::fill_n(out + 1, spec.width_ - 1, fill);
-        }
-      } else {
-        out = writer.GrowBuffer(1);
       }
-      *out = arg.int_value;
-      break;
-    }
-    case STRING: {
-      if (spec.type_ && spec.type_ != 's')
-        internal::ReportUnknownType(spec.type_, "string");
-      const Char *str = arg.string.value;
-      std::size_t size = arg.string.size;
-      if (size == 0) {
-        if (!str)
-          throw FormatError("string pointer is null");
-        if (*str)
-          size = std::char_traits<Char>::length(str);
-      }
-      writer.FormatString(str, size, spec);
-      break;
-    }
-    case POINTER:
-      if (spec.type_ && spec.type_ != 'p')
-        internal::ReportUnknownType(spec.type_, "pointer");
-      spec.flags_= HASH_FLAG;
-      spec.type_ = 'x';
-      writer.FormatInt(reinterpret_cast<uintptr_t>(arg.pointer_value), spec);
-      break;
-    case CUSTOM:
-      if (spec.type_)
-        internal::ReportUnknownType(spec.type_, "object");
-      arg.custom.format(writer, arg.custom.value, spec);
-      break;
-    default:
-      assert(false);
-      break;
     }  
+  } catch (const FormatError &e) {
+    // rethrow FormatError with the format string pointed to by start
+    throw BasicFormatError<Char>(e.what(), original);
   }
+  
   writer.buffer_.append(start, s);
 }
 
@@ -672,6 +679,13 @@ template void fmt::BasicFormatter<char>::CheckSign(
 
 template void fmt::BasicFormatter<char>::DoFormat();
 
+template<> fmt::BasicFormatError<char>::BasicFormatError(const std::string &message, const char *format)
+    : std::runtime_error(message), format_(format) {}
+
+template<> fmt::BasicFormatError<char>::~BasicFormatError() {
+    std::runtime_error::~runtime_error();
+}
+
 // Explicit instantiations for wchar_t.
 
 template void fmt::BasicWriter<wchar_t>::FormatDouble<double>(
@@ -704,3 +718,10 @@ template void fmt::BasicFormatter<wchar_t>::CheckSign(
     const wchar_t *&s, const Arg &arg);
 
 template void fmt::BasicFormatter<wchar_t>::DoFormat();
+
+template<> fmt::BasicFormatError<wchar_t>::BasicFormatError(const std::string &message, const wchar_t *format)
+    : std::runtime_error(message), format_(format){}
+
+template<> fmt::BasicFormatError<wchar_t>::~BasicFormatError() {
+    std::runtime_error::~runtime_error();
+}

format.h

@@ -189,6 +189,9 @@ struct SignedIntTraits {
 template <>
 struct IntTraits<int> : SignedIntTraits<int, unsigned> {};
 
+template <>
+struct IntTraits<uint32_t> : SignedIntTraits<uint32_t, unsigned> {};
+
 template <>
 struct IntTraits<long> : SignedIntTraits<long, unsigned long> {};
 
@@ -289,6 +292,16 @@ class FormatError : public std::runtime_error {
   : std::runtime_error(message) {}
 };
 
+template <typename Char>
+class BasicFormatError : public std::runtime_error {
+private:
+    std::basic_string<Char> format_;
+public:
+    explicit BasicFormatError(const std::string &message, const Char *format);
+    virtual ~BasicFormatError() throw();    
+    const Char *format() const { return format_.c_str(); }
+};
+
 enum Alignment {
   ALIGN_DEFAULT, ALIGN_LEFT, ALIGN_RIGHT, ALIGN_CENTER, ALIGN_NUMERIC
 };
@@ -348,7 +361,7 @@ struct FormatSpec : AlignSpec {
   char type_;
 
   FormatSpec(unsigned width = 0, char type = 0, wchar_t fill = ' ')
-  : AlignSpec(width, fill), flags_(0), type_(type) {}
+    : AlignSpec(width, fill), flags_(0), type_(type) {}
 
   Alignment align() const { return align_; }