diff options
Diffstat (limited to 'example')
-rw-r--r-- | example/ndpiReader.c | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/example/ndpiReader.c b/example/ndpiReader.c index 83144b46e..990b0264e 100644 --- a/example/ndpiReader.c +++ b/example/ndpiReader.c @@ -1199,10 +1199,10 @@ static void printFlow(u_int16_t id, struct ndpi_flow_info *flow, u_int16_t threa } if(flow->http.url[0] != '\0') { - u_int16_t risk = ndpi_validate_url(flow->http.url); + ndpi_risk_enum risk = ndpi_validate_url(flow->http.url); if(risk != NDPI_NO_RISK) - flow->risk |= risk; + NDPI_SET_BIT(flow->risk, risk); fprintf(out, "[URL: %s[StatusCode: %u]", flow->http.url, flow->http.response_status_code); @@ -3046,13 +3046,13 @@ void test_lib() { static void bitmapUnitTest() { u_int32_t val, i, j; - for(i=0; i<16; i++) { - val = 0; + for(i=0; i<32; i++) { + NDPI_ZERO_BIT(val); NDPI_SET_BIT(val, i); assert(NDPI_ISSET_BIT(val, i)); - for(j=0; j<16; j++) { + for(j=0; j<32; j++) { if(j != i) { assert(!NDPI_ISSET_BIT(val, j)); } |