From ef422e69882f4c1ba0e04bc8019350e41fdabd59 Mon Sep 17 00:00:00 2001
From: Kristoffer Dalby <kradalby@kradalby.no>
Date: Wed, 2 Mar 2022 07:15:20 +0000
Subject: [PATCH] Protect against expiry nil

---
 api.go     | 1 +
 machine.go | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/api.go b/api.go
index 9b8fafd..3b3b675 100644
--- a/api.go
+++ b/api.go
@@ -144,6 +144,7 @@ func (h *Headscale) RegistrationHandler(ctx *gin.Context) {
 			Name:       req.Hostinfo.Hostname,
 			NodeKey:    NodePublicKeyStripPrefix(req.NodeKey),
 			LastSeen:   &now,
+			Expiry:     &time.Time{},
 		}
 
 		if !req.Expiry.IsZero() {
diff --git a/machine.go b/machine.go
index bd842ae..c9370c2 100644
--- a/machine.go
+++ b/machine.go
@@ -117,7 +117,7 @@ func (machine Machine) isExpired() bool {
 	// If Expiry is not set, the client has not indicated that
 	// it wants an expiry time, it is therefor considered
 	// to mean "not expired"
-	if machine.Expiry.IsZero() {
+	if machine.Expiry == nil || machine.Expiry.IsZero() {
 		return false
 	}