Go-Excelize API源码阅读三十二—— UnprotectSheet

Go-Excelize API源码阅读（三十二）——UnprotectSheet

开源摘星计划（WeOpen Star）是由腾源会 2022 年推出的全新项目，旨在为开源人提供成长激励，为开源项目提供成长支持，助力开发者更好地了解开源，更快地跨越鸿沟，参与到开源的具体贡献与实践中。

不管你是开源萌新，还是希望更深度参与开源贡献的老兵，跟随“开源摘星计划”开启你的开源之旅，从一篇学习笔记、到一段代码的提交，不断挖掘自己的潜能，最终成长为开源社区的“闪亮之星”。

我们将同你一起，探索更多的可能性！

项目地址： WeOpen-Star：https://github.com/weopenprojects/WeOpen-Star

一、Go-Excelize简介

Excelize 是 Go 语言编写的用于操作 Office Excel 文档基础库，基于 ECMA-376，ISO/IEC 29500 国际标准。可以使用它来读取、写入由 Microsoft Excel™ 2007 及以上版本创建的电子表格文档。支持 XLAM / XLSM / XLSX / XLTM / XLTX 等多种文档格式，高度兼容带有样式、图片(表)、透视表、切片器等复杂组件的文档，并提供流式读写 API，用于处理包含大规模数据的工作簿。可应用于各类报表平台、云计算、边缘计算等系统。使用本类库要求使用的 Go 语言为 1.15 或更高版本。

二、UnprotectSheet

func (f *File) UnprotectSheet(sheet string, password ...string) error

该根据给定的工作表名称取消保护该工作表，指定第二个可选密码参数以通过密码验证来取消工作表保护。先来看一看代码：

// UnprotectSheet provides a function to remove protection for a sheet,
// specified the second optional password parameter to remove sheet
// protection with password verification.
func (f *File) UnprotectSheet(sheet string, password ...string) error {
	ws, err := f.workSheetReader(sheet)
	if err != nil {
		return err
	}
	// password verification
	if len(password) > 0 {
		if ws.SheetProtection == nil {
			return ErrUnprotectSheet
		}
		if ws.SheetProtection.AlgorithmName == "" && ws.SheetProtection.Password != genSheetPasswd(password[0]) {
			return ErrUnprotectSheetPassword
		}
		if ws.SheetProtection.AlgorithmName != "" {
			// check with given salt value
			hashValue, _, err := genISOPasswdHash(password[0], ws.SheetProtection.AlgorithmName, ws.SheetProtection.SaltValue, ws.SheetProtection.SpinCount)
			if err != nil {
				return err
			}
			if ws.SheetProtection.HashValue != hashValue {
				return ErrUnprotectSheetPassword
			}
		}
	}
	ws.SheetProtection = nil
	return err
}

	ws, err := f.workSheetReader(sheet)
	if err != nil {
		return err
	}

先打开工作表，如果有err，就返回err。

接下来的工作主要进行密码验证：

	// password verification
	if len(password) > 0 {
		if ws.SheetProtection == nil {
			return ErrUnprotectSheet
		}
		if ws.SheetProtection.AlgorithmName == "" && ws.SheetProtection.Password != genSheetPasswd(password[0]) {
			return ErrUnprotectSheetPassword
		}
		if ws.SheetProtection.AlgorithmName != "" {
			// check with given salt value
			hashValue, _, err := genISOPasswdHash(password[0], ws.SheetProtection.AlgorithmName, ws.SheetProtection.SaltValue, ws.SheetProtection.SpinCount)
			if err != nil {
				return err
			}
			if ws.SheetProtection.HashValue != hashValue {
				return ErrUnprotectSheetPassword
			}
		}
	}
	ws.SheetProtection = nil
	return err

	// protection.
	ErrUnprotectSheet = errors.New("worksheet has set no protect")

如果ws.SheetProtection为nil，就会报错：工作表没有被保护。自然也不能调用这个API取消保护了。

if ws.SheetProtection.AlgorithmName == "" && ws.SheetProtection.Password != genSheetPasswd(password[0]) {
	return ErrUnprotectSheetPassword
}

如果ws.SheetProtection.AlgorithmName，即工作表保护的加密算法名为空，并且存储的密码与输入的密码不匹配，就会报下面的错误：

	// ErrUnprotectSheetPassword defined the error message on remove sheet
	// protection with password verification failed.
	ErrUnprotectSheetPassword = errors.New("worksheet protect password not match")

if ws.SheetProtection.AlgorithmName != "" {
	// check with given salt value
	hashValue, _, err := genISOPasswdHash(password[0], ws.SheetProtection.AlgorithmName, ws.SheetProtection.SaltValue, ws.SheetProtection.SpinCount)
	if err != nil {
		return err
	}
	if ws.SheetProtection.HashValue != hashValue {
		return ErrUnprotectSheetPassword
	}
}

如果存在加密算法，表示密码被加密保存了，就调用genISOPasswdHash将输入的密码生成密文然后与加密存储的密码进行比对，如果比对成功，会执行下列命令：

ws.SheetProtection = nil

将工作表保护清空。否则返回ErrUnprotectSheetPassword错误。上面介绍了这个错误，此处不再讲述。

		"MD4":     "md4",
		"MD5":     "md5",
		"SHA-1":   "sha1",
		"SHA-256": "sha256",
		"SHA-384": "sha384",
		"SHA-512": "sha512",

需要注意的是此处的所谓“加密”，是指上述的哈希函数：MD4\MD5\SHA-1\SHA-256\SHA-384\SHA-512。

结语

这是Excelize API阅读的第三十二篇，鉴于本人技术水平有限，如有错误，烦请指出斧正！感谢！

原文地址：https://cloud.tencent.com/developer/article/2177698

Go-Excelize API源码阅读三十二—— UnprotectSheet