gstr_domain.go 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. // Copyright 2019 gf Author(https://github.com/gogf/gf). All Rights Reserved.
  2. //
  3. // This Source Code Form is subject to the terms of the MIT License.
  4. // If a copy of the MIT was not distributed with this file,
  5. // You can obtain one at https://github.com/gogf/gf.
  6. package gstr
  7. import "strings"
  8. // IsSubDomain checks whether <subDomain> is sub-domain of mainDomain.
  9. // It supports '*' in <mainDomain>.
  10. func IsSubDomain(subDomain string, mainDomain string) bool {
  11. if p := strings.IndexByte(subDomain, ':'); p != -1 {
  12. subDomain = subDomain[0:p]
  13. }
  14. if p := strings.IndexByte(mainDomain, ':'); p != -1 {
  15. mainDomain = mainDomain[0:p]
  16. }
  17. subArray := strings.Split(subDomain, ".")
  18. mainArray := strings.Split(mainDomain, ".")
  19. subLength := len(subArray)
  20. mainLength := len(mainArray)
  21. // Eg:
  22. // "s.s.goframe.org" is not sub-domain of "*.goframe.org"
  23. // but
  24. // "s.s.goframe.org" is not sub-domain of "goframe.org"
  25. if mainLength > 2 && subLength > mainLength {
  26. return false
  27. }
  28. minLength := subLength
  29. if mainLength < minLength {
  30. minLength = mainLength
  31. }
  32. for i := minLength; i > 0; i-- {
  33. if mainArray[mainLength-i] == "*" {
  34. continue
  35. }
  36. if mainArray[mainLength-i] != subArray[subLength-i] {
  37. return false
  38. }
  39. }
  40. return true
  41. }