105 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			C
		
	
	
	
			
		
		
	
	
			105 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			C
		
	
	
	
| /* Copyright 2021 Google LLC
 | |
| Licensed under the Apache License, Version 2.0 (the "License");
 | |
| you may not use this file except in compliance with the License.
 | |
| You may obtain a copy of the License at
 | |
|       http://www.apache.org/licenses/LICENSE-2.0
 | |
| Unless required by applicable law or agreed to in writing, software
 | |
| distributed under the License is distributed on an "AS IS" BASIS,
 | |
| WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | |
| See the License for the specific language governing permissions and
 | |
| limitations under the License.
 | |
| */
 | |
| 
 | |
| #include "config.h"
 | |
| #include "syshead.h"
 | |
| #include "init.h"
 | |
| #include "packet_id.h"
 | |
| 
 | |
| #include "fuzz_randomizer.h"
 | |
| 
 | |
| int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
 | |
|   fuzz_random_init(data, size);
 | |
| 
 | |
|   struct packet_id pid;
 | |
|   struct packet_id_net pin;
 | |
|   const int seq_backtrack = 10;
 | |
|   const int time_backtrack = 10;
 | |
| 
 | |
|   packet_id_init(&pid, seq_backtrack, time_backtrack, "name", 0);
 | |
| 
 | |
|   int total_sends = fuzz_randomizer_get_int(0, 10);
 | |
|   for (int i = 0; i < total_sends; i++) {
 | |
|     update_time();
 | |
|     pin.time = fuzz_randomizer_get_int(0, 0xfffffff);
 | |
|     pin.id = fuzz_randomizer_get_int(0, 0xfffffff);
 | |
| 
 | |
|     packet_id_reap_test(&pid.rec);
 | |
|     bool test = packet_id_test(&pid.rec, &pin);
 | |
|     if (test) {
 | |
|       packet_id_add(&pid.rec, &pin);
 | |
|     }
 | |
|   }
 | |
|   packet_id_free(&pid);
 | |
| 
 | |
|   // packet id send
 | |
|   char *tmp2 = get_random_string();
 | |
|   if (strlen(tmp2) > sizeof(struct packet_id_send)) {
 | |
|     struct packet_id_send pidsend;
 | |
|     memcmp(&pidsend, tmp2, sizeof(struct packet_id_send));
 | |
| 
 | |
|     struct timeval tv;
 | |
|     tv.tv_sec = pidsend.time;
 | |
|     tv.tv_usec = 0;
 | |
|     if (localtime(&tv)) {
 | |
|       struct buffer iv_buffer;
 | |
|       buf_set_write(&iv_buffer, tmp2, strlen(tmp2));
 | |
|       packet_id_write(&pidsend, &iv_buffer, false, false);
 | |
|       packet_id_write(&pidsend, &iv_buffer, false, true);
 | |
|       packet_id_write(&pidsend, &iv_buffer, true, true);
 | |
|       packet_id_write(&pidsend, &iv_buffer, true, false);
 | |
|     }
 | |
|   }
 | |
|   free(tmp2);
 | |
| 
 | |
|   struct gc_arena gc;
 | |
|   gc = gc_new();
 | |
|   struct buffer buf;
 | |
|   char *tmp = get_random_string();
 | |
|   buf = string_alloc_buf(tmp, &gc);
 | |
|   free(tmp);
 | |
|   packet_id_read(&pid, &buf, false);
 | |
|   packet_id_read(&pid, &buf, true);
 | |
|   gc_free(&gc);
 | |
| 
 | |
| 	char filename[256];
 | |
| 	sprintf(filename, "/tmp/libfuzzer.%d", getpid());
 | |
| 
 | |
| 	FILE *fp = fopen(filename, "wb");
 | |
| 	if (!fp) {
 | |
|     return 0;
 | |
| 	}
 | |
| 	fwrite(data, size, 1, fp);
 | |
| 	fclose(fp);
 | |
|  
 | |
|   struct packet_id_persist p;
 | |
|   memset(&p, 0, sizeof(struct packet_id_persist));
 | |
|   packet_id_persist_init(&p);
 | |
|   packet_id_persist_load(&p, filename);
 | |
|   //p.time = NULL;
 | |
|   struct timeval tv;
 | |
|   tv.tv_sec = p.time;
 | |
|   tv.tv_usec = 0;
 | |
|   if (localtime(&tv) != NULL) {
 | |
|     gc = gc_new();
 | |
|     p.id_last_written = fuzz_randomizer_get_int(0, 0xfffffff);
 | |
|     //packet_id_persist_print(&p, &gc);
 | |
|     packet_id_persist_save(&p);
 | |
|     gc_free(&gc);
 | |
|   }
 | |
| 
 | |
|   packet_id_persist_close(&p); 
 | |
| 
 | |
|   fuzz_random_destroy();
 | |
|   return 0;
 | |
| }
 |