-
Notifications
You must be signed in to change notification settings - Fork 27
/
service.go
53 lines (45 loc) · 1.41 KB
/
service.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
/* cSploit - a simple penetration testing suite
* Copyright (C) 2016 Massimo Dragano aka tux_mind <tux_mind@csploit.org>
*
* cSploit is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* cSploit 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 General Public License
* along with cSploit. If not, see <http://www.gnu.org/licenses/\>.
*
*/
package models
import (
"github.com/cSploit/daemon/models/internal"
"github.com/lair-framework/go-nmap"
)
func init() {
internal.RegisterModels(&Service{})
}
type Service struct {
ID uint `json:"id" gorm:"primary_key"`
Name string `json:"name"`
Product string `json:"product,omitempty"`
Version string `json:"version,omitempty"`
PortID uint `json:"-"`
}
func NewService(s nmap.Service) *Service {
return &Service{Name: s.Name, Version: s.Version, Product: s.Product}
}
func (s *Service) FormatName() string {
var res = s.Name
if s.Product != "" {
res += " - " + s.Product
}
if s.Version != "" {
res += " ( " + s.Version + " )"
}
return res
}