vpn.go 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002
  1. package ovpm
  2. import (
  3. "bytes"
  4. "fmt"
  5. "io"
  6. "math/big"
  7. "net"
  8. "os"
  9. "os/exec"
  10. "path/filepath"
  11. "strings"
  12. "sync"
  13. "text/template"
  14. "time"
  15. "github.com/Sirupsen/logrus"
  16. "github.com/asaskevich/govalidator"
  17. "github.com/cad/ovpm/bindata"
  18. "github.com/cad/ovpm/pki"
  19. "github.com/cad/ovpm/supervisor"
  20. "github.com/coreos/go-iptables/iptables"
  21. "github.com/google/uuid"
  22. "github.com/jinzhu/gorm"
  23. )
  24. // Possible VPN protocols.
  25. const (
  26. TCPProto string = "tcp"
  27. UDPProto string = "udp"
  28. )
  29. // serverModel is database model for storing VPN server related stuff.
  30. type dbServerModel struct {
  31. gorm.Model
  32. Name string `gorm:"unique_index"` // Server name.
  33. SerialNumber string
  34. Hostname string // Server's ip address or FQDN
  35. Port string // Server's listening port
  36. Proto string // Server's proto udp or tcp
  37. Cert string // Server RSA certificate.
  38. Key string // Server RSA private key.
  39. CACert string // Root CA RSA certificate.
  40. CAKey string // Root CA RSA key.
  41. Net string // VPN network.
  42. Mask string // VPN network mask.
  43. CRL string // Certificate Revocation List
  44. DNS string // DNS servers to push to the clients.
  45. }
  46. var serverInstance *Server
  47. var once sync.Once
  48. // Server represents VPN server.
  49. type Server struct {
  50. dbServerModel
  51. webPort string
  52. emitToFileFunc func(path, content string, mode uint) error
  53. openFunc func(path string) (io.Reader, error)
  54. parseStatusLogFunc func(f io.Reader) ([]clEntry, []rtEntry)
  55. }
  56. // TheServer returns a pointer to the server instance.
  57. //
  58. // Server instance is a singleton instance that is initialized
  59. // on the first call made to the TheServer().
  60. func TheServer() *Server {
  61. once.Do(func() {
  62. // Initialize the server instance by setting default mockable funcs & attributes.
  63. serverInstance = &Server{
  64. emitToFileFunc: emitToFile,
  65. openFunc: func(path string) (io.Reader, error) {
  66. return os.Open(path)
  67. },
  68. parseStatusLogFunc: parseStatusLog,
  69. }
  70. })
  71. if db != nil {
  72. serverInstance.Refresh()
  73. } else {
  74. logrus.Warn("database is not connected yet. skipping server instance refresh")
  75. }
  76. return serverInstance
  77. }
  78. // CheckSerial takes a serial number and checks it against the current server's serial number.
  79. func (svr *Server) CheckSerial(serial string) bool {
  80. return serial == svr.SerialNumber
  81. }
  82. // GetSerialNumber returns server's serial number.
  83. func (svr *Server) GetSerialNumber() string {
  84. return svr.SerialNumber
  85. }
  86. // GetServerName returns server's name.
  87. func (svr *Server) GetServerName() string {
  88. if svr.Name != "" {
  89. return svr.Name
  90. }
  91. return "default"
  92. }
  93. // GetHostname returns vpn server's hostname.
  94. func (svr *Server) GetHostname() string {
  95. return svr.Hostname
  96. }
  97. // GetPort returns vpn server's port.
  98. func (svr *Server) GetPort() string {
  99. if svr.Port != "" {
  100. return svr.Port
  101. }
  102. return DefaultVPNPort
  103. }
  104. // GetProto returns vpn server's proto.
  105. func (svr *Server) GetProto() string {
  106. if svr.Proto != "" {
  107. return svr.Proto
  108. }
  109. return DefaultVPNProto
  110. }
  111. // CAExpiresAt returns the expiry date time of the CA.
  112. func (svr *Server) CAExpiresAt() time.Time {
  113. if !svr.IsInitialized() {
  114. return time.Time{}
  115. }
  116. crt, err := pki.ReadCertFromPEM(svr.CACert)
  117. if err != nil {
  118. logrus.Fatalf("can't parse cert: %v", err)
  119. }
  120. return crt.NotAfter
  121. }
  122. // ExpiresAt returns the expiry date time of the server cert.
  123. func (svr *Server) ExpiresAt() time.Time {
  124. if !svr.IsInitialized() {
  125. return time.Time{}
  126. }
  127. crt, err := pki.ReadCertFromPEM(svr.Cert)
  128. if err != nil {
  129. logrus.Fatalf("can't parse cert: %v", err)
  130. }
  131. return crt.NotAfter
  132. }
  133. // GetKey returns vpn server's key.
  134. func (svr *Server) GetKey() string {
  135. return svr.Key
  136. }
  137. // GetCACert returns vpn server's cacert.
  138. func (svr *Server) GetCACert() string {
  139. return svr.CACert
  140. }
  141. // GetCAKey returns vpn server's cakey.
  142. func (svr *Server) GetCAKey() string {
  143. return svr.CAKey
  144. }
  145. // GetNet returns vpn server's net.
  146. func (svr *Server) GetNet() string {
  147. return svr.Net
  148. }
  149. // GetMask returns vpn server's mask.
  150. func (svr *Server) GetMask() string {
  151. return svr.Mask
  152. }
  153. // GetCRL returns vpn server's crl.
  154. func (svr *Server) GetCRL() string {
  155. return svr.CRL
  156. }
  157. // GetDNS returns vpn server's dns.
  158. func (svr *Server) GetDNS() string {
  159. if svr.DNS != "" {
  160. return svr.DNS
  161. }
  162. return DefaultVPNDNS
  163. }
  164. // GetCreatedAt returns server's created at.
  165. func (svr *Server) GetCreatedAt() string {
  166. return svr.CreatedAt.Format(time.UnixDate)
  167. }
  168. // Init regenerates keys and certs for a Root CA, gets initial settings for the VPN server
  169. // and saves them in the database.
  170. //
  171. // 'proto' can be either "udp" or "tcp" and if it's "" it defaults to "udp".
  172. //
  173. // 'ipblock' is a IP network in the CIDR form. VPN clients get their IP addresses from this network.
  174. // It defaults to const 'DefaultVPNNetwork'.
  175. //
  176. // Please note that, Init is potentially destructive procedure, it will cause invalidation of
  177. // existing .ovpn profiles of the current users. So it should be used carefully.
  178. func (svr *Server) Init(hostname string, port string, proto string, ipblock string, dns string) error {
  179. if port == "" {
  180. port = DefaultVPNPort
  181. }
  182. if dns == "" {
  183. dns = DefaultVPNDNS
  184. }
  185. switch proto {
  186. case "":
  187. proto = UDPProto
  188. case UDPProto:
  189. proto = UDPProto
  190. case TCPProto:
  191. proto = TCPProto
  192. default:
  193. return fmt.Errorf("validation error: proto:`%s` should be either 'tcp' or 'udp'", proto)
  194. }
  195. // vpn network to use.
  196. var ipnet *net.IPNet
  197. // If user didn't specify, pick the vpn network from defaults.
  198. if ipblock == "" {
  199. var err error
  200. _, ipnet, err = net.ParseCIDR(DefaultVPNNetwork)
  201. if err != nil {
  202. return fmt.Errorf("can not parse CIDR %s: %v", DefaultVPNNetwork, err)
  203. }
  204. }
  205. // Check if the user specified vpn network is valid.
  206. if ipblock != "" && !govalidator.IsCIDR(ipblock) {
  207. return fmt.Errorf("validation error: ipblock:`%s` should be a CIDR network", ipblock)
  208. }
  209. // Use user specified vpn network.
  210. if ipblock != "" {
  211. var err error
  212. _, ipnet, err = net.ParseCIDR(ipblock)
  213. if err != nil {
  214. return fmt.Errorf("can parse ipblock: %s", err)
  215. }
  216. }
  217. if !govalidator.IsNumeric(port) {
  218. return fmt.Errorf("validation error: port:`%s` should be numeric", port)
  219. }
  220. serverName := "default"
  221. if svr := TheServer(); svr.IsInitialized() {
  222. if err := svr.Deinit(); err != nil {
  223. logrus.Errorf("server can not be deleted: %v", err)
  224. return err
  225. }
  226. }
  227. if !govalidator.IsHost(hostname) {
  228. return fmt.Errorf("validation error: hostname:`%s` should be either an ip address or a FQDN", hostname)
  229. }
  230. if !govalidator.IsIPv4(dns) {
  231. return fmt.Errorf("validation error: dns:`%s` should be an ip address", dns)
  232. }
  233. ca, err := pki.NewCA()
  234. if err != nil {
  235. return fmt.Errorf("can not create ca creds: %s", err)
  236. }
  237. srv, err := pki.NewServerCertHolder(ca)
  238. if err != nil {
  239. return fmt.Errorf("can not create server cert creds: %s", err)
  240. }
  241. serialNumber := uuid.New().String()
  242. serverInstance := dbServerModel{
  243. Name: serverName,
  244. SerialNumber: serialNumber,
  245. Hostname: hostname,
  246. Proto: proto,
  247. Port: port,
  248. Cert: srv.Cert,
  249. Key: srv.Key,
  250. CACert: ca.Cert,
  251. CAKey: ca.Key,
  252. Net: ipnet.IP.To4().String(),
  253. Mask: net.IP(ipnet.Mask).To4().String(),
  254. DNS: dns,
  255. }
  256. db.Create(&serverInstance)
  257. if db.NewRecord(&serverInstance) {
  258. return fmt.Errorf("can not create server instance on database")
  259. }
  260. users, err := GetAllUsers()
  261. if err != nil {
  262. return err
  263. }
  264. // Sign all users in the db with the new server
  265. for _, user := range users {
  266. err := user.Renew()
  267. logrus.Infof("user certificate changed for %s, you should run: $ ovpm user export-config --user %s", user.Username, user.Username)
  268. if err != nil {
  269. logrus.Errorf("can not sign user %s: %v", user.Username, err)
  270. continue
  271. }
  272. // Set dynamic ip to user.
  273. user.HostID = 0
  274. db.Save(&user.dbUserModel)
  275. }
  276. TheServer().EmitWithRestart()
  277. logrus.Infof("server initialized")
  278. return nil
  279. }
  280. // Update updates VPN server attributes.
  281. func (svr *Server) Update(ipblock string, dns string) error {
  282. if !svr.IsInitialized() {
  283. return fmt.Errorf("server is not initialized")
  284. }
  285. var changed bool
  286. if ipblock != "" && govalidator.IsCIDR(ipblock) {
  287. var ipnet *net.IPNet
  288. _, ipnet, err := net.ParseCIDR(ipblock)
  289. if err != nil {
  290. return fmt.Errorf("can not parse CIDR %s: %v", ipblock, err)
  291. }
  292. svr.dbServerModel.Net = ipnet.IP.To4().String()
  293. svr.dbServerModel.Mask = net.IP(ipnet.Mask).To4().String()
  294. changed = true
  295. }
  296. if dns != "" && govalidator.IsIPv4(dns) {
  297. svr.dbServerModel.DNS = dns
  298. changed = true
  299. }
  300. if changed {
  301. db.Save(svr.dbServerModel)
  302. users, err := GetAllUsers()
  303. if err != nil {
  304. return err
  305. }
  306. // Set all users to dynamic ip address.
  307. // This way we prevent any ip range mismatch.
  308. for _, user := range users {
  309. user.HostID = 0
  310. db.Save(user.dbUserModel)
  311. }
  312. svr.EmitWithRestart()
  313. logrus.Infof("server updated")
  314. }
  315. return nil
  316. }
  317. // Deinit deletes the VPN server from the database and frees the allocated resources.
  318. func (svr *Server) Deinit() error {
  319. if !svr.IsInitialized() {
  320. return fmt.Errorf("server not found")
  321. }
  322. db.Unscoped().Delete(&dbServerModel{})
  323. db.Unscoped().Delete(&dbRevokedModel{})
  324. svr.EmitWithRestart()
  325. return nil
  326. }
  327. // DumpsClientConfig generates .ovpn file for the given vpn user and returns it as a string.
  328. func (svr *Server) DumpsClientConfig(username string) (string, error) {
  329. var result bytes.Buffer
  330. user, err := GetUser(username)
  331. if err != nil {
  332. return "", err
  333. }
  334. params := struct {
  335. Hostname string
  336. Port string
  337. CA string
  338. Key string
  339. Cert string
  340. NoGW bool
  341. Proto string
  342. }{
  343. Hostname: svr.GetHostname(),
  344. Port: svr.GetPort(),
  345. CA: svr.GetCACert(),
  346. Key: user.getKey(),
  347. Cert: user.GetCert(),
  348. NoGW: user.IsNoGW(),
  349. Proto: svr.GetProto(),
  350. }
  351. data, err := bindata.Asset("template/client.ovpn.tmpl")
  352. if err != nil {
  353. return "", err
  354. }
  355. t, err := template.New("client.ovpn").Parse(string(data))
  356. if err != nil {
  357. return "", fmt.Errorf("can not parse client.ovpn.tmpl template: %s", err)
  358. }
  359. err = t.Execute(&result, params)
  360. if err != nil {
  361. return "", fmt.Errorf("can not render client.ovpn: %s", err)
  362. }
  363. return result.String(), nil
  364. }
  365. // DumpClientConfig generates .ovpn file for the given vpn user and dumps it to outPath.
  366. func (svr *Server) DumpClientConfig(username, path string) error {
  367. result, err := svr.DumpsClientConfig(username)
  368. if err != nil {
  369. return err
  370. }
  371. // Wite rendered content into openvpn server conf.
  372. return svr.emitToFile(path, result, 0)
  373. }
  374. // GetSystemCA returns the system CA from the database if available.
  375. func (svr *Server) GetSystemCA() (*pki.CA, error) {
  376. server := dbServerModel{}
  377. db.First(&server)
  378. if db.NewRecord(&server) {
  379. return nil, fmt.Errorf("server record does not exists in db")
  380. }
  381. return &pki.CA{
  382. CertHolder: pki.CertHolder{
  383. Cert: server.CACert,
  384. Key: server.CAKey,
  385. },
  386. }, nil
  387. }
  388. // vpnProc represents the OpenVPN process that is managed by the ovpm supervisor globally OpenVPN.
  389. var vpnProc supervisor.Supervisable
  390. // StartVPNProc starts the OpenVPN process.
  391. func (svr *Server) StartVPNProc() {
  392. if !svr.IsInitialized() {
  393. logrus.Error("can not launch OpenVPN because system is not initialized")
  394. return
  395. }
  396. if vpnProc == nil {
  397. panic(fmt.Sprintf("vpnProc is not initialized!"))
  398. }
  399. if vpnProc.Status() == supervisor.RUNNING {
  400. logrus.Error("OpenVPN is already started")
  401. return
  402. }
  403. svr.Emit()
  404. vpnProc.Start()
  405. ensureNatEnabled()
  406. }
  407. // RestartVPNProc restarts the OpenVPN process.
  408. func (svr *Server) RestartVPNProc() {
  409. if !svr.IsInitialized() {
  410. logrus.Error("can not launch OpenVPN because system is not initialized")
  411. return
  412. }
  413. if vpnProc == nil {
  414. panic(fmt.Sprintf("vpnProc is not initialized!"))
  415. }
  416. svr.Emit()
  417. vpnProc.Restart()
  418. ensureNatEnabled()
  419. }
  420. // StopVPNProc stops the OpenVPN process.
  421. func (svr *Server) StopVPNProc() {
  422. if vpnProc == nil {
  423. panic(fmt.Sprintf("vpnProc is not initialized!"))
  424. }
  425. if vpnProc.Status() != supervisor.RUNNING {
  426. logrus.Error("OpenVPN is already not running")
  427. return
  428. }
  429. vpnProc.Stop()
  430. }
  431. // Emit generates all needed files for the OpenVPN server and dumps them to their corresponding paths defined in the config.
  432. func (svr *Server) Emit() error {
  433. // Check dependencies
  434. if !checkOpenVPNExecutable() {
  435. return fmt.Errorf("openvpn executable can not be found! you should install OpenVPN on this machine")
  436. }
  437. if !checkOpenSSLExecutable() {
  438. return fmt.Errorf("openssl executable can not be found! you should install openssl on this machine")
  439. }
  440. if !checkIptablesExecutable() {
  441. return fmt.Errorf("iptables executable can not be found")
  442. }
  443. if !svr.IsInitialized() {
  444. return fmt.Errorf("you should create a server first. e.g. $ ovpm vpn create-server")
  445. }
  446. if err := svr.emitServerConf(); err != nil {
  447. return fmt.Errorf("can not emit server conf: %s", err)
  448. }
  449. if err := svr.emitServerCert(); err != nil {
  450. return fmt.Errorf("can not emit server cert: %s", err)
  451. }
  452. if err := svr.emitServerKey(); err != nil {
  453. return fmt.Errorf("can not emit server key: %s", err)
  454. }
  455. if err := svr.emitCACert(); err != nil {
  456. return fmt.Errorf("can not emit ca cert : %s", err)
  457. }
  458. if err := svr.emitCAKey(); err != nil {
  459. return fmt.Errorf("can not emit ca key: %s", err)
  460. }
  461. if err := svr.emitDHParams(); err != nil {
  462. return fmt.Errorf("can not emit dhparams: %s", err)
  463. }
  464. if err := svr.emitCCD(); err != nil {
  465. return fmt.Errorf("can not emit ccd: %s", err)
  466. }
  467. if err := svr.emitIptables(); err != nil {
  468. return fmt.Errorf("can not emit iptables: %s", err)
  469. }
  470. if err := svr.emitCRL(); err != nil {
  471. return fmt.Errorf("can not emit crl: %s", err)
  472. }
  473. logrus.Info("configurations emitted to the filesystem")
  474. return nil
  475. }
  476. // EmitWithRestart restarts vpnProc after calling EmitWithRestart().
  477. func (svr *Server) EmitWithRestart() error {
  478. if err := svr.Emit(); err != nil {
  479. return err
  480. }
  481. if svr.IsInitialized() {
  482. for {
  483. if vpnProc.Status() == supervisor.RUNNING || vpnProc.Status() == supervisor.STOPPED {
  484. logrus.Info("OpenVPN process is restarting")
  485. svr.RestartVPNProc()
  486. break
  487. }
  488. time.Sleep(1 * time.Second)
  489. }
  490. }
  491. return nil
  492. }
  493. // emitToFile is a proxy that calls svr.emitToFileFunc.
  494. func (svr *Server) emitToFile(path, content string, mode uint) error {
  495. return svr.emitToFileFunc(path, content, mode)
  496. }
  497. // emitToFile is an implementation for svr.emitToFileFunc.
  498. func emitToFile(path, content string, mode uint) error {
  499. // When testing don't emit files to the filesystem. Just pretend you did.
  500. if Testing {
  501. return nil
  502. }
  503. file, err := os.Create(path)
  504. if err != nil {
  505. return fmt.Errorf("Cannot create file %s: %v", path, err)
  506. }
  507. if mode != 0 {
  508. file.Chmod(os.FileMode(mode))
  509. }
  510. defer file.Close()
  511. fmt.Fprintf(file, content)
  512. return nil
  513. }
  514. func (svr *Server) emitServerConf() error {
  515. port := DefaultVPNPort
  516. if serverInstance.Port != "" {
  517. port = serverInstance.Port
  518. }
  519. proto := DefaultVPNProto
  520. if serverInstance.Proto != "" {
  521. proto = serverInstance.Proto
  522. }
  523. dns := DefaultVPNDNS
  524. if serverInstance.DNS != "" {
  525. dns = serverInstance.DNS
  526. }
  527. var result bytes.Buffer
  528. server := struct {
  529. CertPath string
  530. KeyPath string
  531. CACertPath string
  532. CAKeyPath string
  533. CCDPath string
  534. CRLPath string
  535. DHParamsPath string
  536. Net string
  537. Mask string
  538. Port string
  539. Proto string
  540. DNS string
  541. }{
  542. CertPath: _DefaultCertPath,
  543. KeyPath: _DefaultKeyPath,
  544. CACertPath: _DefaultCACertPath,
  545. CAKeyPath: _DefaultCAKeyPath,
  546. CCDPath: _DefaultVPNCCDPath,
  547. CRLPath: _DefaultCRLPath,
  548. DHParamsPath: _DefaultDHParamsPath,
  549. Net: svr.Net,
  550. Mask: svr.Mask,
  551. Port: port,
  552. Proto: proto,
  553. DNS: dns,
  554. }
  555. data, err := bindata.Asset("template/server.conf.tmpl")
  556. if err != nil {
  557. return err
  558. }
  559. t, err := template.New("server.conf").Parse(string(data))
  560. if err != nil {
  561. return fmt.Errorf("can not parse server.conf.tmpl template: %s", err)
  562. }
  563. err = t.Execute(&result, server)
  564. if err != nil {
  565. return fmt.Errorf("can not render server.conf: %s", err)
  566. }
  567. // Wite rendered content into openvpn server conf.
  568. return svr.emitToFile(_DefaultVPNConfPath, result.String(), 0)
  569. }
  570. // Refresh synchronizes the server instance from db.
  571. func (svr *Server) Refresh() error {
  572. //db = CreateDB("sqlite3", "")
  573. var dbServer dbServerModel
  574. q := db.First(&dbServer)
  575. if err := q.Error; err != nil {
  576. return fmt.Errorf("can't get server from db: %v", err)
  577. }
  578. if q.RecordNotFound() {
  579. return fmt.Errorf("server is not initialized")
  580. }
  581. svr.dbServerModel = dbServer
  582. return nil
  583. }
  584. // GetConnectedUsers will return a list of users who are currently connected
  585. // to the VPN service.
  586. func (svr *Server) GetConnectedUsers() ([]User, error) {
  587. var users []User
  588. // Open the status log file.
  589. f, err := svr.openFunc(_DefaultStatusLogPath)
  590. if err != nil {
  591. panic(err)
  592. }
  593. cl, _ := svr.parseStatusLogFunc(f) // client list from OpenVPN status log
  594. for _, c := range cl {
  595. var u dbUserModel
  596. q := db.Where(dbUserModel{Username: c.CommonName}).First(&u)
  597. if q.RecordNotFound() {
  598. logrus.WithFields(
  599. logrus.Fields{"CommonName": c.CommonName},
  600. ).Error("user not found: common name reported by the OpenVPN is not found in the database")
  601. continue
  602. }
  603. if err := q.Error; err != nil {
  604. return nil, fmt.Errorf("unknown db error: %v", err)
  605. }
  606. users = append(users, User{
  607. dbUserModel: u,
  608. isConnected: true,
  609. connectedSince: c.ConnectedSince,
  610. bytesReceived: c.BytesReceived,
  611. bytesSent: c.BytesSent,
  612. })
  613. }
  614. return users, nil
  615. }
  616. // IsInitialized checks if there is a default VPN server configured in the database or not.
  617. func (svr *Server) IsInitialized() bool {
  618. var serverModel dbServerModel
  619. q := db.First(&serverModel)
  620. if err := q.Error; err != nil {
  621. logrus.Errorf("can't retrieve server from db: %v", err)
  622. }
  623. if q.RecordNotFound() {
  624. return false
  625. }
  626. return true
  627. }
  628. func (svr *Server) emitServerKey() error {
  629. // Write rendered content into key file.
  630. return svr.emitToFile(_DefaultKeyPath, svr.Key, 0600)
  631. }
  632. func (svr *Server) emitServerCert() error {
  633. // Write rendered content into the cert file.
  634. return svr.emitToFile(_DefaultCertPath, svr.Cert, 0)
  635. }
  636. func (svr *Server) emitCRL() error {
  637. var revokedDBItems []*dbRevokedModel
  638. db.Find(&revokedDBItems)
  639. var revokedCertSerials []*big.Int
  640. for _, item := range revokedDBItems {
  641. bi := big.NewInt(0)
  642. bi.SetString(item.SerialNumber, 16)
  643. revokedCertSerials = append(revokedCertSerials, bi)
  644. }
  645. systemCA, err := svr.GetSystemCA()
  646. if err != nil {
  647. return fmt.Errorf("can not emit CRL: %v", err)
  648. }
  649. crl, err := pki.NewCRL(systemCA, revokedCertSerials...)
  650. if err != nil {
  651. return fmt.Errorf("can not emit crl: %v", err)
  652. }
  653. return svr.emitToFile(_DefaultCRLPath, crl, 0)
  654. }
  655. func (svr *Server) emitCACert() error {
  656. // Write rendered content into the ca cert file.
  657. return svr.emitToFile(_DefaultCACertPath, svr.CACert, 0)
  658. }
  659. func (svr *Server) emitCAKey() error {
  660. // Write rendered content into the ca key file.
  661. return svr.emitToFile(_DefaultCAKeyPath, svr.CAKey, 0600)
  662. }
  663. func (svr *Server) emitCCD() error {
  664. users, err := GetAllUsers()
  665. if err != nil {
  666. return err
  667. }
  668. // Filesystem related stuff. Skipping when testing.
  669. if !Testing {
  670. // Clean and then create and write rendered ccd data.
  671. err = os.RemoveAll(_DefaultVPNCCDPath)
  672. if err != nil {
  673. if os.IsNotExist(err) {
  674. } else {
  675. return err
  676. }
  677. }
  678. if _, err := os.Stat(_DefaultVPNCCDPath); err != nil {
  679. }
  680. err = os.Mkdir(_DefaultVPNCCDPath, 0755)
  681. if err != nil {
  682. if !os.IsExist(err) {
  683. return err
  684. }
  685. }
  686. }
  687. // Render ccd templates for the users.
  688. for _, user := range users {
  689. var associatedRoutes [][3]string
  690. var serverNets [][2]string
  691. for _, network := range GetAllNetworks() {
  692. switch network.Type {
  693. case ROUTE:
  694. for _, assocUsername := range network.GetAssociatedUsernames() {
  695. if assocUsername == user.Username {
  696. via := network.Via
  697. ip, mask, err := net.ParseCIDR(network.CIDR)
  698. if err != nil {
  699. return err
  700. }
  701. associatedRoutes = append(associatedRoutes, [3]string{ip.To4().String(), net.IP(mask.Mask).To4().String(), via})
  702. }
  703. }
  704. case SERVERNET:
  705. // Push associated servernets to client when client is not getting vpn server as default gw.
  706. if user.IsNoGW() {
  707. for _, assocUsername := range network.GetAssociatedUsernames() {
  708. if assocUsername == user.Username {
  709. ip, mask, err := net.ParseCIDR(network.CIDR)
  710. if err != nil {
  711. return err
  712. }
  713. serverNets = append(serverNets, [2]string{ip.To4().String(), net.IP(mask.Mask).To4().String()})
  714. }
  715. }
  716. }
  717. }
  718. }
  719. var result bytes.Buffer
  720. params := struct {
  721. IP string
  722. NetMask string
  723. Routes [][3]string // [0] is IP, [1] is Netmask, [2] is Via
  724. Servernets [][2]string // [0] is IP, [1] is Netmask
  725. RedirectGW bool
  726. }{IP: user.getIP().String(), NetMask: svr.Mask, Routes: associatedRoutes, Servernets: serverNets, RedirectGW: !user.NoGW}
  727. data, err := bindata.Asset("template/ccd.file.tmpl")
  728. if err != nil {
  729. return err
  730. }
  731. t, err := template.New("ccd.file.tmpl").Parse(string(data))
  732. if err != nil {
  733. return fmt.Errorf("can not parse ccd.file.tmpl template: %s", err)
  734. }
  735. err = t.Execute(&result, params)
  736. if err != nil {
  737. return fmt.Errorf("can not render ccd file %s: %s", user.Username, err)
  738. }
  739. if err = svr.emitToFile(filepath.Join(_DefaultVPNCCDPath, user.Username), result.String(), 0); err != nil {
  740. return err
  741. }
  742. }
  743. return nil
  744. }
  745. func (svr *Server) emitDHParams() error {
  746. var result bytes.Buffer
  747. data, err := bindata.Asset("template/dh4096.pem.tmpl")
  748. if err != nil {
  749. return err
  750. }
  751. t, err := template.New("dh4096.pem.tmpl").Parse(string(data))
  752. if err != nil {
  753. return fmt.Errorf("can not parse dh4096.pem template: %s", err)
  754. }
  755. err = t.Execute(&result, nil)
  756. if err != nil {
  757. return fmt.Errorf("can not render dh4096.pem file: %s", err)
  758. }
  759. return svr.emitToFile(_DefaultDHParamsPath, result.String(), 0)
  760. }
  761. func (svr *Server) emitIptables() error {
  762. if Testing {
  763. return nil
  764. }
  765. ipt, err := iptables.NewWithProtocol(iptables.ProtocolIPv4)
  766. if err != nil {
  767. return fmt.Errorf("can not create new iptables object: %v", err)
  768. }
  769. for _, network := range GetAllNetworks() {
  770. associatedUsernames := network.GetAssociatedUsernames()
  771. switch network.Type {
  772. case SERVERNET:
  773. users, err := GetAllUsers()
  774. if err != nil {
  775. return err
  776. }
  777. // Find associated users and emit iptables configs for the users
  778. // regarding the network's type and attributes.
  779. for _, user := range users {
  780. // Find out if the user is associated or not.
  781. var found bool
  782. for _, auser := range associatedUsernames {
  783. if user.Username == auser {
  784. found = true
  785. break
  786. }
  787. }
  788. userIP, _, err := net.ParseCIDR(user.GetIPNet())
  789. if err != nil {
  790. return err
  791. }
  792. _, networkIPNet, err := net.ParseCIDR(network.CIDR)
  793. if err != nil {
  794. return err
  795. }
  796. // get destination network's iface
  797. iface := interfaceOfIP(networkIPNet)
  798. if iface == nil {
  799. logrus.Warnf("network doesn't exist on server %s[SERVERNET]: cant find interface for %s", network.Name, networkIPNet.String())
  800. return nil
  801. }
  802. // enable nat for the user to the destination network n
  803. if found {
  804. err = ipt.AppendUnique("nat", "POSTROUTING", "-s", userIP.String(), "-o", iface.Name, "-j", "MASQUERADE")
  805. if err != nil {
  806. logrus.Error(err)
  807. return err
  808. }
  809. } else {
  810. err = ipt.Delete("nat", "POSTROUTING", "-s", userIP.String(), "-o", iface.Name, "-j", "MASQUERADE")
  811. if err != nil {
  812. logrus.Debug(err)
  813. }
  814. }
  815. }
  816. }
  817. }
  818. return nil
  819. }
  820. func checkOpenVPNExecutable() bool {
  821. executable := getOpenVPNExecutable()
  822. if executable == "" {
  823. logrus.Error("openvpn is not installed ✘")
  824. return false
  825. }
  826. logrus.Debugf("openvpn executable detected: %s ✔", executable)
  827. return true
  828. }
  829. func getOpenVPNExecutable() string {
  830. cmd := exec.Command("which", "openvpn")
  831. output, err := cmd.Output()
  832. if err != nil {
  833. logrus.Errorf("openvpn is not installed: %s ✘", err)
  834. return ""
  835. }
  836. logrus.Debugf("openvpn executable detected: %s ✔", strings.TrimSpace(string(output[:])))
  837. return strings.TrimSpace(string(output[:]))
  838. }
  839. func checkOpenSSLExecutable() bool {
  840. cmd := exec.Command("which", "openssl")
  841. output, err := cmd.Output()
  842. if err != nil {
  843. logrus.Errorf("openssl is not installed: %s ✘", err)
  844. return false
  845. }
  846. logrus.Debugf("openssl executable detected: %s ✔", strings.TrimSpace(string(output[:])))
  847. return true
  848. }
  849. func checkIptablesExecutable() bool {
  850. cmd := exec.Command("which", "iptables")
  851. output, err := cmd.Output()
  852. if err != nil {
  853. logrus.Errorf("iptables is not installed: %s ✘", err)
  854. return false
  855. }
  856. logrus.Debugf("iptables executable detected: %s ✔", strings.TrimSpace(string(output[:])))
  857. return true
  858. }
  859. func ensureBaseDir() {
  860. if Testing {
  861. return
  862. }
  863. os.Mkdir(varBasePath, 0755)
  864. }
  865. func init() {
  866. ensureBaseDir()
  867. var err error
  868. vpnProc, err = supervisor.NewProcess(getOpenVPNExecutable(), varBasePath, []string{"--config", _DefaultVPNConfPath})
  869. if err != nil {
  870. logrus.Errorf("can not create process: %v", err)
  871. }
  872. }