Skip to content

Commit 151b90f

Browse files
authored
Merge pull request #2388 from greenbone/refactor-nvt-loading
Change: Refactor nvt loading to separate ospd and openvasd logic
2 parents 1e19ca4 + c8d083a commit 151b90f

13 files changed

+2572
-2168
lines changed

src/CMakeLists.txt

+24-6
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,10 @@ add_executable (manage-utils-test
136136
gmp_port_lists.c
137137
gmp_report_configs.c
138138
gmp_report_formats.c gmp_tickets.c
139-
gmp_tls_certificates.c)
139+
gmp_tls_certificates.c
140+
manage_sql_nvts_osp.c
141+
manage_sql_nvts_openvasd.c
142+
manage_sql_nvts_common.c)
140143

141144
add_test (manage-utils-test manage-utils-test)
142145

@@ -168,7 +171,10 @@ add_executable (manage-test
168171
gmp_port_lists.c
169172
gmp_report_configs.c
170173
gmp_report_formats.c gmp_tickets.c
171-
gmp_tls_certificates.c)
174+
gmp_tls_certificates.c
175+
manage_sql_nvts_osp.c
176+
manage_sql_nvts_openvasd.c
177+
manage_sql_nvts_common.c)
172178

173179
add_test (manage-test manage-test)
174180

@@ -200,7 +206,10 @@ add_executable (manage-sql-test
200206
gmp_port_lists.c
201207
gmp_report_configs.c
202208
gmp_report_formats.c gmp_tickets.c
203-
gmp_tls_certificates.c)
209+
gmp_tls_certificates.c
210+
manage_sql_nvts_osp.c
211+
manage_sql_nvts_openvasd.c
212+
manage_sql_nvts_common.c)
204213

205214
add_test (manage-sql-test manage-sql-test)
206215

@@ -232,7 +241,10 @@ add_executable (gmp-tickets-test
232241
gmp_port_lists.c
233242
gmp_report_configs.c
234243
gmp_report_formats.c
235-
gmp_tls_certificates.c)
244+
gmp_tls_certificates.c
245+
manage_sql_nvts_osp.c
246+
manage_sql_nvts_openvasd.c
247+
manage_sql_nvts_common.c)
236248

237249
add_test (gmp-tickets-test gmp-tickets-test)
238250

@@ -264,7 +276,10 @@ add_executable (utils-test
264276
gmp_port_lists.c
265277
gmp_report_configs.c
266278
gmp_report_formats.c gmp_tickets.c
267-
gmp_tls_certificates.c)
279+
gmp_tls_certificates.c
280+
manage_sql_nvts_osp.c
281+
manage_sql_nvts_openvasd.c
282+
manage_sql_nvts_common.c)
268283

269284
add_test (utils-test utils-test)
270285

@@ -313,7 +328,10 @@ add_executable (gvmd
313328
gmp_port_lists.c
314329
gmp_report_configs.c
315330
gmp_report_formats.c gmp_tickets.c
316-
gmp_tls_certificates.c)
331+
gmp_tls_certificates.c
332+
manage_sql_nvts_osp.c
333+
manage_sql_nvts_openvasd.c
334+
manage_sql_nvts_common.c)
317335

318336
target_link_libraries (gvmd m
319337
${GNUTLS_LDFLAGS} ${GPGME_LDFLAGS} ${CMAKE_THREAD_LIBS_INIT} ${LINKER_HARDENING_FLAGS} ${LINKER_DEBUG_FLAGS}

src/gvmd.c

+3-3
Original file line numberDiff line numberDiff line change
@@ -1104,14 +1104,14 @@ handle_sigabrt_simple (int signal)
11041104
* @return 0 success, -1 error, 1 VT integrity check failed.
11051105
*/
11061106
static int
1107-
update_nvt_cache_osp (const gchar *update_socket)
1107+
update_nvt_cache (const gchar *update_socket)
11081108
{
11091109
#ifdef OPENVASD
11101110
setproctitle ("openvasd: Updating NVT cache");
11111111
#else
11121112
setproctitle ("OSP: Updating NVT cache");
11131113
#endif
1114-
return manage_update_nvts_osp (update_socket);
1114+
return manage_update_nvts (update_socket);
11151115
}
11161116

11171117

@@ -1157,7 +1157,7 @@ update_nvt_cache_retry ()
11571157
{
11581158
int ret;
11591159

1160-
ret = update_nvt_cache_osp (osp_update_socket);
1160+
ret = update_nvt_cache (osp_update_socket);
11611161
if (ret == 1)
11621162
{
11631163
g_message ("Rebuilding all NVTs because of a hash value mismatch");

src/manage.c

+2-48
Original file line numberDiff line numberDiff line change
@@ -7549,52 +7549,6 @@ nvts_feed_info_internal (const gchar *update_socket,
75497549
}
75507550
#endif
75517551

7552-
#if OPENVASD
7553-
/**
7554-
* @brief Get VTs feed information from a scanner.
7555-
*
7556-
* @param[in] scanner_uuid The uuid of the scanner to be used.
7557-
* @param[out] vts_version Output of scanner feed version.
7558-
*
7559-
* @return 0 success, 1 connection to scanner failed, 2 scanner still starting,
7560-
* -1 other error.
7561-
*/
7562-
static int
7563-
nvts_feed_info_internal_from_openvasd (const gchar *scanner_uuid,
7564-
gchar **vts_version)
7565-
{
7566-
scanner_t scan;
7567-
openvasd_connector_t connector = NULL;
7568-
openvasd_resp_t resp = NULL;
7569-
int ret;
7570-
if (find_resource_no_acl ("scanner", scanner_uuid, &scan))
7571-
return -1;
7572-
7573-
connector = openvasd_scanner_connect (scan, NULL);
7574-
if (!connector)
7575-
return 1;
7576-
7577-
resp = openvasd_get_health_ready (connector);
7578-
if (resp->code == -1)
7579-
{
7580-
g_warning ("%s: failed to connect to %s:%d", __func__,
7581-
scanner_host (scan), scanner_port (scan));
7582-
ret = 1;
7583-
}
7584-
else if (resp->code == 503)
7585-
ret = 2;
7586-
else
7587-
{
7588-
*vts_version = g_strdup (resp->header);
7589-
ret = 0;
7590-
}
7591-
7592-
openvasd_response_cleanup (resp);
7593-
openvasd_connector_free (connector);
7594-
return ret;
7595-
}
7596-
#endif
7597-
75987552
/**
75997553
* @brief Get VTs feed information from the scanner using VT update socket.
76007554
*
@@ -7753,9 +7707,9 @@ gvm_migrate_secinfo (int feed_type)
77537707
* @return 0 success, -1 error, 1 VT integrity check failed.
77547708
*/
77557709
int
7756-
manage_update_nvts_osp (const gchar *update_socket)
7710+
manage_update_nvts (const gchar *update_socket)
77577711
{
7758-
return manage_update_nvt_cache_osp (update_socket);
7712+
return manage_update_nvt_cache (update_socket);
77597713
}
77607714

77617715

src/manage.h

+1-1
Original file line numberDiff line numberDiff line change
@@ -4066,7 +4066,7 @@ int
40664066
nvts_check_feed (int *, int *, gchar **);
40674067

40684068
int
4069-
manage_update_nvts_osp (const gchar *);
4069+
manage_update_nvts (const gchar *);
40704070

40714071
int
40724072
manage_rebuild (GSList *, const db_conn_info_t *);

src/manage_sql_configs.h

+3
Original file line numberDiff line numberDiff line change
@@ -102,4 +102,7 @@ check_db_configs (int);
102102
void
103103
check_whole_only_in_configs ();
104104

105+
int
106+
check_config_families ();
107+
105108
#endif /* not _GVMD_MANAGE_SQL_CONFIGS_H */

0 commit comments

Comments
 (0)