Subversion Repositories filter_foundry

/*

    This file is part of "Filter Foundry", a filter plugin for Adobe Photoshop

    Copyright (C) 2003-2009 Toby Thain, toby@telegraphics.com.au

    Copyright (C) 2018-2021 Daniel Marschall, ViaThinkSoft

    This program is free software; you can redistribute it and/or modify

    it under the terms of the GNU General Public License as published by

    the Free Software Foundation; either version 2 of the License, or

    (at your option) any later version.

    This program is distributed in the hope that it will be useful,

    but WITHOUT ANY WARRANTY; without even the implied warranty of

    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License

    along with this program; if not, write to the Free Software

    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

*/

#include <stddef.h>

#include <stdint.h>

#include <assert.h>

#include <time.h>

#include "ff.h"

#include "symtab.h"

#include "scripting.h"

#include "PIActions.h"

#include "PITerminology.h"

#include "compat_string.h"

//#define PRINTABLE_HASH_FF16

#define ENABLE_APPLESCRIPT

/*

Find a printable 4-character key, remembering (see Photoshop API guide):

- All IDs starting with an uppercase letter are reserved by Adobe.

- All IDs that are all uppercase are reserved by Apple.

- All IDs that are all lowercase are reserved by Apple.

- This leaves all IDs that begin with a lowercase letter and have at least

  one uppercase letter for you and other plug-in developers.

Note: It is questionable if "a!!!" or "A!!!" are also reseved by Apple. We don't risk it.

*/

unsigned long printablehash(unsigned long hash) {

#ifdef PRINTABLE_HASH_FF16

        // FilterFoundry version 1.6 hashing by Toby Thain

        // Only accepts upper case at the last character

        // 0       = 'a  A'

        // 6100899 = 'z~~Z'

        unsigned long key = 'a' + (hash % 26);  hash /= 26; // first lower case

        key = (key << 8) | (' ' + (hash % 95)); hash /= 95; // any printable

        key = (key << 8) | (' ' + (hash % 95)); hash /= 95; // any printable

        return  (key << 8) | ('A' + (hash % 26));           // last upper case

#else

        // FilterFoundry version 1.7 hashing by Daniel Marschall

        // Accepts upper case at character 2, 3 or 4

        // Spaces are only set the right as padding to make a code shorter

        // 0        = 'aA  '

        // 13530139 = 'zZZZ'

        // The key-space is ~2.22 times larger

        long lowlim;

        long uplim;

        int upperCaseInfo;

        int length;

        int lastThreeCharInfo;

        int firstChar;

        unsigned long key;

        int found;

        int i,j,k;

        uplim = 0;

        for (k=1; k<4; k++) {

                int mask;

                if (k == 1) mask = 1;//0b1;

                if (k == 2) mask = 2;//0b11;

                if (k == 3) mask = 4;//0b111;

                for (i=1; i<=mask; i++) {

                        // 'k' characters

                        long test = 1;

                        for (j=0; j<k; ++j) {

                                test *= ((i&(1<<j)) != 0) ? 26 : 94-26;

                        }

                        uplim += test;

                }

        }

        lastThreeCharInfo = hash % uplim; hash /= uplim;

        firstChar = hash%26; hash /= 26;

        lowlim = 0;

        uplim = 0;

        found = 0;

        length = -1; // avoid compiler warning

        upperCaseInfo = -1; // avoid compiler warning

        for (k=1; k<4; k++) {

                int mask;

                if (k == 1) mask = 1;//0b1;

                if (k == 2) mask = 2;//0b11;

                if (k == 3) mask = 4;//0b111;

                if (!found) for (i=1; i<=mask; i++) {

                        // 'k' characters

                        long test = 1;

                        for (j=0; j<k; ++j) {

                                test *= ((i&(1<<j)) != 0) ? 26 : 94-26;

                        }

                        uplim += test;

                        if ((lastThreeCharInfo >= lowlim) && (lastThreeCharInfo < uplim)) {

                                lastThreeCharInfo -= lowlim;

                                found = 1;

                                length = k;

                                upperCaseInfo = i;

                                break;

                        }

                        lowlim = uplim;

                }

        }

        key = ('a' + firstChar) << 24; // first char is lower-case

        for (i=0; i<length; ++i) {

                char res;

                if ((upperCaseInfo&(1<<i)) == 0) {

                        res = '!' + (lastThreeCharInfo % (94-26));

                        if (res >= 'A') res += 26;

                        lastThreeCharInfo /= (94-26);

                } else {

                        res = 'A' + (lastThreeCharInfo % 26);

                        lastThreeCharInfo /= 26;

                }

                key |= res << (8*(i+(4-length-1))); // 2nd, 3rd, 4th char are either upper-case or any printable char

        }

        if (length == 1) {

                key &= 0xFFFF0000;

                key |= ' ' << 8;

                key |= ' ';

        }

        if (length == 2) {

                key &= 0xFFFFFF00;

                key |= ' ';

        }

        return key;

#endif

}

size_t roundToNext4(size_t x) {

        int pad = 4 - (x % 4);

        if (pad == 0) pad = 4;

        return x + pad;

}

size_t fixpipl(PIPropertyList *pipl, size_t origsize, StringPtr title, long *event_id) {

        PIProperty *prop;

        char *p;

        struct hstm_data {

                /* this structure must be 14+1 bytes long, to match PiPL structure */

                long version; /* = 0 */

                long class_id;

                long event_id;

                short aete_resid;

                char scope[1];

        };

        struct hstm_data *hstm;

        int scopelen;

        unsigned long hash;

        size_t realLength;

        size_t roundedLength;

        char* scope;

        pipl->count += 3; // 3 more keys in PiPL: name, catg, hstm

        p = (char*)pipl + origsize;

        prop = (PIProperty*)p;

        /* add Title/Name property key */

        prop->vendorID = kPhotoshopSignature;

        prop->propertyKey = PINameProperty;

        prop->propertyID = 0;

        prop->propertyLength = (SPInt32)roundToNext4((size_t)title[0] + 1);

        memset(prop->propertyData, 0x00, prop->propertyLength); // fill padding with 00h bytes (cosmetics)

        PLstrcpy((StringPtr)prop->propertyData, title);

        // skip past new property record, and any padding

        p += offsetof(PIProperty, propertyData) + prop->propertyLength;

        prop = (PIProperty*)p;

        /* add Category property key */

        prop->vendorID = kPhotoshopSignature;

        prop->propertyKey = PICategoryProperty;

        prop->propertyID = 0;

        prop->propertyLength = (SPInt32)roundToNext4((size_t)gdata->parm.category[0] + 1);

        memset(prop->propertyData, 0x00, prop->propertyLength); // fill padding with 00h bytes (cosmetics)

        PLstrcpy((StringPtr)prop->propertyData, gdata->parm.category);

        p += offsetof(PIProperty, propertyData) + prop->propertyLength;

        prop = (PIProperty*)p;

        /* add HasTerminology property key */

        hstm = (struct hstm_data*)prop->propertyData;

        scope = (char*)malloc(0x300);

        if (!scope) return -1;

        sprintf(scope, "%s %s",

            INPLACEP2CSTR(gdata->parm.category),

            INPLACEP2CSTR(title));

        #ifdef ENABLE_APPLESCRIPT

        // If the uniqueString/scope is set, the plugin will only communicate with Photoshop.

        // Otherwise it can be accessed with AppleScript, but the AETE keys need to be unique then.

        // This is achieved with getAeteKey().

        scopelen = 0;

        hstm->scope[0] = '\0';

        #else

        // Construct scope string by concatenating Category and Title - hopefully unique!

        // Note: In doresources() we malloc'ed 300h additional bytes,

        // and in the build dialog, title and category are size-limited,

        // so we can write here without malloc.

        scopelen = strlen(scope);

        memcpy(&hstm->scope, scope, scopelen);

        #endif

        /* make up a new event ID for this aete, based on printable base-95 hash of scope */

        hash = djb2(scope);

        *event_id = printablehash(hash); /* this is used by aete_generate() later... */

        prop->vendorID = kPhotoshopSignature;

        prop->propertyKey = PIHasTerminologyProperty;

        prop->propertyID = 0;

        realLength = offsetof(struct hstm_data, scope) + scopelen + 1/*null-term*/;

        roundedLength = roundToNext4(realLength);

        prop->propertyLength = (SPInt32)roundedLength;

        memset(prop->propertyData + realLength, 0x00, roundedLength - realLength); // fill padding with 00h bytes (cosmetics)

        hstm->version = 0;

        hstm->class_id = plugInClassID;

        hstm->event_id = *event_id;

        hstm->aete_resid = AETE_ID;

        p += offsetof(PIProperty, propertyData) + prop->propertyLength;

        return p - (char*)pipl;  // figure how many bytes were added

}

void _aete_write_byte(void** aeteptr, uint8_t val) {

        uint8_t* tmp = *((uint8_t**)aeteptr);

        *tmp = val;

        *aeteptr = (void*)((unsigned char*)tmp + 1);

}

#define AETE_WRITE_BYTE(i) _aete_write_byte(&aeteptr, (i));

void _aete_write_word(void** aeteptr, uint16_t val) {

        uint16_t* tmp = *((uint16_t**)aeteptr);

        *tmp = val;

        *aeteptr = (void*)((unsigned char*)tmp + 2);

}

#define AETE_WRITE_WORD(i) _aete_write_word(&aeteptr, (i));

void _aete_write_dword(void** aeteptr, uint32_t val) {

        uint32_t* tmp = *((uint32_t**)aeteptr);

        *tmp = val;

        *aeteptr = (void*)((unsigned char*)tmp + 4);

}

#define AETE_WRITE_DWORD(i) _aete_write_dword(&aeteptr, (i));

void _aete_write_c2pstr(void** aeteptr, char* str) {

        char* tmp;

        assert(strlen(str) <= 255);

        _aete_write_byte(aeteptr, (uint8_t)strlen(str));

        tmp = *((char**)aeteptr);

        strcpy(tmp, str);

        *aeteptr = (void*)((unsigned char*)tmp + strlen(str));

}

#define AETE_WRITE_C2PSTR(s) _aete_write_c2pstr(&aeteptr, (s));

void _aete_write_p2pstr(void** aeteptr, char* str) {

    char* tmp;

    if (strlen(str) == 0) {

        _aete_write_byte(aeteptr, 0);

    } else {

        tmp = *((char**)aeteptr);

        strcpy(tmp, str);

        *aeteptr = (void*)((unsigned char*)tmp + strlen(str));

    }

}

#define AETE_WRITE_P2PSTR(s) _aete_write_p2pstr(&aeteptr, (s));

void _aete_align_word(void** aeteptr) {

        #ifdef MAC_ENV

        unsigned char* tmp = *((unsigned char**)aeteptr);

        tmp += (intptr_t)tmp & 1;

        *aeteptr = (void*)tmp;

        #endif

}

#define AETE_ALIGN_WORD() _aete_align_word(&aeteptr);

void* _aete_property(void* aeteptr, PARM_T *pparm, int ctlidx, int mapidx, OSType key) {

        char tmp[256];

        if (pparm->ctl_used[ctlidx] || pparm->map_used[mapidx]) {

                if (pparm->map_used[mapidx]) {

                        if (ctlidx & 1) {

                                sprintf(tmp, "... %s", (char*)pparm->map[mapidx]);

                        } else {

                                sprintf(tmp, "%s ...", (char*)pparm->map[mapidx]);

                        }

                        AETE_WRITE_C2PSTR(tmp);

                } else {

                        AETE_WRITE_P2PSTR((char*)pparm->ctl[ctlidx]);

                }

                AETE_ALIGN_WORD();

                AETE_WRITE_DWORD(key);

                AETE_WRITE_DWORD(typeSInt32);

                AETE_WRITE_C2PSTR(_strdup(""));

                AETE_ALIGN_WORD();

                AETE_WRITE_WORD(0x8000); /* FLAGS_1_OPT_PARAM / flagsOptionalSingleParameter */

        }

        return aeteptr;

}

size_t aete_generate(void* aeteptr, PARM_T *pparm, long event_id) {

        // TODO: Can't we use the PI SDK to create an AETE?

        int numprops;

        void *beginptr = aeteptr;

        // Attention!

        // - On some systems (e.g. ARM based CPUs) this will cause an unaligned memory access exception.

        //   For X86, memory access just becomes slower.

        // - If you change something here, please also change it in Scripting.rc (Windows) and scripting.r (Mac OS)

        // Note:

        // - The 'aete' resource for Mac OS has word alignments after strings (but not if the next element is also a string)

        //   see https://developer.apple.com/library/archive/documentation/mac/pdf/Interapplication_Communication/AE_Term_Resources.pdf page 8-9

#ifdef WIN_ENV

        AETE_WRITE_WORD(0x0001); /* Reserved (for Photoshop) */

#endif

        AETE_WRITE_BYTE(0x01); /* aete version */

        AETE_WRITE_BYTE(0x00); /* aete version */

        AETE_WRITE_WORD(english); /* language specifiers */

        AETE_WRITE_WORD(roman);

        AETE_WRITE_WORD(1); /* 1 suite */

        {

                AETE_WRITE_P2PSTR((char*)pparm->author); /* vendor suite name */

                AETE_WRITE_C2PSTR(_strdup("")); /* optional description */

                AETE_ALIGN_WORD();

                AETE_WRITE_DWORD(plugInSuiteID); /* suite ID */

                AETE_WRITE_WORD(1); /* suite code, must be 1. Attention: Filters like 'Pointillize' have set this to 0! */

                AETE_WRITE_WORD(1); /* suite level, must be 1. Attention: Filters like 'Pointillize' have set this to 0! */

                AETE_WRITE_WORD(1); /* 1 event (structure for filters) */

                {

                        AETE_WRITE_P2PSTR((char*)pparm->title); /* event name */

                        AETE_WRITE_C2PSTR(_strdup("")); /* event description */

                        AETE_ALIGN_WORD();

                        AETE_WRITE_DWORD(plugInClassID); /* event class */

                        AETE_WRITE_DWORD(/*plugInEventID*/event_id); /* event ID */

                        /* NO_REPLY: */

                        AETE_WRITE_DWORD(noReply); /* noReply='null' */

                        AETE_WRITE_C2PSTR(_strdup("")); /* reply description */

                        AETE_ALIGN_WORD();

                        AETE_WRITE_WORD(0);

                        /* IMAGE_DIRECT_PARAM: */

                        AETE_WRITE_DWORD(typeImageReference); /* typeImageReference='#ImR' */

                        AETE_WRITE_C2PSTR(_strdup("")); /* direct parm description */

                        AETE_ALIGN_WORD();

                        AETE_WRITE_WORD(0xB000);

                        numprops = 0;

                        if (pparm->ctl_used[0] || pparm->map_used[0]) numprops++;

                        if (pparm->ctl_used[1] || pparm->map_used[0]) numprops++;

                        if (pparm->ctl_used[2] || pparm->map_used[1]) numprops++;

                        if (pparm->ctl_used[3] || pparm->map_used[1]) numprops++;

                        if (pparm->ctl_used[4] || pparm->map_used[2]) numprops++;

                        if (pparm->ctl_used[5] || pparm->map_used[2]) numprops++;

                        if (pparm->ctl_used[6] || pparm->map_used[3]) numprops++;

                        if (pparm->ctl_used[7] || pparm->map_used[3]) numprops++;

                        AETE_WRITE_WORD(numprops);

                        {

                                // Standalone filters don't need RGBA expressions

                                /*

                                AETE_WRITE_C2PSTR(_strdup("R"));

                                AETE_ALIGN_WORD();

                                AETE_WRITE_DWORD(PARAM_R_KEY);

                                AETE_WRITE_DWORD(typeText);

                                AETE_WRITE_C2PSTR(_strdup("R channel expression"));

                                AETE_ALIGN_WORD();

                                AETE_WRITE_WORD(0x8000);

                                AETE_WRITE_C2PSTR(_strdup("G"));

                                AETE_ALIGN_WORD();

                                AETE_WRITE_DWORD(PARAM_G_KEY);

                                AETE_WRITE_DWORD(typeText);

                                AETE_WRITE_C2PSTR(_strdup("G channel expression"));

                                AETE_ALIGN_WORD();

                                AETE_WRITE_WORD(0x8000);

                                AETE_WRITE_C2PSTR(_strdup("B"));

                                AETE_ALIGN_WORD();

                                AETE_WRITE_DWORD(PARAM_B_KEY);

                                AETE_WRITE_DWORD(typeText);

                                AETE_WRITE_C2PSTR(_strdup("B channel expression"));

                                AETE_ALIGN_WORD();

                                AETE_WRITE_WORD(0x8000);

                                AETE_WRITE_C2PSTR(_strdup("A"));

                                AETE_ALIGN_WORD();

                                AETE_WRITE_DWORD(PARAM_A_KEY);

                                AETE_WRITE_DWORD(typeText);

                                AETE_WRITE_C2PSTR(_strdup("A channel expression"));

                                AETE_ALIGN_WORD();

                                AETE_WRITE_WORD(0x8000);

                                */

                                aeteptr = _aete_property(aeteptr, pparm, 0, 0, getAeteKey('0', pparm));

                                aeteptr = _aete_property(aeteptr, pparm, 1, 0, getAeteKey('1', pparm));

                                aeteptr = _aete_property(aeteptr, pparm, 2, 1, getAeteKey('2', pparm));

                                aeteptr = _aete_property(aeteptr, pparm, 3, 1, getAeteKey('3', pparm));

                                aeteptr = _aete_property(aeteptr, pparm, 4, 2, getAeteKey('4', pparm));

                                aeteptr = _aete_property(aeteptr, pparm, 5, 2, getAeteKey('5', pparm));

                                aeteptr = _aete_property(aeteptr, pparm, 6, 3, getAeteKey('6', pparm));

                                aeteptr = _aete_property(aeteptr, pparm, 7, 3, getAeteKey('7', pparm));

                        }

                }

                /* non-filter plug-in class here */

                AETE_WRITE_WORD(0); /* 0 classes */

                {}

                AETE_WRITE_WORD(0); /* 0 comparison ops (not supported) */

                {}

                AETE_WRITE_WORD(0); /* 0 enumerations */

                {}

        }

        AETE_WRITE_DWORD(0); /* padding (FIXME: do we need that? Adobe's Windows filters don't) */

        return (unsigned char*)aeteptr - (unsigned char*)beginptr; // length of stuff written

}

void _xorshift(unsigned char** p, uint32_t* x32, size_t num) {

        size_t i;

        unsigned char* x = *p;

        for (i = 0; i < num; i++) {

                // https://de.wikipedia.org/wiki/Xorshift

                *x32 ^= *x32 << 13;

                *x32 ^= *x32 >> 17;

                *x32 ^= *x32 << 5;

                *x++ ^= *x32;

        }

        *p = x;

}

void obfusc(unsigned char* pparm, size_t size, size_t seed_position) {

        unsigned char* p;

        uint32_t x32;

        unsigned int seed;

        seed = (unsigned int)time(0);

        p = pparm;

        x32 = (uint32_t)seed;

        _xorshift(&p, &x32, seed_position);

        *((unsigned int*)p) = seed; // seed is placed at this position. data will lost! (in deobfusc, it will be set to 0x00000000)

        p += 4;

        _xorshift(&p, &x32, size - seed_position - 4);

}

void deobfusc(unsigned char* pparm, size_t size, size_t seed_position) {

        unsigned char* p;

        uint32_t x32;

        unsigned int seed;

        seed = *((unsigned int*)(pparm + seed_position));

        p = pparm;

        x32 = (uint32_t)seed;

        _xorshift(&p, &x32, seed_position);

        *((unsigned int*)p) = 0; // here was the seed. Fill it with 0x00000000

        p += 4;

        _xorshift(&p, &x32, size - seed_position - 4);

}