// Copyright 2020 The Go Authors. All rights reserved. // Use of this source code is governed by a BSD-style // license that can be found in the LICENSE file. //go:build goexperiment.jsonv2 package json import ( "bytes" "cmp" "errors" "fmt" "math" "math/bits" "reflect" "strconv" "strings" "time" "encoding/json/internal" "encoding/json/internal/jsonflags" "encoding/json/internal/jsonopts" "encoding/json/internal/jsonwire" "encoding/json/jsontext" ) var ( timeDurationType = reflect.TypeFor[time.Duration]() timeTimeType = reflect.TypeFor[time.Time]() ) func makeTimeArshaler(fncs *arshaler, t reflect.Type) *arshaler { // Ideally, time types would implement MarshalerTo and UnmarshalerFrom, // but that would incur a dependency on package json from package time. // Given how widely used time is, it is more acceptable that we incur a // dependency on time from json. // // Injecting the arshaling functionality like this will not be identical // to actually declaring methods on the time types since embedding of the // time types will not be able to forward this functionality. switch t { case timeDurationType: fncs.nonDefault = true marshalNano := fncs.marshal fncs.marshal = func(enc *jsontext.Encoder, va addressableValue, mo *jsonopts.Struct) error { xe := export.Encoder(enc) var m durationArshaler if mo.Format != "" && mo.FormatDepth == xe.Tokens.Depth() { if !m.initFormat(mo.Format) { return newInvalidFormatError(enc, t, mo) } } else if mo.Flags.Get(jsonflags.FormatTimeWithLegacySemantics) { return marshalNano(enc, va, mo) } else { // TODO(https://go.dev/issue/71631): Decide on default duration representation. return newMarshalErrorBefore(enc, t, errors.New("no default representation (see https://go.dev/issue/71631); specify an explicit format")) } // TODO(https://go.dev/issue/62121): Use reflect.Value.AssertTo. m.td = *va.Addr().Interface().(*time.Duration) k := stringOrNumberKind(!m.isNumeric() || xe.Tokens.Last.NeedObjectName() || mo.Flags.Get(jsonflags.StringifyNumbers)) if err := xe.AppendRaw(k, true, m.appendMarshal); err != nil { if !isSyntacticError(err) && !export.IsIOError(err) { err = newMarshalErrorBefore(enc, t, err) } return err } return nil } unmarshalNano := fncs.unmarshal fncs.unmarshal = func(dec *jsontext.Decoder, va addressableValue, uo *jsonopts.Struct) error { xd := export.Decoder(dec) var u durationArshaler if uo.Format != "" && uo.FormatDepth == xd.Tokens.Depth() { if !u.initFormat(uo.Format) { return newInvalidFormatError(dec, t, uo) } } else if uo.Flags.Get(jsonflags.FormatTimeWithLegacySemantics) { return unmarshalNano(dec, va, uo) } else { // TODO(https://go.dev/issue/71631): Decide on default duration representation. return newUnmarshalErrorBeforeWithSkipping(dec, uo, t, errors.New("no default representation (see https://go.dev/issue/71631); specify an explicit format")) } stringify := !u.isNumeric() || xd.Tokens.Last.NeedObjectName() || uo.Flags.Get(jsonflags.StringifyNumbers) var flags jsonwire.ValueFlags td := va.Addr().Interface().(*time.Duration) val, err := xd.ReadValue(&flags) if err != nil { return err } switch k := val.Kind(); k { case 'n': if !uo.Flags.Get(jsonflags.MergeWithLegacySemantics) { *td = time.Duration(0) } return nil case '"': if !stringify { break } val = jsonwire.UnquoteMayCopy(val, flags.IsVerbatim()) if err := u.unmarshal(val); err != nil { return newUnmarshalErrorAfter(dec, t, err) } *td = u.td return nil case '0': if stringify { break } if err := u.unmarshal(val); err != nil { return newUnmarshalErrorAfter(dec, t, err) } *td = u.td return nil } return newUnmarshalErrorAfter(dec, t, nil) } case timeTimeType: fncs.nonDefault = true fncs.marshal = func(enc *jsontext.Encoder, va addressableValue, mo *jsonopts.Struct) (err error) { xe := export.Encoder(enc) var m timeArshaler if mo.Format != "" && mo.FormatDepth == xe.Tokens.Depth() { if !m.initFormat(mo.Format) { return newInvalidFormatError(enc, t, mo) } } // TODO(https://go.dev/issue/62121): Use reflect.Value.AssertTo. m.tt = *va.Addr().Interface().(*time.Time) k := stringOrNumberKind(!m.isNumeric() || xe.Tokens.Last.NeedObjectName() || mo.Flags.Get(jsonflags.StringifyNumbers)) if err := xe.AppendRaw(k, !m.hasCustomFormat(), m.appendMarshal); err != nil { if mo.Flags.Get(jsonflags.ReportErrorsWithLegacySemantics) { return internal.NewMarshalerError(va.Addr().Interface(), err, "MarshalJSON") // unlike unmarshal, always wrapped } if !isSyntacticError(err) && !export.IsIOError(err) { err = newMarshalErrorBefore(enc, t, err) } return err } return nil } fncs.unmarshal = func(dec *jsontext.Decoder, va addressableValue, uo *jsonopts.Struct) (err error) { xd := export.Decoder(dec) var u timeArshaler if uo.Format != "" && uo.FormatDepth == xd.Tokens.Depth() { if !u.initFormat(uo.Format) { return newInvalidFormatError(dec, t, uo) } } else if uo.Flags.Get(jsonflags.FormatTimeWithLegacySemantics) { u.looseRFC3339 = true } stringify := !u.isNumeric() || xd.Tokens.Last.NeedObjectName() || uo.Flags.Get(jsonflags.StringifyNumbers) var flags jsonwire.ValueFlags tt := va.Addr().Interface().(*time.Time) val, err := xd.ReadValue(&flags) if err != nil { return err } switch k := val.Kind(); k { case 'n': if !uo.Flags.Get(jsonflags.MergeWithLegacySemantics) { *tt = time.Time{} } return nil case '"': if !stringify { break } val = jsonwire.UnquoteMayCopy(val, flags.IsVerbatim()) if err := u.unmarshal(val); err != nil { if uo.Flags.Get(jsonflags.ReportErrorsWithLegacySemantics) { return err // unlike marshal, never wrapped } return newUnmarshalErrorAfter(dec, t, err) } *tt = u.tt return nil case '0': if stringify { break } if err := u.unmarshal(val); err != nil { if uo.Flags.Get(jsonflags.ReportErrorsWithLegacySemantics) { return err // unlike marshal, never wrapped } return newUnmarshalErrorAfter(dec, t, err) } *tt = u.tt return nil } return newUnmarshalErrorAfter(dec, t, nil) } } return fncs } type durationArshaler struct { td time.Duration // base records the representation where: // - 0 uses time.Duration.String // - 1e0, 1e3, 1e6, or 1e9 use a decimal encoding of the duration as // nanoseconds, microseconds, milliseconds, or seconds. // - 8601 uses ISO 8601 base uint64 } func (a *durationArshaler) initFormat(format string) (ok bool) { switch format { case "units": a.base = 0 case "sec": a.base = 1e9 case "milli": a.base = 1e6 case "micro": a.base = 1e3 case "nano": a.base = 1e0 case "iso8601": a.base = 8601 default: return false } return true } func (a *durationArshaler) isNumeric() bool { return a.base != 0 && a.base != 8601 } func (a *durationArshaler) appendMarshal(b []byte) ([]byte, error) { switch a.base { case 0: return append(b, a.td.String()...), nil case 8601: return appendDurationISO8601(b, a.td), nil default: return appendDurationBase10(b, a.td, a.base), nil } } func (a *durationArshaler) unmarshal(b []byte) (err error) { switch a.base { case 0: a.td, err = time.ParseDuration(string(b)) case 8601: a.td, err = parseDurationISO8601(b) default: a.td, err = parseDurationBase10(b, a.base) } return err } type timeArshaler struct { tt time.Time // base records the representation where: // - 0 uses RFC 3339 encoding of the timestamp // - 1e0, 1e3, 1e6, or 1e9 use a decimal encoding of the timestamp as // seconds, milliseconds, microseconds, or nanoseconds since Unix epoch. // - math.MaxUint uses time.Time.Format to encode the timestamp base uint64 format string // time format passed to time.Parse looseRFC3339 bool } func (a *timeArshaler) initFormat(format string) bool { // We assume that an exported constant in the time package will // always start with an uppercase ASCII letter. if len(format) == 0 { return false } a.base = math.MaxUint // implies custom format if c := format[0]; !('a' <= c && c <= 'z') && !('A' <= c && c <= 'Z') { a.format = format return true } switch format { case "ANSIC": a.format = time.ANSIC case "UnixDate": a.format = time.UnixDate case "RubyDate": a.format = time.RubyDate case "RFC822": a.format = time.RFC822 case "RFC822Z": a.format = time.RFC822Z case "RFC850": a.format = time.RFC850 case "RFC1123": a.format = time.RFC1123 case "RFC1123Z": a.format = time.RFC1123Z case "RFC3339": a.base = 0 a.format = time.RFC3339 case "RFC3339Nano": a.base = 0 a.format = time.RFC3339Nano case "Kitchen": a.format = time.Kitchen case "Stamp": a.format = time.Stamp case "StampMilli": a.format = time.StampMilli case "StampMicro": a.format = time.StampMicro case "StampNano": a.format = time.StampNano case "DateTime": a.format = time.DateTime case "DateOnly": a.format = time.DateOnly case "TimeOnly": a.format = time.TimeOnly case "unix": a.base = 1e0 case "unixmilli": a.base = 1e3 case "unixmicro": a.base = 1e6 case "unixnano": a.base = 1e9 default: // Reject any Go identifier in case new constants are supported. if strings.TrimFunc(format, isLetterOrDigit) == "" { return false } a.format = format } return true } func (a *timeArshaler) isNumeric() bool { return int(a.base) > 0 } func (a *timeArshaler) hasCustomFormat() bool { return a.base == math.MaxUint } func (a *timeArshaler) appendMarshal(b []byte) ([]byte, error) { switch a.base { case 0: format := cmp.Or(a.format, time.RFC3339Nano) n0 := len(b) b = a.tt.AppendFormat(b, format) // Not all Go timestamps can be represented as valid RFC 3339. // Explicitly check for these edge cases. // See https://go.dev/issue/4556 and https://go.dev/issue/54580. switch b := b[n0:]; { case b[len("9999")] != '-': // year must be exactly 4 digits wide return b, errors.New("year outside of range [0,9999]") case b[len(b)-1] != 'Z': c := b[len(b)-len("Z07:00")] if ('0' <= c && c <= '9') || parseDec2(b[len(b)-len("07:00"):]) >= 24 { return b, errors.New("timezone hour outside of range [0,23]") } } return b, nil case math.MaxUint: return a.tt.AppendFormat(b, a.format), nil default: return appendTimeUnix(b, a.tt, a.base), nil } } func (a *timeArshaler) unmarshal(b []byte) (err error) { switch a.base { case 0: // Use time.Time.UnmarshalText to avoid possible string allocation. if err := a.tt.UnmarshalText(b); err != nil { return err } // TODO(https://go.dev/issue/57912): // RFC 3339 specifies the grammar for a valid timestamp. // However, the parsing functionality in "time" is too loose and // incorrectly accepts invalid timestamps as valid. // Remove these manual checks when "time" checks it for us. newParseError := func(layout, value, layoutElem, valueElem, message string) error { return &time.ParseError{Layout: layout, Value: value, LayoutElem: layoutElem, ValueElem: valueElem, Message: message} } switch { case a.looseRFC3339: return nil case b[len("2006-01-02T")+1] == ':': // hour must be two digits return newParseError(time.RFC3339, string(b), "15", string(b[len("2006-01-02T"):][:1]), "") case b[len("2006-01-02T15:04:05")] == ',': // sub-second separator must be a period return newParseError(time.RFC3339, string(b), ".", ",", "") case b[len(b)-1] != 'Z': switch { case parseDec2(b[len(b)-len("07:00"):]) >= 24: // timezone hour must be in range return newParseError(time.RFC3339, string(b), "Z07:00", string(b[len(b)-len("Z07:00"):]), ": timezone hour out of range") case parseDec2(b[len(b)-len("00"):]) >= 60: // timezone minute must be in range return newParseError(time.RFC3339, string(b), "Z07:00", string(b[len(b)-len("Z07:00"):]), ": timezone minute out of range") } } return nil case math.MaxUint: a.tt, err = time.Parse(a.format, string(b)) return err default: a.tt, err = parseTimeUnix(b, a.base) return err } } // appendDurationBase10 appends d formatted as a decimal fractional number, // where pow10 is a power-of-10 used to scale down the number. func appendDurationBase10(b []byte, d time.Duration, pow10 uint64) []byte { b, n := mayAppendDurationSign(b, d) // append sign whole, frac := bits.Div64(0, n, uint64(pow10)) // compute whole and frac fields b = strconv.AppendUint(b, whole, 10) // append whole field return appendFracBase10(b, frac, pow10) // append frac field } // parseDurationBase10 parses d from a decimal fractional number, // where pow10 is a power-of-10 used to scale up the number. func parseDurationBase10(b []byte, pow10 uint64) (time.Duration, error) { suffix, neg := consumeSign(b, false) // consume sign wholeBytes, fracBytes := bytesCutByte(suffix, '.', true) // consume whole and frac fields whole, okWhole := jsonwire.ParseUint(wholeBytes) // parse whole field; may overflow frac, okFrac := parseFracBase10(fracBytes, pow10) // parse frac field hi, lo := bits.Mul64(whole, uint64(pow10)) // overflow if hi > 0 sum, co := bits.Add64(lo, uint64(frac), 0) // overflow if co > 0 switch d := mayApplyDurationSign(sum, neg); { // overflow if neg != (d < 0) case (!okWhole && whole != math.MaxUint64) || !okFrac: return 0, fmt.Errorf("invalid duration %q: %w", b, strconv.ErrSyntax) case !okWhole || hi > 0 || co > 0 || neg != (d < 0): return 0, fmt.Errorf("invalid duration %q: %w", b, strconv.ErrRange) default: return d, nil } } // appendDurationISO8601 appends an ISO 8601 duration with a restricted grammar, // where leading and trailing zeroes and zero-value designators are omitted. // It only uses hour, minute, and second designators since ISO 8601 defines // those as being "accurate", while year, month, week, and day are "nominal". func appendDurationISO8601(b []byte, d time.Duration) []byte { if d == 0 { return append(b, "PT0S"...) } b, n := mayAppendDurationSign(b, d) b = append(b, "PT"...) n, nsec := bits.Div64(0, n, 1e9) // compute nsec field n, sec := bits.Div64(0, n, 60) // compute sec field hour, min := bits.Div64(0, n, 60) // compute hour and min fields if hour > 0 { b = append(strconv.AppendUint(b, hour, 10), 'H') } if min > 0 { b = append(strconv.AppendUint(b, min, 10), 'M') } if sec > 0 || nsec > 0 { b = append(appendFracBase10(strconv.AppendUint(b, sec, 10), nsec, 1e9), 'S') } return b } // daysPerYear is the exact average number of days in a year according to // the Gregorian calender, which has an extra day each year that is // a multiple of 4, unless it is evenly divisible by 100 but not by 400. // This does not take into account leap seconds, which are not deterministic. const daysPerYear = 365.2425 var errInaccurateDateUnits = errors.New("inaccurate year, month, week, or day units") // parseDurationISO8601 parses a duration according to ISO 8601-1:2019, // section 5.5.2.2 and 5.5.2.3 with the following restrictions or extensions: // // - A leading minus sign is permitted for negative duration according // to ISO 8601-2:2019, section 4.4.1.9. We do not permit negative values // for each "time scale component", which is permitted by section 4.4.1.1, // but rarely supported by parsers. // // - A leading plus sign is permitted (and ignored). // This is not required by ISO 8601, but not forbidden either. // There is some precedent for this as it is supported by the principle of // duration arithmetic as specified in ISO 8601-2-2019, section 14.1. // Of note, the JavaScript grammar for ISO 8601 permits a leading plus sign. // // - A fractional value is only permitted for accurate units // (i.e., hour, minute, and seconds) in the last time component, // which is permissible by ISO 8601-1:2019, section 5.5.2.3. // // - Both periods ('.') and commas (',') are supported as the separator // between the integer part and fraction part of a number, // as specified in ISO 8601-1:2019, section 3.2.6. // While ISO 8601 recommends comma as the default separator, // most formatters uses a period. // // - Leading zeros are ignored. This is not required by ISO 8601, // but also not forbidden by the standard. Many parsers support this. // // - Lowercase designators are supported. This is not required by ISO 8601, // but also not forbidden by the standard. Many parsers support this. // // If the nominal units of year, month, week, or day are present, // this produces a best-effort value and also reports [errInaccurateDateUnits]. // // The accepted grammar is identical to JavaScript's Duration: // // https://tc39.es/proposal-temporal/#prod-Duration // // We follow JavaScript's grammar as JSON itself is derived from JavaScript. // The Temporal.Duration.toJSON method is guaranteed to produce an output // that can be parsed by this function so long as arithmetic in JavaScript // do not use a largestUnit value higher than "hours" (which is the default). // Even if it does, this will do a best-effort parsing with inaccurate units, // but report [errInaccurateDateUnits]. func parseDurationISO8601(b []byte) (time.Duration, error) { var invalid, overflow, inaccurate, sawFrac bool var sumNanos, n, co uint64 // cutBytes is like [bytes.Cut], but uses either c0 or c1 as the separator. cutBytes := func(b []byte, c0, c1 byte) (prefix, suffix []byte, ok bool) { for i, c := range b { if c == c0 || c == c1 { return b[:i], b[i+1:], true } } return b, nil, false } // mayParseUnit attempts to parse another date or time number // identified by the desHi and desLo unit characters. // If the part is absent for current unit, it returns b as is. mayParseUnit := func(b []byte, desHi, desLo byte, unit time.Duration) []byte { number, suffix, ok := cutBytes(b, desHi, desLo) if !ok || sawFrac { return b // designator is not present or already saw fraction, which can only be in the last component } // Parse the number. // A fraction allowed for the accurate units in the last part. whole, frac, ok := cutBytes(number, '.', ',') if ok { sawFrac = true invalid = invalid || len(frac) == len("") || unit > time.Hour if unit == time.Second { n, ok = parsePaddedBase10(frac, uint64(time.Second)) invalid = invalid || !ok } else { f, err := strconv.ParseFloat("0."+string(frac), 64) invalid = invalid || err != nil || len(bytes.Trim(frac[len("."):], "0123456789")) > 0 n = uint64(math.Round(f * float64(unit))) // never overflows since f is within [0..1] } sumNanos, co = bits.Add64(sumNanos, n, 0) // overflow if co > 0 overflow = overflow || co > 0 } for len(whole) > 1 && whole[0] == '0' { whole = whole[len("0"):] // trim leading zeros } n, ok := jsonwire.ParseUint(whole) // overflow if !ok && MaxUint64 hi, lo := bits.Mul64(n, uint64(unit)) // overflow if hi > 0 sumNanos, co = bits.Add64(sumNanos, lo, 0) // overflow if co > 0 invalid = invalid || (!ok && n != math.MaxUint64) overflow = overflow || (!ok && n == math.MaxUint64) || hi > 0 || co > 0 inaccurate = inaccurate || unit > time.Hour return suffix } suffix, neg := consumeSign(b, true) prefix, suffix, okP := cutBytes(suffix, 'P', 'p') durDate, durTime, okT := cutBytes(suffix, 'T', 't') invalid = invalid || len(prefix) > 0 || !okP || (okT && len(durTime) == 0) || len(durDate)+len(durTime) == 0 if len(durDate) > 0 { // nominal portion of the duration durDate = mayParseUnit(durDate, 'Y', 'y', time.Duration(daysPerYear*24*60*60*1e9)) durDate = mayParseUnit(durDate, 'M', 'm', time.Duration(daysPerYear/12*24*60*60*1e9)) durDate = mayParseUnit(durDate, 'W', 'w', time.Duration(7*24*60*60*1e9)) durDate = mayParseUnit(durDate, 'D', 'd', time.Duration(24*60*60*1e9)) invalid = invalid || len(durDate) > 0 // unknown elements } if len(durTime) > 0 { // accurate portion of the duration durTime = mayParseUnit(durTime, 'H', 'h', time.Duration(60*60*1e9)) durTime = mayParseUnit(durTime, 'M', 'm', time.Duration(60*1e9)) durTime = mayParseUnit(durTime, 'S', 's', time.Duration(1e9)) invalid = invalid || len(durTime) > 0 // unknown elements } d := mayApplyDurationSign(sumNanos, neg) overflow = overflow || (neg != (d < 0) && d != 0) // overflows signed duration switch { case invalid: return 0, fmt.Errorf("invalid ISO 8601 duration %q: %w", b, strconv.ErrSyntax) case overflow: return 0, fmt.Errorf("invalid ISO 8601 duration %q: %w", b, strconv.ErrRange) case inaccurate: return d, fmt.Errorf("invalid ISO 8601 duration %q: %w", b, errInaccurateDateUnits) default: return d, nil } } // mayAppendDurationSign appends a negative sign if n is negative. func mayAppendDurationSign(b []byte, d time.Duration) ([]byte, uint64) { if d < 0 { b = append(b, '-') d *= -1 } return b, uint64(d) } // mayApplyDurationSign inverts n if neg is specified. func mayApplyDurationSign(n uint64, neg bool) time.Duration { if neg { return -1 * time.Duration(n) } else { return +1 * time.Duration(n) } } // appendTimeUnix appends t formatted as a decimal fractional number, // where pow10 is a power-of-10 used to scale up the number. func appendTimeUnix(b []byte, t time.Time, pow10 uint64) []byte { sec, nsec := t.Unix(), int64(t.Nanosecond()) if sec < 0 { b = append(b, '-') sec, nsec = negateSecNano(sec, nsec) } switch { case pow10 == 1e0: // fast case where units is in seconds b = strconv.AppendUint(b, uint64(sec), 10) return appendFracBase10(b, uint64(nsec), 1e9) case uint64(sec) < 1e9: // intermediate case where units is not seconds, but no overflow b = strconv.AppendUint(b, uint64(sec)*uint64(pow10)+uint64(uint64(nsec)/(1e9/pow10)), 10) return appendFracBase10(b, (uint64(nsec)*pow10)%1e9, 1e9) default: // slow case where units is not seconds and overflow would occur b = strconv.AppendUint(b, uint64(sec), 10) b = appendPaddedBase10(b, uint64(nsec)/(1e9/pow10), pow10) return appendFracBase10(b, (uint64(nsec)*pow10)%1e9, 1e9) } } // parseTimeUnix parses t formatted as a decimal fractional number, // where pow10 is a power-of-10 used to scale down the number. func parseTimeUnix(b []byte, pow10 uint64) (time.Time, error) { suffix, neg := consumeSign(b, false) // consume sign wholeBytes, fracBytes := bytesCutByte(suffix, '.', true) // consume whole and frac fields whole, okWhole := jsonwire.ParseUint(wholeBytes) // parse whole field; may overflow frac, okFrac := parseFracBase10(fracBytes, 1e9/pow10) // parse frac field var sec, nsec int64 switch { case pow10 == 1e0: // fast case where units is in seconds sec = int64(whole) // check overflow later after negation nsec = int64(frac) // cannot overflow case okWhole: // intermediate case where units is not seconds, but no overflow sec = int64(whole / pow10) // check overflow later after negation nsec = int64((whole%pow10)*(1e9/pow10) + frac) // cannot overflow case !okWhole && whole == math.MaxUint64: // slow case where units is not seconds and overflow occurred width := int(math.Log10(float64(pow10))) // compute len(strconv.Itoa(pow10-1)) whole, okWhole = jsonwire.ParseUint(wholeBytes[:len(wholeBytes)-width]) // parse the upper whole field mid, _ := parsePaddedBase10(wholeBytes[len(wholeBytes)-width:], pow10) // parse the lower whole field sec = int64(whole) // check overflow later after negation nsec = int64(mid*(1e9/pow10) + frac) // cannot overflow } if neg { sec, nsec = negateSecNano(sec, nsec) } switch t := time.Unix(sec, nsec).UTC(); { case (!okWhole && whole != math.MaxUint64) || !okFrac: return time.Time{}, fmt.Errorf("invalid time %q: %w", b, strconv.ErrSyntax) case !okWhole || neg != (t.Unix() < 0): return time.Time{}, fmt.Errorf("invalid time %q: %w", b, strconv.ErrRange) default: return t, nil } } // negateSecNano negates a Unix timestamp, where nsec must be within [0, 1e9). func negateSecNano(sec, nsec int64) (int64, int64) { sec = ^sec // twos-complement negation (i.e., -1*sec + 1) nsec = -nsec + 1e9 // negate nsec and add 1e9 (which is the extra +1 from sec negation) sec += int64(nsec / 1e9) // handle possible overflow of nsec if it started as zero nsec %= 1e9 // ensure nsec stays within [0, 1e9) return sec, nsec } // appendFracBase10 appends the fraction of n/max10, // where max10 is a power-of-10 that is larger than n. func appendFracBase10(b []byte, n, max10 uint64) []byte { if n == 0 { return b } return bytes.TrimRight(appendPaddedBase10(append(b, '.'), n, max10), "0") } // parseFracBase10 parses the fraction of n/max10, // where max10 is a power-of-10 that is larger than n. func parseFracBase10(b []byte, max10 uint64) (n uint64, ok bool) { switch { case len(b) == 0: return 0, true case len(b) < len(".0") || b[0] != '.': return 0, false } return parsePaddedBase10(b[len("."):], max10) } // appendPaddedBase10 appends a zero-padded encoding of n, // where max10 is a power-of-10 that is larger than n. func appendPaddedBase10(b []byte, n, max10 uint64) []byte { if n < max10/10 { // Formatting of n is shorter than log10(max10), // so add max10/10 to ensure the length is equal to log10(max10). i := len(b) b = strconv.AppendUint(b, n+max10/10, 10) b[i]-- // subtract the addition of max10/10 return b } return strconv.AppendUint(b, n, 10) } // parsePaddedBase10 parses b as the zero-padded encoding of n, // where max10 is a power-of-10 that is larger than n. // Truncated suffix is treated as implicit zeros. // Extended suffix is ignored, but verified to contain only digits. func parsePaddedBase10(b []byte, max10 uint64) (n uint64, ok bool) { pow10 := uint64(1) for pow10 < max10 { n *= 10 if len(b) > 0 { if b[0] < '0' || '9' < b[0] { return n, false } n += uint64(b[0] - '0') b = b[1:] } pow10 *= 10 } if len(b) > 0 && len(bytes.TrimRight(b, "0123456789")) > 0 { return n, false // trailing characters are not digits } return n, true } // consumeSign consumes an optional leading negative or positive sign. func consumeSign(b []byte, allowPlus bool) ([]byte, bool) { if len(b) > 0 { if b[0] == '-' { return b[len("-"):], true } else if b[0] == '+' && allowPlus { return b[len("+"):], false } } return b, false } // bytesCutByte is similar to bytes.Cut(b, []byte{c}), // except c may optionally be included as part of the suffix. func bytesCutByte(b []byte, c byte, include bool) ([]byte, []byte) { if i := bytes.IndexByte(b, c); i >= 0 { if include { return b[:i], b[i:] } return b[:i], b[i+1:] } return b, nil } // parseDec2 parses b as an unsigned, base-10, 2-digit number. // The result is undefined if digits are not base-10. func parseDec2(b []byte) byte { if len(b) < 2 { return 0 } return 10*(b[0]-'0') + (b[1] - '0') }