/
db.go
103 lines (93 loc) · 1.79 KB
/
db.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
package main
import (
"database/sql"
"fmt"
_ "github.com/mattn/go-sqlite3"
"io/ioutil"
"log"
"os"
"os/exec"
"strconv"
"strings"
"time"
)
type SMS struct {
address string
date int64
body string
sub_id int
}
type Database struct {
*sql.DB
last int64
db_path string
last_path string
}
func (d *Database) Save(l int64) error {
if l == 0 {
return nil
}
d.last = l
b := []byte(strconv.FormatInt(d.last, 10))
err := ioutil.WriteFile(d.last_path, b, 0644)
if err != nil {
return err
}
return nil
}
func (d *Database) Pull() {
cmd := exec.Command("adb", "pull", "/data/data/com.android.providers.telephony/databases/mmssms.db", d.db_path)
cmd.Run()
}
func (d *Database) Clear() {
os.Remove(d.db_path)
}
func (d *Database) Load() error {
b, err := ioutil.ReadFile(d.last_path)
if err != nil {
return err
}
sb := strings.Trim(string(b), "\r\t\n ")
last, err := strconv.ParseInt(sb, 10, 64)
if err != nil {
return err
}
d.last = last
return nil
}
func (d *Database) Init() {
err := d.Load()
if err != nil {
log.Println(err)
d.Save(time.Now().Unix() * 1000)
log.Println("Last file not found. Created and set last time to now.")
}
}
func (d *Database) Read() ([]SMS, error) {
d.Pull()
defer d.Clear()
ret := make([]SMS, 0, 10)
db, err := sql.Open("sqlite3", d.db_path)
if err != nil {
return ret, err
}
defer db.Close()
rows, err := db.Query(fmt.Sprintf("SELECT `address`, `date`, `body`, `sub_id` FROM sms WHERE `date` > %d ORDER BY `date` ASC", d.last))
if err != nil {
return ret, err
}
defer rows.Close()
for rows.Next() {
sms := SMS{}
err = rows.Scan(&sms.address, &sms.date, &sms.body, &sms.sub_id)
if err != nil {
return ret, err
}
ret = append(ret, sms)
}
err = rows.Err()
if err != nil {
return ret, err
}
return ret, nil
}