golang检测上传文件类型

Golang是一种被广泛应用于网络编程和后端开发的编程语言。在很多Web应用中，用户经常需要上传文件，这就涉及到了文件上传安全性的问题。对于一个可靠的文件上传功能，我们需要确保用户只能上传特定类型的文件。在本文中，我们将使用Golang来实现一个文件上传功能，并且通过文件头检测的方式来验证文件的类型。

文件上传功能的实现

在Golang中，我们可以使用net/http包来处理HTTP请求和响应。首先，我们需要创建一个HTTP服务器来接收文件上传请求。可以通过下面的代码来实现：

```go package main import ( "fmt" "net/http" ) func main() { http.HandleFunc("/upload", uploadHandler) err := http.ListenAndServe(":8080", nil) if err != nil { fmt.Println("Server error:", err) } } func uploadHandler(w http.ResponseWriter, r *http.Request) { if r.Method == "POST" { file, header, err := r.FormFile("file") if err != nil { http.Error(w, err.Error(), http.StatusInternalServerError) return } defer file.Close() // 继续处理文件上传逻辑 } else { http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed) return } } ```

文件头检测

在文件上传中，我们可以通过读取文件的前几个字节来判断文件的类型。不同类型的文件有不同的文件头信息，我们可以通过比较文件头和已知文件类型的文件头来判断上传文件的类型。下面是一个简单的文件头检测函数的实现：

```go package main import ( "fmt" "io" "net/http" "os" ) func main() { http.HandleFunc("/upload", uploadHandler) err := http.ListenAndServe(":8080", nil) if err != nil { fmt.Println("Server error:", err) } } func uploadHandler(w http.ResponseWriter, r *http.Request) { if r.Method == "POST" { file, header, err := r.FormFile("file") if err != nil { http.Error(w, err.Error(), http.StatusInternalServerError) return } defer file.Close() buf := make([]byte, 512) _, err = file.Read(buf) if err != nil { http.Error(w, err.Error(), http.StatusInternalServerError) return } fileType := http.DetectContentType(buf) if fileType != "image/jpeg" && fileType != "image/png" { http.Error(w, "Invalid file type", http.StatusBadRequest) return } // 保存文件到服务器 dst, err := os.Create(header.Filename) if err != nil { http.Error(w, err.Error(), http.StatusInternalServerError) return } defer dst.Close() _, err = io.Copy(dst, file) if err != nil { http.Error(w, err.Error(), http.StatusInternalServerError) return } w.WriteHeader(http.StatusOK) fmt.Fprintf(w, "File uploaded successfully") } else { http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed) return } } ```

简单的文件类型判断

在上面的示例中，我们只检测了两种文件类型：JPEG和PNG。实际应用中，可能还会需要检测更多类型的文件，因此我们可以扩展上面的代码来支持更多类型的文件。常见的文件类型通常与文件的后缀名相关联，我们可以通过文件的后缀名来判断文件的类型。

```go package main import ( "fmt" "io" "net/http" "os" "path/filepath" "strings" ) func main() { http.HandleFunc("/upload", uploadHandler) err := http.ListenAndServe(":8080", nil) if err != nil { fmt.Println("Server error:", err) } } func getExtension(filename string) string { ext := filepath.Ext(filename) if ext != "" { return ext[1:] } return "" } func uploadHandler(w http.ResponseWriter, r *http.Request) { if r.Method == "POST" { file, header, err := r.FormFile("file") if err != nil { http.Error(w, err.Error(), http.StatusInternalServerError) return } defer file.Close() buf := make([]byte, 512) _, err = file.Read(buf) if err != nil { http.Error(w, err.Error(), http.StatusInternalServerError) return } fileType := http.DetectContentType(buf) if fileType != "image/jpeg" && fileType != "image/png" { ext := getExtension(header.Filename) if ext != "jpg" && ext != "png" { http.Error(w, "Invalid file type", http.StatusBadRequest) } } // 保存文件到服务器 dst, err := os.Create(header.Filename) if err != nil { http.Error(w, err.Error(), http.StatusInternalServerError) return } defer dst.Close() _, err = io.Copy(dst, file) if err != nil { http.Error(w, err.Error(), http.StatusInternalServerError) return } w.WriteHeader(http.StatusOK) fmt.Fprintf(w, "File uploaded successfully") } else { http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed) return } } ``` 通过添加getExtension函数，我们可以获取文件的后缀名，并与已知的文件类型进行比较来判断上传文件的类型。