contrib/Utils/MakeUTF8.c

   1 /******************************************************************************
   2     MakeUTF8.c
   3
   4 Copyright (C) 2002 - 2006, 2013 Simon Large
   5
   6 This program is free software; you can redistribute it and/or
   7 modify it under the terms of the GNU General Public License
   8 as published by the Free Software Foundation; either version 2
   9 of the License, or (at your option) any later version.
  10
  11 This program is distributed in the hope that it will be useful,
  12 but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 GNU General Public License for more details.
  15
  16 You should have received a copy of the GNU General Public License
  17 along with this program; if not, write to the Free Software Foundation,
  18 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19
  20 Description:
  21 This program checks text files for the presence of a byte-order-mark (BOM)
  22 and for a UTF-8 encoding indicator in the XML version tag. You can also
  23 opt to add either or both of these features.
  24
  25 Use:
  26 MakeUTF8 [ -b ] [ -x ] file [ file ... ]
  27 Wildcard filenames are supported. Subdirectory recursion is not at present.
  28 -b option adds/corrects BOM in file if not already present.
  29 -x option adds/corrects XML tag if not already present.
  30 With no options, the current stateis reported but nothing is changed.
  31
  32 Example:
  33 MakeUTF8 -b *.xml tsvn_dug\*.xml
  34 Fixes BOMs (but not XML tags) in all .xml files in the current directory,
  35 and in the tsvn_dug subdirectory.
  36
  37 This program has only been built using the Microsoft Visual C++ compiler.
  38 Library calls for finding files (_findfirst64) will probably need to be
  39 changed in other environments.
  40
  41 No special compiler options were used. CL MakeUTF8.c works OK.
  42 ******************************************************************************/
  43
  44 #ifndef _CRT_SECURE_NO_WARNINGS
  45 #define _CRT_SECURE_NO_WARNINGS 1
  46 #endif
  47
  48 #include <stdio.h>
  49 #include <string.h>
  50 #include <stdlib.h>
  51 #include <io.h>
  52
  53 // Status flags returned from the file processor.
  54 #define ADD_BOM     1       // BOM is missing
  55 #define DOUBLE_BOM  2       // Double BOM found
  56 #define XML_TAG     4       // XML tag missing, or UTF-8 not included
  57 #define FIXED_BOM   64      // BOM has been added or fixed
  58 #define FIXED_TAG   128     // XML tag has been added or fixed
  59
  60 char *help =
  61     "MakeUTF8     Version 1.1\n"
  62     "Add UTF-8 byte-order-mark and XML-tag to start of text file.\n\n"
  63     "Use: MakeUTF8 [ -b ] [ -x ] file [ file ... ]\n"
  64     "     -b option adds/corrects BOM in file if not already present\n"
  65     "     -x option adds/corrects XML tag if not already present\n"
  66     "     With no options, just report current state\n\n";
  67
  68 int ProcessFile(const char *FName, const char *TName, int Action);
  69
  70 int main(int argc, char *argv[])
  71 {
  72     int n, Action = 0, Result = 0;
  73     char Path[_MAX_PATH], Temp[_MAX_PATH];
  74     char *FName;
  75     struct __finddata64_t FileInfo;
  76     intptr_t hFile;
  77
  78     if (argc < 2)
  79     {
  80         fprintf(stderr, "%s", help);
  81         exit(0);
  82     }
  83
  84     for (n = 1; n < argc; n++)
  85     {
  86         if (_stricmp(argv[n], "-b") == 0)
  87         {
  88             Action |= ADD_BOM | DOUBLE_BOM;
  89             continue;
  90         }
  91         if (_stricmp(argv[n], "-x") == 0)
  92         {
  93             Action |= XML_TAG;
  94             continue;
  95         }
  96         // Unscramble wildcard filenames
  97         if ((hFile = _findfirst64(argv[n], &FileInfo)) != -1)
  98         {
  99             printf("BOM\tXML-tag\tFile\n");
 100             printf("--------------------\n");
 101             // Extract path from original argument.
 102             strcpy(Path, argv[n]);
 103             // Set FName to point to filename portion of path
 104             FName = strrchr(Path, '\\');
 105             if (FName == NULL)
 106             {
 107                 FName = strrchr(Path, '/');
 108             }
 109             if (FName == NULL)
 110             {
 111                 FName = strrchr(Path, ':');
 112             }
 113             if (FName == NULL)
 114             {
 115                 FName = Path;
 116             }
 117             else
 118             {
 119                 ++FName;
 120             }
 121
 122             // Process all matching files.
 123             do
 124             {
 125                 if (!(FileInfo.attrib & _A_SUBDIR))
 126                 {
 127                     // Append filename to path
 128                     char *p;
 129                     strcpy(FName, FileInfo.name);
 130                     // Create temp filename by replacing extension with $$$
 131                     strcpy(Temp, Path);
 132                     p = strrchr(Temp, '.');
 133                     if (p != NULL)
 134                     {
 135                         *p = '\0';      // Trim off extension
 136                     }
 137                     strcat(Temp, ".$$$");
 138                     Result = ProcessFile(Path, Temp, Action);
 139                     if (Result < 0)
 140                     {
 141                         break;          // Failed.
 142                     }
 143                     // Show results of analysis / repair
 144                     if (Result & ADD_BOM)
 145                     {
 146                         if (Result & FIXED_BOM)
 147                         {
 148                             p = "Added";
 149                         }
 150                         else
 151                         {
 152                             p = "None";
 153                         }
 154                     }
 155                     else if (Result & DOUBLE_BOM)
 156                     {
 157                         if (Result & FIXED_BOM)
 158                         {
 159                             p = "Fixed";
 160                         }
 161                         else
 162                         {
 163                             p = "Multi";
 164                         }
 165                     }
 166                     else
 167                     {
 168                         p = "OK";
 169                     }
 170                     printf("%s\t", p);
 171                     if (Result & XML_TAG)
 172                     {
 173                         if (Result & FIXED_TAG)
 174                         {
 175                             p = "Fixed";
 176                         }
 177                         else
 178                         {
 179                             p = "None";
 180                         }
 181                     }
 182                     else
 183                     {
 184                         p = "OK";
 185                     }
 186                     printf("%s\t%s\n", p, FileInfo.name);
 187                 }
 188             }
 189             while (_findnext64(hFile, &FileInfo) == 0);
 190             _findclose(hFile);
 191         }
 192     }
 193     exit((Result < 0) ? 1 : 0);
 194 }
 195
 196 // These 3 bytes are the BOM we want
 197 char BOMbuf[3] = { 0xef, 0xbb, 0xbf };
 198
 199 // This is the XML tag we want
 200 char *UTFtag = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>";
 201
 202 // Read this amount at start of file to check for BOM and tag
 203 #define BUFSIZE 2048
 204
 205 int ProcessFile(const char *FName, const char *TName, int Action)
 206 {
 207     FILE *fp, *fpout;
 208     char Buffer[BUFSIZE + 1024];
 209     size_t Len;
 210     size_t NumRead;
 211     int Changed = 0, Checked = 0;
 212     size_t UTFtaglen;
 213     char *TagStart, *TagStop;
 214     char *AfterBOM = Buffer;
 215
 216     if ((fp = fopen(FName, "r")) == NULL)
 217     {
 218         return -1;
 219     }
 220
 221     // Check if output file exists already
 222     if ((fpout = fopen(TName, "r")) != NULL) {
 223         fprintf(stderr, "%s:\tTemp file already exists\n", TName);
 224         fclose(fpout);
 225         fclose(fp);
 226         return -1;
 227     }
 228
 229     while ((NumRead = fread(Buffer, 1, BUFSIZE, fp)) > 0)
 230     {
 231         if (!Checked)
 232         {
 233             Checked = 1;
 234             // Check for no BOM or multiple BOM.
 235             if (memcmp(BOMbuf, Buffer, 3) == 0)
 236             {
 237                 // BOM already exists.
 238                 AfterBOM = Buffer + 3;
 239                 while (memcmp(BOMbuf, AfterBOM, 3) == 0)
 240                 {
 241                     // Multiple BOM found.
 242                     Changed |= DOUBLE_BOM;
 243                     if (Action & DOUBLE_BOM)
 244                     {
 245                         // Delete BOM from source
 246                         NumRead -= 3;
 247                         memmove(Buffer, AfterBOM, NumRead);
 248                         Buffer[NumRead] = '\0';
 249                     }
 250                     else
 251                     {
 252                         break;
 253                     }
 254                 }
 255             }
 256             else
 257             {
 258                 // No BOM found.
 259                 Changed |= ADD_BOM;
 260                 if (Action & ADD_BOM)
 261                 {
 262                     // Add BOM to source
 263                     AfterBOM = Buffer + 3;
 264                     memmove(AfterBOM, Buffer, NumRead);
 265                     memcpy(Buffer, BOMbuf, 3);
 266                     NumRead += 3;
 267                 }
 268             }
 269
 270             // Check for XML tag <?xml version="1.0" encoding="UTF-8"?>
 271             Buffer[NumRead] = '\0';     // Add null terminator for string search.
 272             UTFtaglen = strlen(UTFtag);
 273             if (strstr(Buffer, "encoding=\"UTF-8\"") == NULL)
 274             {
 275                 // No XML tag found.
 276                 Changed |= XML_TAG;
 277                 if (Action & XML_TAG)
 278                 {
 279                     TagStart = strstr(Buffer, "<?xml version");
 280                     if (TagStart != NULL)
 281                     {
 282                         TagStop = strstr(TagStart, "?>");
 283                         if (TagStop != NULL)
 284                         {
 285                             // Version tag present without UTF-8
 286                             Len = UTFtaglen - (TagStop - TagStart + 2);
 287                             if (Len != 0)
 288                             {
 289                                 // Expand/contract the space
 290                                 memmove(TagStop + Len, TagStop, NumRead - (TagStop - Buffer));
 291                                 NumRead += Len;
 292                             }
 293                             memcpy(TagStart, UTFtag, UTFtaglen);
 294                         }
 295                         else
 296                         {
 297                             // Version tag is not terminated. Cannot fix.
 298                             Action &= ~XML_TAG;
 299                         }
 300                     }
 301                     else
 302                     {
 303                         // No version tag found. Add one after BOM, with newline.
 304                         memmove(AfterBOM + UTFtaglen + 1, AfterBOM, NumRead);
 305                         memcpy(AfterBOM, UTFtag, UTFtaglen);
 306                         AfterBOM[UTFtaglen] = '\n';
 307                         NumRead += UTFtaglen + 1;
 308                     }
 309                 }
 310             }
 311
 312             if (!(Action & Changed))
 313             {
 314                 // If no problems marked for fixing, leave it here.
 315                 break;
 316             }
 317             // Changes made - open a temp file for the BOM'ed version
 318             if ((fpout = fopen(TName, "w")) == NULL)
 319             {
 320                 fprintf(stderr, "Cannot open temp file\n");
 321                 fclose(fp);
 322                 return -1;
 323             }
 324         }
 325         if (fwrite(Buffer, 1, NumRead, fpout) != NumRead)
 326         {
 327             fprintf(stderr, "Error writing to temp file\n");
 328             fclose(fpout);
 329             fclose(fp);
 330             return -1;
 331         }
 332     }
 333
 334     fclose(fp);
 335
 336     // If changes have been made, replace original file with temp file.
 337     if (Changed & Action)
 338     {
 339         // Replace original with temp file
 340         fclose(fpout);
 341         if (remove(FName) != 0)
 342         {
 343             fprintf(stderr, "Cannot delete original file\n");
 344             return -1;
 345         }
 346         if (rename(TName, FName) != 0)
 347         {
 348             fprintf(stderr, "Cannot replace original file with fixed version\n");
 349             return -1;
 350         }
 351         // Add flags to indicate what we have actually fixed
 352         if (Changed & Action & (DOUBLE_BOM | ADD_BOM))
 353         {
 354             Changed |= FIXED_BOM;
 355         }
 356         if (Changed & Action & XML_TAG)
 357         {
 358             Changed |= FIXED_TAG;
 359         }
 360     }
 361
 362     return Changed;
 363 }