upstream/mercurial-mirror Files · contrib/fuzz/fuzzutil.h

tests: change how sockets are closed...

tests: change how sockets are closed Python 3 uses a different type to represent a socket file object than Python 2. We need to conditionalize how the socket is closed accordingly. While we're here, we switch to use socket.shutdown() to close the socket. This is because socket.close() may not actually close the socket until it is GCd. socket.shutdown() forces an immediate shutdown. I suspect Python 3 changed semantic behavior here, as I can't get test-http-bad-server.t to work with socket.close(). socket.shutdown() does appear to work, however. Differential Revision: https://phab.mercurial-scm.org/D5751

Augie Fackler - - Load All Authors

File last commit:

r38192:36d55f90 default


                r41605:d343d9ac

default

Download file

             fuzzutil.h
        
                    47 lines
            
             | 1.2 KiB
            
                | text/x-c
            
             |
                CLexer
            
             / contrib / fuzz / fuzzutil.h
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      #ifndef CONTRIB_FUZZ_FUZZUTIL_H

      #define CONTRIB_FUZZ_FUZZUTIL_H

      #include <iostream>

      #include <memory>

      #include <stdint.h>

      /* Try and use std::optional, but failing that assume we'll have a

       * workable https://abseil.io/ install on the include path to get

       * their backport of std::optional. */

      #ifdef __has_include

      #if __has_include(<optional>) && __cplusplus >= 201703L

      #include <optional>

      #define CONTRIB_FUZZ_HAVE_STD_OPTIONAL

      #endif

      #endif

      #ifdef CONTRIB_FUZZ_HAVE_STD_OPTIONAL

      namespace contrib

      {

      using std::nullopt;

      using std::optional;

      } /* namespace contrib */

      #else

      #include "third_party/absl/types/optional.h"

      namespace contrib

      {

      using absl::nullopt;

      using absl::optional;

      } /* namespace contrib */

      #endif

      /* set DEBUG to 1 for a few debugging prints, or 2 for a lot */

      #define DEBUG 0

      #define LOG(level)                                                             \

      	if (level <= DEBUG)                                                    \

      	std::cout

      struct two_inputs {

      	std::unique_ptr<char[]> right;

      	size_t right_size;

      	std::unique_ptr<char[]> left;

      	size_t left_size;

      };

      /* Split a non-zero-length input into two inputs. */

      contrib::optional<two_inputs> SplitInputs(const uint8_t *Data, size_t Size);

      #endif /* CONTRIB_FUZZ_FUZZUTIL_H */

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

				#ifndef CONTRIB_FUZZ_FUZZUTIL_H
				#define CONTRIB_FUZZ_FUZZUTIL_H
				#include <iostream>
				#include <memory>
				#include <stdint.h>

				/* Try and use std::optional, but failing that assume we'll have a
				* workable https://abseil.io/ install on the include path to get
				* their backport of std::optional. */
				#ifdef __has_include
				#if __has_include(<optional>) && __cplusplus >= 201703L
				#include <optional>
				#define CONTRIB_FUZZ_HAVE_STD_OPTIONAL
				#endif
				#endif
				#ifdef CONTRIB_FUZZ_HAVE_STD_OPTIONAL
				namespace contrib
				{
				using std::nullopt;
				using std::optional;
				} /* namespace contrib */
				#else
				#include "third_party/absl/types/optional.h"
				namespace contrib
				{
				using absl::nullopt;
				using absl::optional;
				} /* namespace contrib */
				#endif

				/* set DEBUG to 1 for a few debugging prints, or 2 for a lot */
				#define DEBUG 0
				#define LOG(level) \
				if (level <= DEBUG) \
				std::cout

				struct two_inputs {
				std::unique_ptr<char[]> right;
				size_t right_size;
				std::unique_ptr<char[]> left;
				size_t left_size;
				};

				/* Split a non-zero-length input into two inputs. */
				contrib::optional<two_inputs> SplitInputs(const uint8_t *Data, size_t Size);

				#endif /* CONTRIB_FUZZ_FUZZUTIL_H */