Example #1
0
func (s *SessionManager) Set(session map[string]interface{}, rw http.ResponseWriter, req *http.Request) {
	c, cerr := req.Cookie(s.CookieName)
	lsess := len(session)
	if cerr == nil {
		sessionSign := c.Value
		if lsess > 0 {
			if encodeSession, err := encodeGob(session); err == nil {
				writeFile(s.sessionDir+sessionSign+".golanger", encodeSession)
			} else {
				log.Error("<SessionManager.Set> ", "encodeGob:", err)
			}
		} else {
			s.Clear(sessionSign)
		}
	} else {
		if lsess > 0 {
			if encodeSession, err := encodeGob(session); err == nil {
				sessionSign := s.new(rw)
				writeFile(s.sessionDir+sessionSign+".golanger", encodeSession)
			} else {
				log.Error("<SessionManager.Set> ", "encodeGob:", err)
			}
		}
	}
}
Example #2
0
func decodeCookie(encodedCookie string, key, iv []byte) (map[string]interface{}, error) {
	sessionBytes, err := base64.URLEncoding.DecodeString(encodedCookie)
	if err != nil {
		log.Error("<decodeCookie> ", "base64.Decodestring:", err)
		return nil, err
	}
	aesCipher, err := aes.NewCipher(key)
	if err != nil {
		log.Error("<decodeCookie> ", "aes.NewCipher:", err)
		return nil, err
	}

	decrypter := cipher.NewCBCDecrypter(aesCipher, iv)
	decrypter.CryptBlocks(sessionBytes, sessionBytes)

	buf := bytes.NewBuffer(sessionBytes)
	var gobLen int32
	binary.Read(buf, binary.BigEndian, &gobLen)
	gobBytes := sessionBytes[4 : 4+gobLen]
	session, err := decodeGob(gobBytes)
	if err != nil {
		log.Error("<decodeCookie> ", "decodeGob:", err)
		return nil, err
	}
	return session, nil
}
Example #3
0
func (p *Page) setGlobalTpl(globalTplModTime int64, reset bool) {
	if globalTpls, err := filepath.Glob(p.Config.TemplateDirectory + p.Config.ThemeDirectory + p.Config.TemplateGlobalDirectory + p.Config.TemplateGlobalFile); err == nil && globalTpls != nil && len(globalTpls) > 0 {
		var buf bytes.Buffer
		for _, tpl := range globalTpls {
			if b, err := ioutil.ReadFile(tpl); err == nil {
				buf.Write(b)
			} else {
				log.Error("<Page.setGlobalTpl> ", err)
			}
		}

		sBuf := strings.TrimSpace(buf.String())
		if sBuf != "" {
			p.site.SetTemplateCacheObject("globalTpl", buf.String(), globalTplModTime)
			if reset {
				newT := template.New("globalTpl").Funcs(p.site.templateFunc)
				if t, err := newT.Parse(sBuf); t != nil {
					p.site.globalTemplate = t
				} else {
					log.Error("<Page.setGlobalTpl> ", err)
				}
			} else {
				if t, err := p.site.globalTemplate.Parse(sBuf); t != nil {
					p.site.globalTemplate = t
				} else {
					log.Error("<Page.setGlobalTpl> ", err)
				}
			}
		}
	}
}
Example #4
0
func Test(t *Template, err error) *Template {
	if err != nil {
		log.Error("<Test> Name: ", t.Name(), ", error:", err)
	}

	return t
}
Example #5
0
func (p *Page) loadGeneratedHtml(w http.ResponseWriter, r *http.Request) (b bool) {
	log.Debug("<Page.loadGeneratedHtml> ", "p.Config.SupportTemplate:", p.Config.SupportTemplate)
	log.Debug("<Page.loadGeneratedHtml> ", "p.Config.LoadStaticHtmlWithLogic:", p.Config.LoadStaticHtmlWithLogic)
	log.Debug("<Page.loadGeneratedHtml> ", "p.Config.AutoGenerateHtml:", p.Config.AutoGenerateHtml)

	if p.Config.SupportTemplate && !p.Config.LoadStaticHtmlWithLogic && p.Config.AutoGenerateHtml {
		tplFi, tplErr := os.Stat(p.Config.TemplateDirectory + p.Config.ThemeDirectory + p.Template)
		if tplErr != nil {
			log.Error("<Page.loadGeneratedHtml> ", tplErr)
		} else {
			p.site.base.rmutex.RLock()
			tmplCache := p.site.GetTemplateCache(p.Template)
			p.site.base.rmutex.RUnlock()
			if tplFi.ModTime().Unix() > tmplCache.ModTime {
				return false
			}
		}

		htmlFile := ""
		assetsHtmlDir := p.Config.AssetsDirectory + p.Config.HtmlDirectory
		if strings.HasPrefix(p.Template, p.Config.IndexDirectory) {
			htmlFile = assetsHtmlDir + strings.Replace(p.Template, p.Config.IndexDirectory, "", 1)
		} else {
			htmlFile = assetsHtmlDir + p.Template
		}

		if r.URL.RawQuery != "" {
			htmlFile += "?" + r.URL.RawQuery
		}

		if htmlFi, htmlErr := os.Stat(htmlFile); htmlErr == nil {
			if p.checkHtmlDoWrite(tplFi, htmlFi, htmlErr) {
			}

			htmlContent, err := ioutil.ReadFile(htmlFile)
			if err == nil {
				w.Header().Add("Content-Length", strconv.FormatInt(htmlFi.Size(), 10))
				w.Header().Add("Last-Modified", htmlFi.ModTime().UTC().Format("Mon, 02 Jan 2006 15:04:05 GMT"))
				w.Write(htmlContent)
				return true
			}
		}
	}

	return
}
Example #6
0
func (s *SessionManager) Get(rw http.ResponseWriter, req *http.Request) map[string]interface{} {
	m := map[string]interface{}{}

	if c, err := req.Cookie(s.CookieName); err == nil {
		sessionSign := c.Value
		if content, err := readFile(s.sessionDir + sessionSign + ".golanger"); err == nil {
			if len(content) > 0 {
				if dm, err := decodeGob(content); err == nil {
					m = dm
				} else {
					log.Error("<SessionManager.Get> ", "decodeGob:", err)
				}
			}
		}
	} else {
		s.new(rw)
	}

	return m
}
Example #7
0
func encodeCookie(content map[string]interface{}, key, iv []byte) (string, error) {
	sessionGob, err := encodeGob(content)
	if err != nil {
		log.Error("<encodeCookie> ", err)
		return "", err
	}

	//实现动态填充,达到aes.BlockSize的倍数,+4是为了后面提供4个字节来保存字符串长度使用的
	padLen := aes.BlockSize - (len(sessionGob)+4)%aes.BlockSize
	buf := bytes.NewBuffer(nil)
	var sessionLen int32 = (int32)(len(sessionGob))
	binary.Write(buf, binary.BigEndian, sessionLen)
	buf.WriteString(sessionGob)
	buf.WriteString(strings.Repeat("\000", padLen))
	sessionBytes := buf.Bytes()
	aesCipher, err := aes.NewCipher(key)
	if err != nil {
		return "", err
	}
	encrypter := cipher.NewCBCEncrypter(aesCipher, iv)
	encrypter.CryptBlocks(sessionBytes, sessionBytes)
	b64 := base64.URLEncoding.EncodeToString(sessionBytes)
	return b64, nil
}
Example #8
0
func (p *Page) routeTemplate(w http.ResponseWriter, r *http.Request) {
	if globalTplFi, err := os.Stat(p.Config.TemplateDirectory + p.Config.ThemeDirectory + p.Config.TemplateGlobalDirectory); err != nil {
		log.Error("<Page.routeTemplate> ", err)
	} else {
		if globalTplCache := p.site.GetTemplateCache("globalTpl"); globalTplCache.ModTime > 0 {
			if globalTplFi.ModTime().Unix() > globalTplCache.ModTime {
				p.setGlobalTpl(globalTplFi.ModTime().Unix(), true)
			}
		}
	}

	if tplFi, err := os.Stat(p.Config.TemplateDirectory + p.Config.ThemeDirectory + p.Template); err != nil {
		p.site.base.rmutex.RLock()
		tmplCache := p.site.GetTemplateCache(p.Template)
		if tmplCache.ModTime > 0 {
			p.site.base.mutex.Lock()
			p.DelTemplateCache(p.Template)
			p.site.base.mutex.Unlock()
			log.Info("<Page.routeTemplate> ", "Delete Template Cache:", p.Template)
		}
		p.site.base.rmutex.RUnlock()
	} else {
		p.site.base.rmutex.RLock()
		tmplCache := p.site.GetTemplateCache(p.Template)

		if tplFi.ModTime().Unix() > tmplCache.ModTime {
			p.site.base.mutex.Lock()
			p.SetTemplateCache(p.Template, p.Config.TemplateDirectory+p.Config.ThemeDirectory+p.Template)
			p.site.base.mutex.Unlock()

			tmplCache = p.site.GetTemplateCache(p.Template)
		}

		globalTemplate, _ := p.site.globalTemplate.Clone()
		p.site.base.rmutex.RUnlock()
		pageTemplate, err := globalTemplate.New(filepath.Base(p.Template)).Parse(tmplCache.Content)

		if err != nil {
			log.Error("<Page.routeTemplate> ", err)
			w.Write([]byte(fmt.Sprint(err)))
		} else {
			templateVar := map[string]interface{}{
				"G":        p.GET,
				"P":        p.POST,
				"S":        p.SESSION,
				"O_S":      p.ONCE_SESSION,
				"C":        p.COOKIE,
				"CS":       p.COOKIE_SESSION,
				"D":        p.Document,
				"L":        p.LANG,
				"Config":   p.Config.M,
				"Template": p.Template,
			}

			p.site.base.rmutex.RLock()
			templateVar["Siteroot"] = p.site.Root
			templateVar["Version"] = p.site.Version
			p.site.base.rmutex.RUnlock()

			if !p.Document.GenerateHtml {
				err := pageTemplate.Execute(w, templateVar)
				if err != nil {
					log.Error("<Page.routeTemplate> ", err)
					w.Write([]byte(fmt.Sprint(err)))
				}
			} else {
				htmlFile := ""
				assetsHtmlDir := p.Config.AssetsDirectory + p.Config.HtmlDirectory
				if strings.HasPrefix(p.Template, p.Config.IndexDirectory) {
					htmlFile = assetsHtmlDir + strings.Replace(p.Template, p.Config.IndexDirectory, "", 1)
				} else {
					htmlFile = assetsHtmlDir + p.Template
				}

				if r.URL.RawQuery != "" {
					htmlFile += "?" + r.URL.RawQuery
				}

				htmlDir := filepath.Dir(htmlFile)
				if htmlDirFi, err := os.Stat(htmlDir); err != nil || !htmlDirFi.IsDir() {
					os.MkdirAll(htmlDir, 0777)
					log.Info("<Page.routeTemplate> ", "MkdirAll:", htmlDir)
				}

				htmlFi, htmlErr := os.Stat(htmlFile)
				if p.checkHtmlDoWrite(tplFi, htmlFi, htmlErr) {
					if file, err := os.OpenFile(htmlFile, os.O_CREATE|os.O_WRONLY, 0777); err != nil {
						log.Error("<Page.routeTemplate> ", err)
					} else {
						if p.Config.AutoJumpToHtml || p.Config.ChangeSiteRoot {
							p.site.base.rmutex.RLock()
							templateVar["Siteroot"] = p.site.Root + p.Config.HtmlDirectory
							p.site.base.rmutex.RUnlock()
						}

						pageTemplate.Execute(file, templateVar)
					}
				}

				if p.Config.AutoJumpToHtml {
					p.site.base.rmutex.RLock()
					siteRoot := p.site.Root
					p.site.base.rmutex.RUnlock()
					http.Redirect(w, r, siteRoot+htmlFile[len(p.Config.AssetsDirectory):], http.StatusFound)
				} else if p.Config.AutoLoadStaticHtml && htmlFi != nil && htmlErr == nil {
					htmlContent, err := ioutil.ReadFile(htmlFile)
					if err == nil {
						w.Write(htmlContent)
					} else {
						log.Error("<Page.routeTemplate> ", err)
					}
				} else {
					err := pageTemplate.Execute(w, templateVar)
					if err != nil {
						log.Error("<Page.routeTemplate> ", err)
					}
				}
			}
		}
	}
}
Example #9
0
func (p *Page) reset(update bool) {
	p.setLog(p.Config.SupportLog, p.Config.LogWriteTo, p.Config.LogLevel)
	p.setUrlManage(p.Config.SupportUrlManage, p.Config.SupportUrlManageWithCache, p.Config.UrlManageRule)

	if update {
		if p.site.supportSession != p.Config.SupportSession {
			p.site.supportSession = p.Config.SupportSession
		}

		if p.site.supportCookieSession != p.Config.SupportCookieSession {
			p.site.supportCookieSession = p.Config.SupportCookieSession
		}

		if p.site.supportI18n != p.Config.SupportI18n {
			p.site.supportI18n = p.Config.SupportI18n
		}

		if p.Document.Theme != p.Config.Theme {
			p.Document.Theme = p.Config.Theme
		}

		if p.Document.Static != p.Config.SiteRoot+p.Config.StaticDirectory {
			p.Document.Static = p.Config.SiteRoot + p.Config.StaticDirectory
		}

		if p.site.Root == p.Config.SiteRoot {
			return
		} else {
			log.Info("<Page.reset> ", "p.Config.SiteRoot be changed:", p.Config.SiteRoot)
			p.SetDefaultController(p.GetController(p.Config.IndexDirectory))
			p.UpdateController(p.site.Root, p.Config.SiteRoot, p.DefaultController)
			p.site.Root = p.Config.SiteRoot
		}
	} else {
		p.site.supportSession = p.Config.SupportSession
		p.site.supportCookieSession = p.Config.SupportCookieSession
		p.site.supportI18n = p.Config.SupportI18n
		p.Document.Theme = p.Config.Theme
		p.site.Root = p.Config.SiteRoot
		p.Document.Static = p.site.Root + p.Config.StaticDirectory
		p.SetDefaultController(p.GetController(p.Config.IndexDirectory))
		p.RegisterController(p.site.Root, p.DefaultController)
		p.site.globalTemplate = template.New("globalTpl").Funcs(p.site.templateFunc)
	}

	if globalCssFi, err := os.Stat(p.Config.StaticCssDirectory + "/global/"); err == nil && globalCssFi.IsDir() {
		DcssPath := p.Config.StaticCssDirectory + "global/"
		p.Document.Css["global"] = p.site.Root + DcssPath[len(p.Config.AssetsDirectory):]
		log.Debug("<Page.reset> ", `p.Document.Css["global"]:`, p.Document.Css["global"])

		if _, err := os.Stat(DcssPath + "global.css"); err == nil {
			p.Document.GlobalCssFile = p.Document.Css["global"] + "global.css"
		}
	}

	if globalJsFi, err := os.Stat(p.Config.StaticJsDirectory + "/global/"); err == nil && globalJsFi.IsDir() {
		DjsPath := p.Config.StaticJsDirectory + "global/"
		p.Document.Js["global"] = p.site.Root + DjsPath[len(p.Config.AssetsDirectory):]
		log.Debug("<Page.reset> ", `p.Document.Js["global"]:`, p.Document.GlobalCssFile)
		if _, err := os.Stat(DjsPath + "global.js"); err == nil {
			p.Document.GlobalJsFile = p.Document.Js["global"] + "global.js"
		}
	}

	if globalImgFi, err := os.Stat(p.Config.StaticImgDirectory + "/global/"); err == nil && globalImgFi.IsDir() {
		DimgPath := p.Config.StaticImgDirectory + "global/"
		p.Document.Img["global"] = p.site.Root + DimgPath[len(p.Config.AssetsDirectory):]
		log.Debug("<Page.reset> ", `p.Document.Img["global"]:`, p.Document.Img["global"])
	}

	if globalTplFi, err := os.Stat(p.Config.TemplateDirectory + p.Config.ThemeDirectory + p.Config.TemplateGlobalDirectory); err != nil {
		log.Error("<Page.reset> ", err)
	} else {
		p.setGlobalTpl(globalTplFi.ModTime().Unix(), false)
	}
}