GoLang实现google authenticator的CLI工具

两步认证在很多验证中都要使用。如果在手机客户端上,如果使用电脑,每次都要拿出手机,手动输入。还要担心会过时。效率不是很高。

我现在的处理方式以下方式:

  • MAC上alfred workflow支持
  • Chrom扩展支持
  • 手机客户端
  • CLI记录工具

今天主要介绍CLI工具,我现在在golang,找一些练手的项目.

先上代码

package main

import (
    "fmt"
    "os"
    "log"
    "sort"
    "github.com/urfave/cli"
    "gopkg.in/ini.v1"
    "crypto/hmac"
    "crypto/sha1"
    "strings"
    "encoding/base32"
    "time"
    "github.com/atotto/clipboard"
    "strconv"
)


func toBytes(value int64) []byte {
    var result []byte
    mask := int64(0xFF)
    shifts := [8]uint16{56, 48, 40, 32, 24, 16, 8, 0}
    for _, shift := range shifts {
        result = append(result, byte((value>>shift)&mask))
    }
    return result
}

func toUint32(bytes []byte) uint32 {
    return (uint32(bytes[0]) << 24) + (uint32(bytes[1]) << 16) +
        (uint32(bytes[2]) << 8) + uint32(bytes[3])
}

func oneTimePassword(key []byte, value []byte) uint32 {
    // sign the value using HMAC-SHA1
    hmacSha1 := hmac.New(sha1.New, key)
    hmacSha1.Write(value)
    hash := hmacSha1.Sum(nil)
    
    offset := hash[len(hash)-1] & 0x0F

    // get a 32-bit (4-byte) chunk from the hash starting at offset
    hashParts := hash[offset : offset+4]

    // ignore the most significant bit as per RFC 4226
    hashParts[0] = hashParts[0] & 0x7F

    number := toUint32(hashParts)

    // size to 6 digits
    // one million is the first number with 7 digits so the remainder
    // of the division will always return < 7 digits
    pwd := number % 1000000

    return pwd
}



func Cli() {

    app := cli.NewApp()
    app.Name = "Google Authentiator CLI"
    app.Usage = "Create, List, Delete, Copy your GA"
    app.Version = "0.0.1"
    app.Commands = []cli.Command {
        {
            Name: "list",
            Aliases: []string{"l"},
            Usage: "list | l",
            Description: "list all item",
            Category: "Show",
            Action: func(c *cli.Context) error {
                list()
                return nil
            },
        },
        {
            Name: "show",
            Aliases: []string{"s"},
            Usage: "show | s github.com",
            Description: "show all item",
            Category: "Show",
            Action: func(c *cli.Context) error {
                item := c.Args().First()
                show(item)
                return nil
            },
        },
        {
            Name: "copy",
            Aliases: []string{"c"},
            Usage: "copy | c github.com",
            UsageText: "copy ga to Clipboard",
            Category: "Show",
            Action: func(c *cli.Context) error {
                item := c.Args().First()
                copy(item)
                return nil
            },
        },
        {
            Name: "create",
            Aliases: []string{"r"},
            Usage: "create| r github.com xxxxxxxxxx",
            UsageText: "create a new ga Secrete item",
            Category: "Store",
            Action: func(c *cli.Context) error {
                item := c.Args().First()
                secret := c.Args()[1]
                create(item, secret)
                return nil
            },
        },
        {
            Name: "modify",
            Aliases: []string{"m"},
            Usage: "modify | m github.com xxxxxxxxx",
            UsageText: "Modify One item Secrete",
            Category: "Store",
            Action: func(c *cli.Context) error {
                item := c.Args().First()
                secret := c.Args()[1]
                modify(item, secret)
                return nil
            },
        },
        {
            Name: "delete",
            Aliases: []string{"d"},
            Usage: "delete | d github.com",
            UsageText: "Delete One item Secrete",
            Category: "Store",
            Action: func(c *cli.Context) error {
                delete(c.Args().First())
                return nil
            },
        },

    }
    app.Action = func(c *cli.Context) error {
        fmt.Println("you can use list")
        return nil
    }

    sort.Sort(cli.FlagsByName(app.Flags))

    cli.HelpFlag = cli.BoolFlag {
        Name: "help, h",
        Usage: "Help!Help!",
    }

    cli.VersionFlag = cli.BoolFlag {
        Name: "print-version, v",
        Usage: "print version",
    }

    err := app.Run(os.Args)
    if err != nil {
        log.Fatal(err)
    }
}


func create(item, secret string) error {
    file,er:=os.Open("/tmp/myga.ini")

    if er != nil {
        file, _ = os.Create("/tmp/myga.ini")
    }
    file.Close()

    cfg, err := ini.Load("/tmp/myga.ini")
    if err != nil {
        panic(err)
    }

    if _, err := cfg.GetSection(item); err == nil {
        fmt.Println(item + " is Exist!!!!")
        os.Exit(1)
    }

    cfg.NewSection(item)
    cfg.Section(item).Key("secret").SetValue(secret)
    cfg.SaveTo("/tmp/myga.ini")
    return nil
}

func show(item string) error {
    file,er:=os.Open("/tmp/myga.ini")

    if er!=nil && os.IsNotExist(er){
        fmt.Println("no data!!!")
        os.Exit(1)
    }
    file.Close()

    cfg, err := ini.Load("/tmp/myga.ini")
    if err != nil {
        panic(err)
    }


    section, err := cfg.GetSection(item)
    if err != nil {
        panic(err)
    }

    key, err := section.GetKey("secret")

    if err != nil {
        panic(err)
    }

    inputNoSpaces := strings.Replace(key.String(), " ", "", -1)
    inputNoSpacesUpper := strings.ToUpper(inputNoSpaces)
    keyS, erro := base32.StdEncoding.DecodeString(inputNoSpacesUpper)
    if erro != nil {
        fmt.Fprintln(os.Stderr, err.Error())
        os.Exit(1)
    }

    // generate a one-time password using the time at 30-second intervals
    epochSeconds := time.Now().Unix()
    pwd := oneTimePassword(keyS, toBytes(epochSeconds/30))

    secondsRemaining := 30 - (epochSeconds % 30)
    fmt.Printf("%06d (%d second(s) remaining)\n", pwd, secondsRemaining)
    return nil

}


func copy(item string) error {
    file,er:=os.Open("/tmp/myga.ini")

    if er!=nil && os.IsNotExist(er){
        fmt.Println("no data!!!")
        os.Exit(1)
    }
    file.Close()

    cfg, err := ini.Load("/tmp/myga.ini")
    if err != nil {
        panic(err)
    }


    section, err := cfg.GetSection(item)
    if err != nil {
        panic(err)
    }

    key, err := section.GetKey("secret")

    if err != nil {
        panic(err)
    }

    inputNoSpaces := strings.Replace(key.String(), " ", "", -1)
    inputNoSpacesUpper := strings.ToUpper(inputNoSpaces)
    keyS, erro := base32.StdEncoding.DecodeString(inputNoSpacesUpper)
    if erro != nil {
        fmt.Fprintln(os.Stderr, err.Error())
        os.Exit(1)
    }

    // generate a one-time password using the time at 30-second intervals
    epochSeconds := time.Now().Unix()
    pwd := oneTimePassword(keyS, toBytes(epochSeconds/30))

    secondsRemaining := 30 - (epochSeconds % 30)
    clipStr := strconv.Itoa(int(pwd))
    clipboard.WriteAll(clipStr)
    fmt.Printf("%06d (%d second(s) remaining)\n", pwd, secondsRemaining)
    return nil

}




func list() error {
    file,er:=os.Open("/tmp/myga.ini")

    if er!=nil && os.IsNotExist(er){
        fmt.Println("no data!!!")
        os.Exit(1)
    }
    file.Close()

    cfg, err := ini.Load("/tmp/myga.ini")
    if err != nil {
        panic(err)
    }


    sections := cfg.Sections()

    for _,v := range sections {
        fmt.Println(v.Name())
    }
    return nil

}




func delete(item string) error {
    file,er:=os.Open("/tmp/myga.ini")

    if er!=nil && os.IsNotExist(er){
        file, _ = os.Create("/tmp/myga.ini")
    }
    file.Close()

    cfg, err := ini.Load("/tmp/myga.ini")
    if err != nil {
        panic(err)
    }

    if _, err := cfg.GetSection(item); err != nil {
        fmt.Println(item + " is not Exist!!!!")
        os.Exit(1)
    }

    cfg.DeleteSection(item)
    cfg.SaveTo("/tmp/myga.ini")
    fmt.Println("ok!!!")
    return nil
}

func modify(item string, secret string) error {
    file,er:=os.Open("/tmp/myga.ini")

    if er!=nil && os.IsNotExist(er){
        file, _ = os.Create("/tmp/myga.ini")
    }
    file.Close()

    cfg, err := ini.Load("/tmp/myga.ini")
    if err != nil {
        panic(err)
    }

    if _, err := cfg.GetSection(item); err != nil {
        fmt.Println(item + " is not Exist!!!!")
        os.Exit(1)
    }

    cfg.Section(item).Key("secret").SetValue(secret)
    cfg.SaveTo("/tmp/myga.ini")
    return nil
}


func main() {
    Cli()
}

上面实现的功能有

  • ga的创建,更新,删除,数据都是保存在本地的
  • 显示所有条目
  • 显示验证码,复制验证码到黏贴板上

后续介绍其他的方式以及ga的工作原理。

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏更流畅、简洁的软件开发方式

用接口实现事件的一种方法,只是玩玩。

  前一阵子,firelong说,应该用接口实现事件,而不应该用委托。我就希望他能给出一个用接口实现事件的方法,我是一直等呀,等到了现在也没有看到。   昨天又...

24880
来自专栏水击三千

Silverlight学习(二)

好久没来写博客了,这期间经历了春节,也因为忙于一个项目,所以博客被疏忽了。最近一段时间一直在用silverlight做项目,从来一开始的不熟悉渐渐的开始上手。今...

299100
来自专栏林德熙的博客

win 10 UWP 标签

本文主要翻译:http://visuallylocated.com/post/2015/02/20/Creating-a-WrapPanel-for-your-...

11220
来自专栏阿炬.NET

JQuery文件上传插件ajaxFileUpload在Asp.net MVC中的使用

36290
来自专栏我有一个梦想

Python 项目实践一(外星人入侵小游戏)第三篇

接着上节的继续学习, 一 重构:模块game_functions 在大型项目中,经常需要在添加新代码前重构既有代码。重构旨在简化既有代码的结构,使其更容易扩展。...

34290
来自专栏林德熙的博客

WPF 修改图片颜色

在 WPF 可以使用很多图片处理的方法,本文告诉大家的是一个图片处理,可以把处理的图片保存在文件。

41710
来自专栏听雨堂

从MapX到MapXtreme2004[2]-图层操作

Mapx中基本的图层操作还是比较简单的,集中在对Layers和Layer的处理上,对别的没有太多要求。   在MapXtreme中,要完成类似功能,发生了一点...

22980
来自专栏搞前端的李蚊子

在vue-cli项目中使用echarts

这个示例使用 vue-cli 脚手架搭建 安装echarts依赖 npm install echarts -S 或者使用国内的淘宝镜像: 安装 npm in...

76790
来自专栏hotqin888的专栏

用beego进行onlyoffice document server的二次开发

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/hotqin888/article/det...

21320
来自专栏我有一个梦想

UE4新手之编程指南

  虚幻引擎4为程序员提供了两套工具集,可共同使用来加速开发的工作流程。 新的游戏类、Slate和Canvas用户接口元素以及编辑器功能可以使用C++语言来编写...

55280

扫码关注云+社区

领取腾讯云代金券