usr.bin/sort/file.h

   1 /*      $FreeBSD$       */
   2
   3 /*-
   4  * Copyright (C) 2009 Gabor Kovesdan <gabor@FreeBSD.org>
   5  * Copyright (C) 2012 Oleg Moskalenko <mom040267@gmail.com>
   6  * All rights reserved.
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions
  10  * are met:
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  13  * 2. Redistributions in binary form must reproduce the above copyright
  14  *    notice, this list of conditions and the following disclaimer in the
  15  *    documentation and/or other materials provided with the distribution.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  27  * SUCH DAMAGE.
  28  */
  29
  30 #if !defined(__SORT_FILE_H__)
  31 #define __SORT_FILE_H__
  32
  33 #include "coll.h"
  34 #include "sort.h"
  35
  36 #define SORT_DEFAULT    0
  37 #define SORT_QSORT      1
  38 #define SORT_MERGESORT  2
  39 #define SORT_HEAPSORT   3
  40 #define SORT_RADIXSORT  4
  41
  42 #define DEFAULT_SORT_ALGORITHM SORT_HEAPSORT
  43 #define DEFAULT_SORT_FUNC heapsort
  44
  45 /*
  46  * List of data to be sorted.
  47  */
  48 struct sort_list
  49 {
  50         struct sort_list_item   **list;
  51         unsigned long long       memsize;
  52         size_t                   count;
  53         size_t                   size;
  54         size_t                   sub_list_pos;
  55 };
  56
  57 /*
  58  * File reader object
  59  */
  60 struct file_reader;
  61
  62 /*
  63  * List of files to be sorted
  64  */
  65 struct file_list
  66 {
  67         char                    **fns;
  68         size_t                   count;
  69         size_t                   sz;
  70         bool                     tmp;
  71 };
  72
  73 /*
  74  * Structure for zero-separated file reading (for input files list)
  75  */
  76 struct file0_reader
  77 {
  78         char                    *current_line;
  79         FILE                    *f;
  80         size_t                   current_sz;
  81 };
  82
  83 /* memory */
  84
  85 /**/
  86
  87 extern unsigned long long free_memory;
  88 extern unsigned long long available_free_memory;
  89
  90 /* Are we using mmap ? */
  91 extern bool use_mmap;
  92
  93 /* temporary file dir */
  94
  95 extern const char *tmpdir;
  96
  97 /*
  98  * Max number of simultaneously open files (including the output file).
  99  */
 100 extern size_t max_open_files;
 101
 102 /*
 103  * Compress program
 104  */
 105 extern const char* compress_program;
 106
 107 /* funcs */
 108
 109 struct file_reader *file_reader_init(const char *fsrc);
 110 struct bwstring *file_reader_readline(struct file_reader *fr);
 111 void file_reader_free(struct file_reader *fr);
 112
 113 char *read_file0_line(struct file0_reader *f0r);
 114
 115 void init_tmp_files(void);
 116 void clear_tmp_files(void);
 117 char *new_tmp_file_name(void);
 118 void tmp_file_atexit(const char *tmp_file);
 119
 120 void file_list_init(struct file_list *fl, bool tmp);
 121 void file_list_add(struct file_list *fl, char *fn, bool allocate);
 122 void file_list_populate(struct file_list *fl, int argc, char **argv, bool allocate);
 123 void file_list_clean(struct file_list *fl);
 124
 125 int check(const char *);
 126 void merge_files(struct file_list *fl, const char *fn_out);
 127 FILE *openfile(const char *, const char *);
 128 void closefile(FILE *, const char *);
 129 int procfile(const char *fn, struct sort_list *list, struct file_list *fl);
 130
 131 void sort_list_init(struct sort_list *l);
 132 void sort_list_add(struct sort_list *l, struct bwstring *str);
 133 void sort_list_clean(struct sort_list *l);
 134 void sort_list_dump(struct sort_list *l, const char *fn);
 135
 136 void sort_list_to_file(struct sort_list *list, const char *outfile);
 137
 138 #endif /* __SORT_FILE_H__ */