[go] Go 언어를 활용한 웹 소켓 보안

07 Dec 2023

소개

웹 소켓은 양방향 통신을 지원하는 프로토콜로, 실시간으로 데이터를 주고받을 수 있습니다. 웹 소켓을 사용하면 서버와 클라이언트 간의 효율적인 통신이 가능해지지만, 보안에 대한 고려가 필요합니다. 이 글에서는 Go 언어를 활용하여 웹 소켓 통신을 안전하게 만드는 방법에 대해 알아보겠습니다.

TLS 적용

TLS (Transport Layer Security)은 데이터 통신을 암호화하는 프로토콜입니다. 웹 소켓 통신에서 TLS를 적용하면 데이터가 암호화되어 전송되므로, 중간에 누군가가 데이터를 엿듣는 것을 방지할 수 있습니다.

package main

import (
    "crypto/tls"
    "net/http"

    "github.com/gorilla/websocket"
)

func main() {
    http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
        // 웹 소켓 연결을 위한 업그레이드
        upgrader := websocket.Upgrader{
            ReadBufferSize:  1024,
            WriteBufferSize: 1024,
            // TLS 사용 설정
            CheckOrigin: func(r *http.Request) bool { return true },
        }
        conn, err := upgrader.Upgrade(w, r, nil)
        if err != nil {
            panic(err)
        }
        defer conn.Close()

        // 웹 소켓 통신 시작
        for {
            // 메시지 읽기
            _, msg, err := conn.ReadMessage()
            if err != nil {
                break
            }
            // 메시지 처리
            // ...

            // 응답 메시지 쓰기
            err = conn.WriteMessage(websocket.TextMessage, []byte("Received message"))
            if err != nil {
                break
            }
        }
    })

    // TLS 설정
    tlsConfig := &tls.Config{
        // 인증서와 개인 키 로드
        CertFile: "server.crt",
        KeyFile:  "server.key",
    }

    // HTTPS 서버 시작
    err := http.ListenAndServeTLS(":8080", "server.crt", "server.key", nil)
    if err != nil {
        panic(err)
    }
}

위의 예제 코드에서는 net/http 패키지와 github.com/gorilla/websocket 패키지를 사용하여 TLS를 적용한 웹 소켓 서버를 만들고 있습니다.

TLS를 적용하기 위해 http.ListenAndServeTLS 함수를 사용하고, tls.Config를 설정하여 인증서와 개인 키를 로드합니다.

인증서 발급

TLS를 적용하기 위해서는 인증서가 필요합니다. 인증서는 공인된 인증 기관에서 발급 받을 수도 있지만, 개발 환경에서는 자체 서명된 인증서를 생성하여 사용할 수도 있습니다.

자체 서명된 인증서를 생성하기 위해 다음과 같은 명령을 실행합니다:

$ openssl req -x509 -newkey rsa:2048 -nodes -keyout server.key -out server.crt -days 365

위 명령을 실행하면 개인 키(server.key)와 인증서(server.crt) 파일이 생성됩니다. 이 파일들을 TLS 설정에 사용하면 됩니다.

요약

Go 언어를 사용하여 웹 소켓 통신에 TLS 보안을 적용하는 방법에 대해 알아보았습니다. TLS를 사용하면 웹 소켓 통신 중간에 데이터가 탈취되는 것을 방지할 수 있습니다. TLS 설정을 통해 인증서와 개인 키를 로드하여 안전한 웹 소켓 서버를 구축할 수 있습니다.

더 자세한 내용은 다음 참고 자료를 참고하시기 바랍니다: