From ed26172c1b1131ae799c245b08a7ab598b4dc73d Mon Sep 17 00:00:00 2001 From: shaoying Date: Mon, 13 Nov 2017 08:47:18 +0800 Subject: [PATCH] =?UTF-8?q?mac=20add=20aaa=20ssh=20=E6=B7=BB=E5=8A=A0?= =?UTF-8?q?=E4=BA=86=E8=AE=A4=E8=AF=81=E5=92=8C=E5=8A=A0=E5=AF=86=E9=80=9A?= =?UTF-8?q?=E4=BF=A1=E6=A8=A1=E5=9D=97?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/context/aaa/aaa.go | 159 +++++++++++++++++++++++++++++++++++++++++ src/context/ssh/ssh.go | 44 ++++++++++++ 2 files changed, 203 insertions(+) create mode 100644 src/context/aaa/aaa.go create mode 100644 src/context/ssh/ssh.go diff --git a/src/context/aaa/aaa.go b/src/context/aaa/aaa.go new file mode 100644 index 00000000..71f95cf9 --- /dev/null +++ b/src/context/aaa/aaa.go @@ -0,0 +1,159 @@ +package aaa // {{{ +// }}} +import ( // {{{ + "context" + _ "context/cli" + + "crypto/md5" + "encoding/hex" + "math/rand" + "time" + + "fmt" + "log" +) + +// }}} + +type AAA struct { + sessions map[string]*ctx.Context + *ctx.Context +} + +func (aaa *AAA) session(meta string) string { // {{{ + bs := md5.Sum([]byte(fmt.Sprintln("%d%d%s", time.Now().Unix(), rand.Int(), meta))) + sessid := hex.EncodeToString(bs[:]) + return sessid +} + +// }}} + +func (aaa *AAA) Begin(m *ctx.Message, arg ...string) ctx.Server { // {{{ + return aaa +} + +// }}} +func (aaa *AAA) Start(m *ctx.Message, arg ...string) bool { // {{{ + return true +} + +// }}} +func (aaa *AAA) Spawn(c *ctx.Context, m *ctx.Message, arg ...string) ctx.Server { // {{{ + c.Caches = map[string]*ctx.Cache{ + "username": &ctx.Cache{Name: "用户名", Value: arg[0], Help: "显示已经启动运行模块的数量"}, + "password": &ctx.Cache{}, + "time": &ctx.Cache{Name: "登录时间", Value: fmt.Sprintf("%d", time.Now().Unix()), Help: "用户登录时间"}, + "sessid": &ctx.Cache{Name: "会话标识", Value: aaa.session(arg[0]), Help: "用户的会话标识"}, + "group": &ctx.Cache{Name: "群组", Value: "void", Help: "用户所属群组"}, + } + + *c.Caches["password"] = *Index.Caches["password"] + if len(arg) > 2 { + c.Caches["group"].Value = arg[2] + c.Caches["password"].Value = arg[1] + } else if len(arg) > 1 { + m.Cap("password", arg[1]) + } + + c.Configs = map[string]*ctx.Config{} + c.Commands = map[string]*ctx.Command{} + + s := new(AAA) + s.Context = c + return s +} + +// }}} +func (aaa *AAA) Exit(m *ctx.Message, arg ...string) bool { // {{{ + return true +} + +// }}} + +var Index = &ctx.Context{Name: "aaa", Help: "认证中心", + Caches: map[string]*ctx.Cache{ + "password": &ctx.Cache{Name: "密码", Value: "", Help: "用户密码，加密存储", Hand: func(m *ctx.Message, x *ctx.Cache, arg ...string) string { + if len(arg) > 0 { // {{{ + if arg[0] == "" { + return "" + } + + if x.Value == "" { + bs := md5.Sum([]byte(fmt.Sprintln("用户密码:%s", arg[0]))) + return hex.EncodeToString(bs[:]) + } else { + bs := md5.Sum([]byte(fmt.Sprintln("用户密码:%s", arg[0]))) + if x.Value != hex.EncodeToString(bs[:]) { + log.Println(m.Target.Name, "login in:", arg[0], "密码错误") + panic("密码错误") + } + } + } + return x.Value + // }}} + }}, + }, + Configs: map[string]*ctx.Config{ + "rootname": &ctx.Config{Name: "根用户的名称", Value: "root", Help: "系统根据此名确定是否超级用户"}, + }, + Commands: map[string]*ctx.Command{ + "login": &ctx.Command{Name: "login [sessid]|[username password [group]]]", Help: "", Hand: func(c *ctx.Context, m *ctx.Message, key string, arg ...string) string { + aaa := c.Server.(*AAA) // {{{ + switch len(arg) { + case 0: + target := m.Target + m.Target.Travel(func(c *ctx.Context) bool { + m.Target = c + m.Echo("%s %s %s\n", c.Name, m.Cap("group"), m.Cap("sessid")) + return true + }) + m.Target = target + case 1: + target := m.Target + if s, ok := aaa.sessions[arg[0]]; ok { + m.Target = s + m.Source.Owner = s + log.Println(aaa.Name, "login on:", aaa.sessions) + return m.Cap("username") + } + m.Target = target + case 2: + if s := m.Target.Find(arg[0]); s != nil { + m.Target = s + log.Println(aaa.Name, "login in:", arg[0]) + m.Cap("password", arg[1]) + log.Println(aaa.Name, "login in:", arg[0]) + + m.Source.Owner = s + if arg[0] == m.Conf("rootname") { + ctx.Index.Owner = s + c.Owner = s + } + return m.Cap("sessid") + } else { + m.Start(arg[0], arg...) + sessid := m.Cap("sessid") + aaa.sessions[sessid] = m.Target + log.Println(aaa.Name, "login up:", arg[0]) + m.Source.Owner = s + if arg[0] == m.Conf("rootname") { + ctx.Index.Owner = s + c.Owner = s + } + return sessid + } + case 3: + m.Start(arg[0], arg...) + } + return "" + // }}} + }}, + }, +} + +func init() { + aaa := &AAA{} + aaa.Context = Index + aaa.sessions = make(map[string]*ctx.Context) + ctx.Index.Register(Index, aaa) +} diff --git a/src/context/ssh/ssh.go b/src/context/ssh/ssh.go new file mode 100644 index 00000000..e4fe117a --- /dev/null +++ b/src/context/ssh/ssh.go @@ -0,0 +1,44 @@ +package ssh + +import ( + "context" + _ "context/cli" +) + +type SSH struct { + *ctx.Context +} + +func (ssh *SSH) Begin(m *ctx.Message, arg ...string) ctx.Server { + return ssh +} + +func (ssh *SSH) Start(m *ctx.Message, arg ...string) bool { + return true +} + +func (ssh *SSH) Spawn(c *ctx.Context, m *ctx.Message, arg ...string) ctx.Server { + c.Caches = map[string]*ctx.Cache{} + c.Configs = map[string]*ctx.Config{} + c.Commands = map[string]*ctx.Command{} + + s := new(SSH) + s.Context = c + return s +} + +func (ssh *SSH) Exit(m *ctx.Message, arg ...string) bool { + return true +} + +var Index = &ctx.Context{Name: "ssh", Help: "加密终端", + Caches: map[string]*ctx.Cache{}, + Configs: map[string]*ctx.Config{}, + Commands: map[string]*ctx.Command{}, +} + +func init() { + ssh := &SSH{} + ssh.Context = Index + ctx.Index.Register(Index, ssh) +}