From 72d70d68dfcbdbe112d9f82ead6d987898417f95 Mon Sep 17 00:00:00 2001
From: Pablo Mazzini <pmazzini@gmail.com>
Date: Sun, 12 Aug 2018 16:17:11 +0200
Subject: fix OptClientArchType

---
 dhcpv4/option_archtype.go | 42 +++++-------------------------------------
 1 file changed, 5 insertions(+), 37 deletions(-)

(limited to 'dhcpv4/option_archtype.go')

diff --git a/dhcpv4/option_archtype.go b/dhcpv4/option_archtype.go
index 16ca98d..8eafc55 100644
--- a/dhcpv4/option_archtype.go
+++ b/dhcpv4/option_archtype.go
@@ -6,43 +6,14 @@ package dhcpv4
 import (
 	"encoding/binary"
 	"fmt"
-)
-
-//ArchType encodes an architecture type in an uint16
-type ArchType uint16
 
-// see rfc4578
-const (
-	INTEL_X86PC       ArchType = 0
-	NEC_PC98          ArchType = 1
-	EFI_ITANIUM       ArchType = 2
-	DEC_ALPHA         ArchType = 3
-	ARC_X86           ArchType = 4
-	INTEL_LEAN_CLIENT ArchType = 5
-	EFI_IA32          ArchType = 6
-	EFI_BC            ArchType = 7
-	EFI_XSCALE        ArchType = 8
-	EFI_X86_64        ArchType = 9
+	"github.com/insomniacslk/dhcp/iana"
 )
 
-// ArchTypeToStringMap maps an ArchType to a mnemonic name
-var ArchTypeToStringMap = map[ArchType]string{
-	INTEL_X86PC:       "Intel x86PC",
-	NEC_PC98:          "NEC/PC98",
-	EFI_ITANIUM:       "EFI Itanium",
-	DEC_ALPHA:         "DEC Alpha",
-	ARC_X86:           "Arc x86",
-	INTEL_LEAN_CLIENT: "Intel Lean Client",
-	EFI_IA32:          "EFI IA32",
-	EFI_BC:            "EFI BC",
-	EFI_XSCALE:        "EFI Xscale",
-	EFI_X86_64:        "EFI x86-64",
-}
-
 // OptClientArchType represents an option encapsulating the Client System
 // Architecture Type option Definition.
 type OptClientArchType struct {
-	ArchTypes []ArchType
+	ArchTypes []iana.ArchType
 }
 
 // Code returns the option code.
@@ -71,10 +42,7 @@ func (o *OptClientArchType) Length() int {
 func (o *OptClientArchType) String() string {
 	var archTypes string
 	for idx, at := range o.ArchTypes {
-		name, ok := ArchTypeToStringMap[at]
-		if !ok {
-			name = "Unknown"
-		}
+		name := iana.ArchTypeToString(at)
 		archTypes += name
 		if idx < len(o.ArchTypes)-1 {
 			archTypes += ", "
@@ -100,10 +68,10 @@ func ParseOptClientArchType(data []byte) (*OptClientArchType, error) {
 	if len(data) < 2+length {
 		return nil, ErrShortByteStream
 	}
-	archTypes := make([]ArchType, 0, length%2)
+	archTypes := make([]iana.ArchType, 0, length%2)
 	for idx := 0; idx < length; idx += 2 {
 		b := data[2+idx : 2+idx+2]
-		archTypes = append(archTypes, ArchType(binary.BigEndian.Uint16(b)))
+		archTypes = append(archTypes, iana.ArchType(binary.BigEndian.Uint16(b)))
 	}
 	return &OptClientArchType{ArchTypes: archTypes}, nil
 }
-- 
cgit v1.2.3