Go 언어에서 동시 파일의 암호화 및 복호화를 처리하는 방법은 무엇입니까?
소개:
인터넷이 발달하고 정보 전송이 대중화되면서 파일 암호화 및 복호화는 데이터 보안을 보호하는 중요한 수단이 되었습니다. 또한, 컴퓨터 처리 능력과 저장 용량이 증가함에 따라 여러 파일을 동시에 처리해야 하는 필요성도 커지고 있습니다. Go 언어에서는 동시성 기능을 사용하여 여러 파일의 암호화 및 암호 해독 작업을 동시에 구현할 수 있습니다.
package main import ( "crypto/aes" "crypto/cipher" "fmt" "io" "os" "path/filepath" "sync" ) // 加密文件 func encryptFile(inPath, outPath string, key []byte) error { inFile, err := os.Open(inPath) if err != nil { return err } defer inFile.Close() outFile, err := os.OpenFile(outPath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666) if err != nil { return err } defer outFile.Close() block, err := aes.NewCipher(key) if err != nil { return err } iv := make([]byte, aes.BlockSize) stream := cipher.NewCTR(block, iv) writer := &cipher.StreamWriter{S: stream, W: outFile} if _, err := io.Copy(writer, inFile); err != nil { return err } return nil } // 解密文件 func decryptFile(inPath, outPath string, key []byte) error { inFile, err := os.Open(inPath) if err != nil { return err } defer inFile.Close() outFile, err := os.OpenFile(outPath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0666) if err != nil { return err } defer outFile.Close() block, err := aes.NewCipher(key) if err != nil { return err } iv := make([]byte, aes.BlockSize) stream := cipher.NewCTR(block, iv) reader := &cipher.StreamReader{S: stream, R: inFile} if _, err := io.Copy(outFile, reader); err != nil { return err } return nil } func main() { var wg sync.WaitGroup key := []byte("YOUR_KEY") // 要加密的文件列表 files := []string{"file1.txt", "file2.txt", "file3.txt"} wg.Add(len(files)) for _, file := range files { go func(f string) { defer wg.Done() inPath := filepath.Join("input", f) outPath := filepath.Join("output", "encrypted_"+f) if err := encryptFile(inPath, outPath, key); err != nil { fmt.Printf("Error encrypting file %s: %s ", inPath, err.Error()) } }(file) } wg.Wait() // 要解密的文件列表 files = []string{"encrypted_file1.txt", "encrypted_file2.txt", "encrypted_file3.txt"} wg.Add(len(files)) for _, file := range files { go func(f string) { defer wg.Done() inPath := filepath.Join("output", f) outPath := filepath.Join("output", "decrypted_"+f[len("encrypted_"):]) if err := decryptFile(inPath, outPath, key); err != nil { fmt.Printf("Error decrypting file %s: %s ", inPath, err.Error()) } }(file) } wg.Wait() fmt.Println("All files encrypted and decrypted successfully.") }
위 코드는 파일 암호화 및 복호화를 동시에 처리하는 예를 구현합니다. 이 예에서는 먼저 모든 동시 작업이 완료되기를 기다리는 sync.WaitGroup
객체를 정의한 다음 암호화 및 복호화할 파일 목록을 지정하고 동시에 암호화 및 복호화 작업을 수행합니다. . sync.WaitGroup
对象,然后指定了要加密和解密的文件列表,并以并发方式进行加密和解密操作。
在encryptFile
和decryptFile
函数中,我们首先打开输入文件和输出文件,然后创建一个AES块,并使用16字节IV对其进行初始化。接下来,我们将生成的流与输入文件或输出文件进行关联,然后使用io.Copy
函数复制数据,完成文件加密和解密的操作。
最后,在main
函数中,我们使用filepath.Join
encryptFile
및 decryptFile
함수에서는 먼저 입력 파일과 출력 파일을 연 다음 AES 블록을 생성하고 16바이트 IV로 초기화합니다. 다음으로 생성된 스트림을 입력 파일 또는 출력 파일과 연결한 다음 io.Copy
함수를 사용하여 데이터를 복사하고 파일 암호화 및 암호 해독 작업을 완료합니다. main
함수에서 filepath.Join
함수를 사용하여 파일 경로를 구축하고 동시 암호화 및 암호 해독 작업을 시작합니다. 모든 작업이 완료될 때까지 기다렸다가 성공 메시지를 인쇄합니다. 위 내용은 Go 언어에서 동시 파일의 암호화 및 복호화를 처리하는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!