func main() { flag.Parse() var clientConfig ClientDomainVerificationTestConfigFile err := utils.ReadConfigFile(configFilePath, &clientConfig) config.ShelterConfig = clientConfig.Config if err == utils.ErrConfigFileUndefined { fmt.Println(err.Error()) fmt.Println("Usage:") flag.PrintDefaults() return } else if err != nil { utils.Fatalln("Error reading configuration file", err) } utils.StartDNSServer(clientConfig.DNSServerPort, clientConfig.Scan.UDPMaxSize) finishRESTServer := utils.StartRESTServer() defer finishRESTServer() utils.StartWebClient() scanDomain() queryDomain() utils.Println("SUCCESS!") }
func main() { flag.Parse() err := utils.ReadConfigFile(configFilePath, &config.ShelterConfig) if err == utils.ErrConfigFileUndefined { fmt.Println(err.Error()) fmt.Println("Usage:") flag.PrintDefaults() return } else if err != nil { utils.Fatalln("Error reading configuration file", err) } finishRESTServer := utils.StartRESTServer() defer finishRESTServer() utils.StartWebClient() database, databaseSession, err := mongodb.Open( config.ShelterConfig.Database.URIs, config.ShelterConfig.Database.Name, config.ShelterConfig.Database.Auth.Enabled, config.ShelterConfig.Database.Auth.Username, config.ShelterConfig.Database.Auth.Password, ) if err != nil { utils.Fatalln("Error connecting the database", err) } defer databaseSession.Close() // If there was some problem in the last test, there could be some data in the // database, so let's clear it to don't affect this test. We avoid checking the error, // because if the collection does not exist yet, it will be created in the first // insert utils.ClearDatabase(database) createScans(database) retrieveScans() retrieveScansWithPagination() retrieveScansWithCache() retrieveCurrentScan() utils.Println("SUCCESS!") }