You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
103 lines
2.5 KiB
103 lines
2.5 KiB
/*
|
|
* This program is free software: you can redistribute it and/or
|
|
* modify it under the terms of the GNU Affero General Public License as
|
|
* published by the Free Software Foundation, either version 3 of the
|
|
* License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public
|
|
* License along with this program. If not, see
|
|
* <http://www.gnu.org/licenses/>.
|
|
*
|
|
* (c) Vincenzo "KatolaZ" Nicosia 2017 -- <katolaz@freaknet.org>
|
|
*
|
|
*
|
|
* This file is part of "binnit", a minimal no-fuss pastebin-like
|
|
* server written in golang
|
|
*
|
|
*/
|
|
|
|
|
|
/*
|
|
*
|
|
* Store/Retrieve functions for FS-based paste storage
|
|
*
|
|
*/
|
|
|
|
package paste
|
|
|
|
import(
|
|
"crypto/sha256"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"io/ioutil"
|
|
"errors"
|
|
"bufio"
|
|
"strings"
|
|
)
|
|
|
|
|
|
|
|
func Store(title, date, content, dest_dir string) (string, error) {
|
|
|
|
h := sha256.New()
|
|
|
|
h.Write([]byte(title))
|
|
h.Write([]byte(date))
|
|
h.Write([]byte(content))
|
|
|
|
paste := fmt.Sprintf("# Title: %s\n# Date: %s\n%s", title, date, content)
|
|
|
|
paste_hash := fmt.Sprintf("%x", h.Sum(nil))
|
|
log.Printf(" `-- hash: %s\n", paste_hash)
|
|
paste_dir := dest_dir + "/"
|
|
|
|
|
|
// Now we save the file
|
|
for i := 0; i < len(paste_hash)-16; i++ {
|
|
paste_name := paste_hash[i:i+16]
|
|
if _, err := os.Stat(paste_dir + paste_name); os.IsNotExist(err) {
|
|
// The file does not exist, so we can create it
|
|
if err := ioutil.WriteFile(paste_dir + paste_name, []byte(paste), 0644); err == nil {
|
|
// and then we return the URL:
|
|
log.Printf(" `-- saving new paste to : %s", paste_dir + paste_name)
|
|
return paste_name, nil
|
|
} else {
|
|
log.Printf("Cannot create the paste: %s!\n", paste_dir + paste_name)
|
|
}
|
|
}
|
|
}
|
|
return "", errors.New("Cannot store the paste...Sorry!")
|
|
}
|
|
|
|
|
|
func Retrieve(URI string) (title, date, content string, err error) {
|
|
|
|
|
|
f_cont, err := os.Open(URI)
|
|
defer f_cont.Close()
|
|
|
|
|
|
if err == nil {
|
|
stuff := bufio.NewScanner(f_cont)
|
|
// The first line contains the title
|
|
stuff.Scan()
|
|
title = strings.Trim(strings.Split(stuff.Text(), ":")[1], " ")
|
|
stuff.Scan()
|
|
date = strings.Trim(strings.Join(strings.Split(stuff.Text(), ":")[1:], ":"), " ")
|
|
for stuff.Scan() {
|
|
content += stuff.Text() + "\n"
|
|
}
|
|
} else {
|
|
|
|
return "", "", "", errors.New("Cannot retrieve paste!!!")
|
|
}
|
|
|
|
return title, date, content, nil
|
|
}
|
|
|
|
|