mirror of
https://github.com/tailscale/tailscale.git
synced 2026-10-10 12:21:53 -04:00
Compare parsed each numeric field with strconv.ParseUint and panicked if that failed, so a version with a field of 20 or more digits, such as a long build stamp, crashed the caller. Compare digit runs by length after trimming leading zeros, then lexically. This gives the same results as before for numbers that fit in a uint64 and works for numbers of any length. Fixes #21532 Signed-off-by: Raphael Fakhri <153192858+RaphaelFakhri@users.noreply.github.com>
103 lines
2.9 KiB
Go
103 lines
2.9 KiB
Go
// Copyright (c) Tailscale Inc & contributors
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
// Package cmpver implements a variant of debian version number
|
|
// comparison.
|
|
//
|
|
// A version is a string consisting of alternating non-numeric and
|
|
// numeric fields. When comparing two versions, each one is broken
|
|
// down into its respective fields, and the fields are compared
|
|
// pairwise. The comparison is lexicographic for non-numeric fields,
|
|
// numeric for numeric fields. The first non-equal field pair
|
|
// determines the ordering of the two versions.
|
|
//
|
|
// This comparison scheme is a simplified version of Debian's version
|
|
// number comparisons. Debian differs in a few details of
|
|
// lexicographical field comparison, where certain characters have
|
|
// special meaning and ordering. We don't need that, because Tailscale
|
|
// version numbers don't need it.
|
|
package cmpver
|
|
|
|
import "strings"
|
|
|
|
// Less reports whether v1 is less than v2.
|
|
//
|
|
// Note that "12" is less than "12.0".
|
|
func Less(v1, v2 string) bool {
|
|
return Compare(v1, v2) < 0
|
|
}
|
|
|
|
// LessEq reports whether v1 is less than or equal to v2.
|
|
//
|
|
// Note that "12" is less than "12.0".
|
|
func LessEq(v1, v2 string) bool {
|
|
return Compare(v1, v2) <= 0
|
|
}
|
|
|
|
func isnum(r rune) bool {
|
|
return r >= '0' && r <= '9'
|
|
}
|
|
|
|
func notnum(r rune) bool {
|
|
return !isnum(r)
|
|
}
|
|
|
|
// Compare returns an integer comparing two strings as version numbers.
|
|
// The result will be -1, 0, or 1 representing the sign of v1 - v2:
|
|
//
|
|
// Compare(v1, v2) < 0 if v1 < v2
|
|
// == 0 if v1 == v2
|
|
// > 0 if v1 > v2
|
|
func Compare(v1, v2 string) int {
|
|
var f1, f2 string
|
|
for v1 != "" || v2 != "" {
|
|
// Compare the non-numeric character run lexicographically.
|
|
f1, v1 = splitPrefixFunc(v1, notnum)
|
|
f2, v2 = splitPrefixFunc(v2, notnum)
|
|
|
|
if res := strings.Compare(f1, f2); res != 0 {
|
|
return res
|
|
}
|
|
|
|
// Compare the numeric character run numerically.
|
|
f1, v1 = splitPrefixFunc(v1, isnum)
|
|
f2, v2 = splitPrefixFunc(v2, isnum)
|
|
|
|
// Compare the digit runs numerically. Empty strings mean zero,
|
|
// which follows the Debian convention because empirically that
|
|
// produces reasonable-feeling comparison behavior.
|
|
if res := compareDigits(f1, f2); res != 0 {
|
|
return res
|
|
}
|
|
}
|
|
|
|
// Only way to reach here is if v1 and v2 run out of fields
|
|
// simultaneously - i.e. exactly equal versions.
|
|
return 0
|
|
}
|
|
|
|
// compareDigits compares two strings of ASCII digits by numeric value and
|
|
// returns -1, 0 or 1. Unlike parsing the strings as integers, it has no
|
|
// limit on the length of the numbers. An empty string is treated as zero.
|
|
func compareDigits(a, b string) int {
|
|
a = strings.TrimLeft(a, "0")
|
|
b = strings.TrimLeft(b, "0")
|
|
if len(a) != len(b) {
|
|
if len(a) < len(b) {
|
|
return -1
|
|
}
|
|
return 1
|
|
}
|
|
return strings.Compare(a, b)
|
|
}
|
|
|
|
// splitPrefixFunc splits s at the first rune where f(rune) is false.
|
|
func splitPrefixFunc(s string, f func(rune) bool) (string, string) {
|
|
for i, r := range s {
|
|
if !f(r) {
|
|
return s[:i], s[i:]
|
|
}
|
|
}
|
|
return s, s[:0]
|
|
}
|