/* $NetBSD: udf_osta.c,v 1.11 2022/02/11 16:33:18 reinoud Exp $ */
#if HAVE_NBTOOL_CONFIG_H
#include "nbtool_config.h"
#endif

#include <sys/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD: udf_osta.c,v 1.11 2022/02/11 16:33:18 reinoud Exp $");

/*
* Various routines from the OSTA 2.01 specs.  Copyrights are included with
* each code segment.  Slight whitespace modifications have been made for
* formatting purposes.  Typos/bugs have been fixed.
*
*/

#include "udf_osta.h"

#ifndef _KERNEL
#include <ctype.h>
#endif

/*****************************************************************************/
/***********************************************************************
* OSTA compliant Unicode compression, uncompression routines.
* Copyright 1995 Micro Design International, Inc.
* Written by Jason M. Rinn.
* Micro Design International gives permission for the free use of the
* following source code.
*/

/***********************************************************************
* Takes an OSTA CS0 compressed unicode name, and converts
* it to Unicode.
* The Unicode output will be in the byte order
* that the local compiler uses for 16-bit values.
* NOTE: This routine only performs error checking on the compID.
* It is up to the user to ensure that the unicode buffer is large
* enough, and that the compressed unicode name is correct.
*
* RETURN VALUE
*
* The number of unicode characters which were uncompressed.
* A -1 is returned if the compression ID is invalid.
*/
int
udf_UncompressUnicode(
       int numberOfBytes,      /* (Input) number of bytes read from media. */
       byte *UDFCompressed,    /* (Input) bytes read from media. */
       unicode_t *unicode)     /* (Output) uncompressed unicode characters. */
{
       unsigned int compID;
       int returnValue, unicodeIndex, byteIndex;

       /* Use UDFCompressed to store current byte being read. */
       compID = UDFCompressed[0];

       /* Translate 254/255 compID values used for deleted entries */
       if (compID == 254)
               compID = 8;
       if (compID == 255)
               compID = 16;

       /* First check for valid compID. */
       if (compID != 8 && compID != 16) {
               returnValue = -1;
       } else {
               unicodeIndex = 0;
               byteIndex = 1;

               /* Loop through all the bytes. */
               while (byteIndex < numberOfBytes) {
                       if (compID == 16) {
                               /* Move the first byte to the high bits of the
                                * unicode char.
                                */
                               unicode[unicodeIndex] =
                                   UDFCompressed[byteIndex++] << 8;
                       } else {
                               unicode[unicodeIndex] = 0;
                       }
                       if (byteIndex < numberOfBytes) {
                               /*Then the next byte to the low bits. */
                               unicode[unicodeIndex] |=
                                   UDFCompressed[byteIndex++];
                       }
                       unicodeIndex++;
               }
               returnValue = unicodeIndex;
       }
       return(returnValue);
}

/***********************************************************************
* DESCRIPTION:
* Takes a string of unicode wide characters and returns an OSTA CS0
* compressed unicode string. The unicode MUST be in the byte order of
* the compiler in order to obtain correct results. Returns an error
* if the compression ID is invalid.
*
* NOTE: This routine assumes the implementation already knows, by
* the local environment, how many bits are appropriate and
* therefore does no checking to test if the input characters fit
* into that number of bits or not.
*
* RETURN VALUE
*
* The total number of bytes in the compressed OSTA CS0 string,
* including the compression ID.
* A -1 is returned if the compression ID is invalid.
*/
int
udf_CompressUnicode(
       int numberOfChars,      /* (Input) number of unicode characters. */
       int compID,             /* (Input) compression ID to be used. */
       unicode_t *unicode,     /* (Input) unicode characters to compress. */
       byte *UDFCompressed)    /* (Output) compressed string, as bytes. */
{
       int byteIndex, unicodeIndex;

       if (compID != 8 && compID != 16) {
               byteIndex = -1; /* Unsupported compression ID ! */
       } else {
               /* Place compression code in first byte. */
               UDFCompressed[0] = compID;

               byteIndex = 1;
               unicodeIndex = 0;
               while (unicodeIndex < numberOfChars) {
                       if (compID == 16) {
                               /* First, place the high bits of the char
                                * into the byte stream.
                                */
                               UDFCompressed[byteIndex++] =
                                   (unicode[unicodeIndex] & 0xFF00) >> 8;
                       }
                       /*Then place the low bits into the stream. */
                       UDFCompressed[byteIndex++] =
                           unicode[unicodeIndex] & 0x00FF;
                       unicodeIndex++;
               }
       }
       return(byteIndex);
}

/*****************************************************************************/
/*
* CRC 010041
*/
static unsigned short crc_table[256] = {
       0x0000, 0x1021, 0x2042, 0x3063, 0x4084, 0x50A5, 0x60C6, 0x70E7,
       0x8108, 0x9129, 0xA14A, 0xB16B, 0xC18C, 0xD1AD, 0xE1CE, 0xF1EF,
       0x1231, 0x0210, 0x3273, 0x2252, 0x52B5, 0x4294, 0x72F7, 0x62D6,
       0x9339, 0x8318, 0xB37B, 0xA35A, 0xD3BD, 0xC39C, 0xF3FF, 0xE3DE,
       0x2462, 0x3443, 0x0420, 0x1401, 0x64E6, 0x74C7, 0x44A4, 0x5485,
       0xA56A, 0xB54B, 0x8528, 0x9509, 0xE5EE, 0xF5CF, 0xC5AC, 0xD58D,
       0x3653, 0x2672, 0x1611, 0x0630, 0x76D7, 0x66F6, 0x5695, 0x46B4,
       0xB75B, 0xA77A, 0x9719, 0x8738, 0xF7DF, 0xE7FE, 0xD79D, 0xC7BC,
       0x48C4, 0x58E5, 0x6886, 0x78A7, 0x0840, 0x1861, 0x2802, 0x3823,
       0xC9CC, 0xD9ED, 0xE98E, 0xF9AF, 0x8948, 0x9969, 0xA90A, 0xB92B,
       0x5AF5, 0x4AD4, 0x7AB7, 0x6A96, 0x1A71, 0x0A50, 0x3A33, 0x2A12,
       0xDBFD, 0xCBDC, 0xFBBF, 0xEB9E, 0x9B79, 0x8B58, 0xBB3B, 0xAB1A,
       0x6CA6, 0x7C87, 0x4CE4, 0x5CC5, 0x2C22, 0x3C03, 0x0C60, 0x1C41,
       0xEDAE, 0xFD8F, 0xCDEC, 0xDDCD, 0xAD2A, 0xBD0B, 0x8D68, 0x9D49,
       0x7E97, 0x6EB6, 0x5ED5, 0x4EF4, 0x3E13, 0x2E32, 0x1E51, 0x0E70,
       0xFF9F, 0xEFBE, 0xDFDD, 0xCFFC, 0xBF1B, 0xAF3A, 0x9F59, 0x8F78,
       0x9188, 0x81A9, 0xB1CA, 0xA1EB, 0xD10C, 0xC12D, 0xF14E, 0xE16F,
       0x1080, 0x00A1, 0x30C2, 0x20E3, 0x5004, 0x4025, 0x7046, 0x6067,
       0x83B9, 0x9398, 0xA3FB, 0xB3DA, 0xC33D, 0xD31C, 0xE37F, 0xF35E,
       0x02B1, 0x1290, 0x22F3, 0x32D2, 0x4235, 0x5214, 0x6277, 0x7256,
       0xB5EA, 0xA5CB, 0x95A8, 0x8589, 0xF56E, 0xE54F, 0xD52C, 0xC50D,
       0x34E2, 0x24C3, 0x14A0, 0x0481, 0x7466, 0x6447, 0x5424, 0x4405,
       0xA7DB, 0xB7FA, 0x8799, 0x97B8, 0xE75F, 0xF77E, 0xC71D, 0xD73C,
       0x26D3, 0x36F2, 0x0691, 0x16B0, 0x6657, 0x7676, 0x4615, 0x5634,
       0xD94C, 0xC96D, 0xF90E, 0xE92F, 0x99C8, 0x89E9, 0xB98A, 0xA9AB,
       0x5844, 0x4865, 0x7806, 0x6827, 0x18C0, 0x08E1, 0x3882, 0x28A3,
       0xCB7D, 0xDB5C, 0xEB3F, 0xFB1E, 0x8BF9, 0x9BD8, 0xABBB, 0xBB9A,
       0x4A75, 0x5A54, 0x6A37, 0x7A16, 0x0AF1, 0x1AD0, 0x2AB3, 0x3A92,
       0xFD2E, 0xED0F, 0xDD6C, 0xCD4D, 0xBDAA, 0xAD8B, 0x9DE8, 0x8DC9,
       0x7C26, 0x6C07, 0x5C64, 0x4C45, 0x3CA2, 0x2C83, 0x1CE0, 0x0CC1,
       0xEF1F, 0xFF3E, 0xCF5D, 0xDF7C, 0xAF9B, 0xBFBA, 0x8FD9, 0x9FF8,
       0x6E17, 0x7E36, 0x4E55, 0x5E74, 0x2E93, 0x3EB2, 0x0ED1, 0x1EF0
};

unsigned short
udf_cksum(unsigned char *s, int n)
{
       unsigned short crc=0;

       while (n-- > 0)
               crc = crc_table[(crc>>8 ^ *s++) & 0xff] ^ (crc<<8);
       return crc;
}

/* UNICODE Checksum */
unsigned short
udf_unicode_cksum(unsigned short *s, int n)
{
       unsigned short crc=0;

       while (n-- > 0) {
               /* Take high order byte first--corresponds to a big endian
                * byte stream.
                */
               crc = crc_table[(crc>>8 ^ (*s>>8)) & 0xff] ^ (crc<<8);
               crc = crc_table[(crc>>8 ^ (*s++ & 0xff)) & 0xff] ^ (crc<<8);
       }
       return crc;
}


/*
 * Calculates a 16-bit checksum of the Implementation Use
 * Extended Attribute header or Application Use Extended Attribute
 * header. The fields AttributeType through ImplementationIdentifier
 * (or ApplicationIdentifier) inclusively represent the
 * data covered by the checksum (48 bytes).
 *
 */
uint16_t udf_ea_cksum(uint8_t *data) {
       uint16_t checksum = 0;
       int      count;

       for (count = 0; count < 48; count++) {
              checksum += *data++;
       }

       return checksum;
}


#ifdef MAIN
unsigned char bytes[] = { 0x70, 0x6A, 0x77 };

main(void)
{
       unsigned short x;
       x = cksum(bytes, sizeof bytes);
       printf("checksum: calculated=%4.4x, correct=%4.4x\en", x, 0x3299);
       exit(0);
}
#endif

/*****************************************************************************/
/* #ifdef NEEDS_ISPRINT */
/***********************************************************************
* OSTA UDF compliant file name translation routine for OS/2,
* Windows 95, Windows NT, Macintosh and UNIX.
* Copyright 1995 Micro Design International, Inc.
* Written by Jason M. Rinn.
* Micro Design International gives permission for the free use of the
* following source code.
*/

/***********************************************************************
* To use these routines with different operating systems.
*
* OS/2
* Define OS2
* Define MAXLEN = 254
*
* Windows 95
* Define WIN_95
* Define MAXLEN = 255
*
* Windows NT
* Define WIN_NT
* Define MAXLEN = 255
*
* Macintosh:
* Define MAC.
* Define MAXLEN = 31.
*
* UNIX
* Define UNIX.
* Define MAXLEN as specified by unix version.
*/

#define ILLEGAL_CHAR_MARK       0x005F
#define CRC_MARK        0x0023
#define EXT_SIZE        5
#define PERIOD  0x002E
#define SPACE   0x0020

/*** PROTOTYPES ***/
int IsIllegal(unicode_t ch);

/* Define a function or macro which determines if a Unicode character is
* printable under your implementation.
*/


/* #include <stdio.h> */
static int UnicodeIsPrint(unicode_t ch) {
       return (ch >=' ') && (ch != 127);
}


int UnicodeLength(unicode_t *string) {
       int length;
       length = 0;
       while (*string++) length++;

       return length;
}


#ifdef _KERNEL
static int isprint(int c) {
       return (c >= ' ') && (c != 127);
}
#endif


/***********************************************************************
* Translates a long file name to one using a MAXLEN and an illegal
* char set in accord with the OSTA requirements. Assumes the name has
* already been translated to Unicode.
*
* RETURN VALUE
*
* Number of unicode characters in translated name.
*/
int UDFTransName(
       unicode_t *newName,     /* (Output)Translated name. Must be of length
                                * MAXLEN */
       unicode_t *udfName,     /* (Input) Name from UDF volume.*/
       int udfLen)             /* (Input) Length of UDF Name. */
{
       int Index, newIndex = 0, needsCRC = false;      /* index is shadowed */
       int extIndex = 0, newExtIndex = 0, hasExt = false;
#if defined OS2 || defined WIN_95 || defined WIN_NT
       int trailIndex = 0;
#endif
       unsigned short valueCRC;
       unicode_t current;
       const char hexChar[] = "0123456789ABCDEF";

       for (Index = 0; Index < udfLen; Index++) {
               current = udfName[Index];

               if (IsIllegal(current) || !UnicodeIsPrint(current)) {
                       needsCRC = true;
                       /* Replace Illegal and non-displayable chars with
                        * underscore.
                        */
                       current = ILLEGAL_CHAR_MARK;
                       /* Skip any other illegal or non-displayable
                        * characters.
                        */
                       while(Index+1 < udfLen && (IsIllegal(udfName[Index+1])
                           || !UnicodeIsPrint(udfName[Index+1]))) {
                               Index++;
                       }
               }

               /* Record position of extension, if one is found. */
               if (current == PERIOD && (udfLen - Index -1) <= EXT_SIZE) {
                       if (udfLen == Index + 1) {
                               /* A trailing period is NOT an extension. */
                               hasExt = false;
                       } else {
                               hasExt = true;
                               extIndex = Index;
                               newExtIndex = newIndex;
                       }
               }

#if defined OS2 || defined WIN_95 || defined WIN_NT
               /* Record position of last char which is NOT period or space. */
               else if (current != PERIOD && current != SPACE) {
                       trailIndex = newIndex;
               }
#endif

               if (newIndex < MAXLEN) {
                       newName[newIndex++] = current;
               } else {
                       needsCRC = true;
               }
       }

#if defined OS2 || defined WIN_95 || defined WIN_NT
       /* For OS2, 95 & NT, truncate any trailing periods and\or spaces. */
       if (trailIndex != newIndex - 1) {
               newIndex = trailIndex + 1;
               needsCRC = true;
               hasExt = false; /* Trailing period does not make an
                                * extension. */
       }
#endif

       if (needsCRC) {
               unicode_t ext[EXT_SIZE];
               int localExtIndex = 0;
               if (hasExt) {
                       int maxFilenameLen;
                       /* Translate extension, and store it in ext. */
                       for(Index = 0; Index<EXT_SIZE &&
                           extIndex + Index +1 < udfLen; Index++ ) {
                               current = udfName[extIndex + Index + 1];
                               if (IsIllegal(current) ||
                                   !UnicodeIsPrint(current)) {
                                       needsCRC = 1;
                                       /* Replace Illegal and non-displayable
                                        * chars with underscore.
                                        */
                                       current = ILLEGAL_CHAR_MARK;
                                       /* Skip any other illegal or
                                        * non-displayable characters.
                                        */
                                       while(Index + 1 < EXT_SIZE
                                           && (IsIllegal(udfName[extIndex +
                                           Index + 2]) ||
                                           !isprint(udfName[extIndex +
                                           Index + 2]))) {
                                               Index++;
                                       }
                               }
                               ext[localExtIndex++] = current;
                       }

                       /* Truncate filename to leave room for extension and
                        * CRC.
                        */
                       maxFilenameLen = ((MAXLEN - 5) - localExtIndex - 1);
                       if (newIndex > maxFilenameLen) {
                               newIndex = maxFilenameLen;
                       } else {
                               newIndex = newExtIndex;
                       }
               } else if (newIndex > MAXLEN - 5) {
                       /*If no extension, make sure to leave room for CRC. */
                       newIndex = MAXLEN - 5;
               }
               newName[newIndex++] = CRC_MARK; /* Add mark for CRC. */

               /*Calculate CRC from original filename from FileIdentifier. */
               valueCRC = udf_unicode_cksum(udfName, udfLen);
               /* Convert 16-bits of CRC to hex characters. */
               newName[newIndex++] = hexChar[(valueCRC & 0xf000) >> 12];
               newName[newIndex++] = hexChar[(valueCRC & 0x0f00) >> 8];
               newName[newIndex++] = hexChar[(valueCRC & 0x00f0) >> 4];
               newName[newIndex++] = hexChar[(valueCRC & 0x000f)];

               /* Place a translated extension at end, if found. */
               if (hasExt) {
                       newName[newIndex++] = PERIOD;
                       for (Index = 0;Index < localExtIndex ;Index++ ) {
                               newName[newIndex++] = ext[Index];
                       }
               }
       }
       return(newIndex);
}

#if defined OS2 || defined WIN_95 || defined WIN_NT
/***********************************************************************
* Decides if a Unicode character matches one of a list
* of ASCII characters.
* Used by OS2 version of IsIllegal for readability, since all of the
* illegal characters above 0x0020 are in the ASCII subset of Unicode.
* Works very similarly to the standard C function strchr().
*
* RETURN VALUE
*
* Non-zero if the Unicode character is in the given ASCII string.
*/
int UnicodeInString(
       unsigned char *string,  /* (Input) String to search through. */
       unicode_t ch)           /* (Input) Unicode char to search for. */
{
       int found = false;
       while (*string != '\0' && found == false) {
               /* These types should compare, since both are unsigned
                * numbers. */
               if (*string == ch) {
                       found = true;
               }
               string++;
       }
       return(found);
}
#endif /* OS2 */

/***********************************************************************
* Decides whether the given character is illegal for a given OS.
*
* RETURN VALUE
*
* Non-zero if char is illegal.
*/
int IsIllegal(unicode_t ch)
{
#ifdef MAC
       /* Only illegal character on the MAC is the colon. */
       if (ch == 0x003A) {
               return(1);
       } else {
               return(0);
       }

#elif defined UNIX
       /* Illegal UNIX characters are NULL and slash. */
       if (ch == 0x0000 || ch == 0x002F) {
               return(1);
       } else {
               return(0);
       }

#elif defined OS2 || defined WIN_95 || defined WIN_NT
       /* Illegal char's for OS/2 according to WARP toolkit. */
       if (ch < 0x0020 || UnicodeInString("\\/:*?\"<>|", ch)) {
               return(1);
       } else {
               return(0);
       }
#endif
}
/* #endif*/     /* NEEDS_ISPRINT */