scalos/main/Patches-aos4.c

   1 // Patches-aos4.c
   2 // $Date$
   3 // $Revision$
   4
   5
   6 #include <exec/types.h>
   7 #include <graphics/rastport.h>
   8 #include <utility/hooks.h>
   9 #include <intuition/gadgetclass.h>
  10 #include <workbench/workbench.h>
  11 #include <workbench/startup.h>
  12 #include <workbench/icon.h>
  13 #include <libraries/gadtools.h>
  14 #include <libraries/iffparse.h>
  15 #include <datatypes/pictureclass.h>
  16 #include <dos/dostags.h>
  17 #include <guigfx/guigfx.h>
  18
  19 #define __USE_SYSBASE
  20
  21 #include <proto/dos.h>
  22 #include <proto/exec.h>
  23 #include <proto/layers.h>
  24 #include <proto/intuition.h>
  25 #include <proto/graphics.h>
  26 #include <proto/utility.h>
  27 #include <proto/icon.h>
  28 #include <proto/iconobject.h>
  29 #include <proto/guigfx.h>
  30 #include <proto/datatypes.h>
  31 #include <proto/gadtools.h>
  32 #include <proto/wb.h>
  33 #include "debug.h"
  34 #include <proto/scalos.h>
  35
  36 #include <clib/alib_protos.h>
  37
  38 #include <defs.h>
  39 #include <datatypes/iconobject.h>
  40 #include <scalos/palette.h>
  41 #include <scalos/pattern.h>
  42 #include <scalos/scalos.h>
  43
  44 #include <stdlib.h>
  45 #include <stdio.h>
  46 #include <string.h>
  47
  48 #include "scalos_structures.h"
  49 #include "functions.h"
  50 #include "Variables.h"
  51 #include "Patches.h"
  52
  53 //----------------------------------------------------------------------------
  54 // Revision history :
  55 //
  56 // 20011203     jl      initial history
  57 //----------------------------------------------------------------------------
  58
  59 // local data structures
  60
  61 struct PatchEntry
  62         {
  63         struct Interface **pe_Interface;
  64         LONG pe_IFaceOffset;
  65         APTR pe_NewFunction;
  66         APTR *pe_OldFunction;
  67         };
  68
  69 //----------------------------------------------------------------------------
  70
  71 // local functions
  72
  73 static void SetPatches(const struct PatchEntry *PatchList);
  74 static BOOL RemovePatches(const struct PatchEntry *PatchList);
  75
  76 //----------------------------------------------------------------------------
  77
  78 // public data items
  79
  80 //----------------------------------------------------------------------------
  81
  82 // local data items
  83 static LIBFUNC_P7VA_PROTO(struct AppIcon *, sca_AddAppIcon,
  84         D0, ULONG, Id,
  85         D1, ULONG, UserData,
  86         A0, CONST_STRPTR, IconText,
  87         A1, struct MsgPort *, iconPort,
  88         A2, BPTR, lock,
  89         A3, struct DiskObject *, DiskObj,
  90         A6, struct Library *, wbBase);
  91 static LIBFUNC_P5VA_PROTO(struct AppWindow *, sca_AddAppWindow,
  92         D0, ULONG, Id,
  93         D1, ULONG, UserData,
  94         A0, struct Window *, window,
  95         A1, struct MsgPort *, iconPort,
  96         A6, struct Library *, wbBase);
  97 static LIBFUNC_P5VA_PROTO(struct AppMenuItem *, sca_AddAppMenuItem,
  98         D0, ULONG, Id,
  99         D1, ULONG, UserData,
 100         A0, CONST_STRPTR, text,
 101         A1, struct MsgPort *, iconPort,
 102         A6, struct Library *, wbBase);
 103 static LIBFUNC_P2VA_PROTO(struct Screen *, sca_OpenScreen,
 104         A0, struct NewScreen *, newScr,
 105         A6, struct IntuitionBase *, iBase);
 106
 107 static APTR OldAddAppIcon;
 108 static APTR OldAddAppWindow;
 109 static APTR OldAppAppMenuItem;
 110 static APTR OldOpenScreen;
 111
 112 //----------------------------------------------------------------------------
 113
 114 #define OffsetCloseWorkBench            offsetof(struct IntuitionIFace, CloseWorkBench)
 115 #define OffsetOpenWorkBench             offsetof(struct IntuitionIFace, OpenWorkBench)
 116 #define OffsetOpenScreenTagList         offsetof(struct IntuitionIFace, OpenScreenTagList)
 117 #define OffsetOpenScreen                offsetof(struct IntuitionIFace, OpenScreen)
 118
 119 #define OffsetUpdateWorkbench           offsetof(struct WorkbenchIFace, UpdateWorkbench)
 120 //#define OffsetwbPrivate2              offsetof(struct WorkbenchIFace, Private1)
 121 #define OffsetAddAppWindowA             offsetof(struct WorkbenchIFace, AddAppWindowA)
 122 #define OffsetAddAppWindow              offsetof(struct WorkbenchIFace, AddAppWindow)
 123 #define OffsetRemoveAppWindow           offsetof(struct WorkbenchIFace, RemoveAppWindow)
 124 #define OffsetAddAppIconA               offsetof(struct WorkbenchIFace, AddAppIconA)
 125 #define OffsetAddAppIcon                offsetof(struct WorkbenchIFace, AddAppIcon)
 126 #define OffsetRemoveAppIcon             offsetof(struct WorkbenchIFace, RemoveAppIcon)
 127 #define OffsetAddAppMenuItemA           offsetof(struct WorkbenchIFace, AddAppMenuItemA)
 128 #define OffsetAddAppMenuItem            offsetof(struct WorkbenchIFace, AddAppMenuItem)
 129 #define OffsetRemoveAppMenuItem         offsetof(struct WorkbenchIFace, RemoveAppMenuItem)
 130 #define OffsetWBInfo                    offsetof(struct WorkbenchIFace, WBInfo)
 131 //#define OffsetSetBackFill             offsetof(struct WorkbenchIFace, ...)
 132
 133 #define OffsetPutDiskObject             offsetof(struct IconIFace, PutDiskObject)
 134 #define OffsetDeleteDiskObject          offsetof(struct IconIFace, DeleteDiskObject)
 135 #define OffsetPutIconTagList            offsetof(struct IconIFace, PutIconTagList)
 136
 137 #define OffsetRename                    offsetof(struct DOSIFace, Rename)
 138 #define OffsetDeleteFile                offsetof(struct DOSIFace, DeleteFile)
 139 #define OffsetOpen                      offsetof(struct DOSIFace, Open)
 140 #define OffsetClose                     offsetof(struct DOSIFace, Close)
 141 #define OffsetCreateDir                 offsetof(struct DOSIFace, CreateDir)
 142
 143 //----------------------------------------------------------------------------
 144 static const struct PatchEntry EmulationPatches[] =
 145         {
 146         { (struct Interface **)(APTR) &IIntuition,      OffsetCloseWorkBench,           (APTR)sca_CloseWorkBench,         (APTR) &OldCloseWB },
 147         { (struct Interface **)(APTR) &IIntuition,      OffsetOpenWorkBench,            (APTR)sca_OpenWorkBench,          (APTR) &OldOpenWB },
 148         { (struct Interface **)(APTR) &IIntuition,      OffsetOpenScreenTagList,        (APTR)sca_OpenScreenTagList,      (APTR) &OldOpenScreenTagList },
 149         { (struct Interface **)(APTR) &IIntuition,      OffsetOpenScreen,               (APTR)sca_OpenScreen,             (APTR) &OldOpenScreen },
 150         { (struct Interface **)(APTR) &IWorkbench,      OffsetAddAppIconA,              (APTR)sca_AddAppIconA,            (APTR) &OldAddAppIconA },
 151         { (struct Interface **)(APTR) &IWorkbench,      OffsetAddAppIcon,               (APTR)sca_AddAppIcon,             (APTR) &OldAddAppIcon },
 152         { (struct Interface **)(APTR) &IWorkbench,      OffsetRemoveAppIcon,            (APTR)sca_RemoveAppIcon,          (APTR) &OldRemoveAppIcon },
 153         { (struct Interface **)(APTR) &IWorkbench,      OffsetAddAppWindowA,            (APTR)sca_AddAppWindowA,          (APTR) &OldAddAppWindowA },
 154         { (struct Interface **)(APTR) &IWorkbench,      OffsetAddAppWindow,             (APTR)sca_AddAppWindow,           (APTR) &OldAddAppWindow },
 155         { (struct Interface **)(APTR) &IWorkbench,      OffsetRemoveAppWindow,          (APTR)sca_RemoveAppWindow,        (APTR) &OldRemoveAppWindow },
 156         { (struct Interface **)(APTR) &IWorkbench,      OffsetAddAppMenuItemA,          (APTR)sca_AddAppMenuItemA,        (APTR) &OldAppAppMenuItemA },
 157         { (struct Interface **)(APTR) &IWorkbench,      OffsetAddAppMenuItem,           (APTR)sca_AddAppMenuItem,         (APTR) &OldAppAppMenuItem },
 158         { (struct Interface **)(APTR) &IWorkbench,      OffsetRemoveAppMenuItem,        (APTR)sca_RemoveAppMenuItem,      (APTR) &OldRemoveAppMenuItem },
 159         { (struct Interface **)(APTR) &IWorkbench,      OffsetWBInfo,                   (APTR)sca_WBInfo,                 (APTR) &OldWBInfo },
 160
 161 //      { (struct Interface **)(APTR) &IWorkbench,      OffsetwbPrivate2,               (APTR)sca_GetLocString,           (APTR) &OldwbPrivate2 },
 162         { NULL,                                         0,                              (APTR)NULL,                       NULL },
 163         };
 164 static const struct PatchEntry StandardPatches[] =
 165         {
 166         { (struct Interface **)(APTR) &IIcon,           OffsetPutDiskObject,            (APTR)sca_PutDiskObject,        (APTR) &OldPutDiskObject },
 167         { (struct Interface **)(APTR) &IIcon,           OffsetDeleteDiskObject,         (APTR)sca_DeleteDiskObject,     (APTR) &OldDeleteDiskObject },
 168         { (struct Interface **)(APTR) &IIcon,           OffsetPutIconTagList,           (APTR)sca_PutIconTagList,       (APTR) &OldPutIconTagList},
 169
 170         { (struct Interface **)(APTR) &IDOS,            OffsetDeleteFile,               (APTR)sca_DeleteFile,           (APTR) &OldDeleteFile },
 171         { (struct Interface **)(APTR) &IDOS,            OffsetRename,                   (APTR)sca_Rename,               (APTR) &OldRename },
 172         { (struct Interface **)(APTR) &IDOS,            OffsetOpen,                     (APTR)sca_Open,                 (APTR) &OldOpen },
 173         { (struct Interface **)(APTR) &IDOS,            OffsetClose,                    (APTR)sca_Close,                (APTR) &OldClose },
 174         { (struct Interface **)(APTR) &IDOS,            OffsetCreateDir,                (APTR)sca_CreateDir,            (APTR) &OldCreateDir },
 175         { NULL,                                         0,                              (APTR)NULL,                       NULL },
 176         };
 177 static const struct PatchEntry HardEmulationPatches[] =
 178         {
 179         { (struct Interface **)(APTR) &IWorkbench,      OffsetUpdateWorkbench,          (APTR)sca_UpdateWorkbench,        (APTR) &OldUpdateWorkbench },
 180 //      { (struct Interface **)(APTR) &IWorkbench,      OffsetSetBackFill,              (APTR)sca_SetBackFill,            (APTR) &OldSetBackFill },
 181         { NULL,                                         0,                              (APTR)NULL,                       NULL },
 182         };
 183
 184 //----------------------------------------------------------------------------
 185
 186 void SetAllPatches(struct MainTask *mt)
 187 {
 188         PatchInit();
 189
 190         SetPatches(StandardPatches);
 191
 192         if (mt->emulation)
 193                 {
 194                 if (CurrentPrefs.pref_HardEmulation)
 195                         SetPatches(HardEmulationPatches);
 196
 197                 SetPatches(EmulationPatches);
 198                 }
 199 }
 200
 201
 202 ULONG RemoveAllPatches(struct MainTask *mt)
 203 {
 204         if (!RemovePatches(StandardPatches))
 205                 return FALSE;
 206
 207         if (mt->emulation)
 208                 {
 209                 if (!RemovePatches(EmulationPatches))
 210                         {
 211                         SetPatches(StandardPatches);
 212                         return FALSE;
 213                         }
 214
 215                 if (CurrentPrefs.pref_HardEmulation)
 216                         {
 217                         if (!RemovePatches(HardEmulationPatches))
 218                                 {
 219                                 SetPatches(EmulationPatches);
 220                                 SetPatches(StandardPatches);
 221                                 }
 222                         }
 223
 224                 }
 225
 226         PatchCleanup();
 227
 228         return TRUE;
 229 }
 230
 231 static void SetPatches(const struct PatchEntry *PatchList)
 232 {
 233         Forbid();
 234
 235         d1(kprintf("%s/%s/%ld: PatchList=%08lx\n", __FILE__, __FUNC__, __LINE__, PatchList));
 236
 237         while (PatchList->pe_Interface)
 238                 {
 239                 if (NULL == *PatchList->pe_OldFunction)
 240                         {
 241                         *PatchList->pe_OldFunction = SetMethod(*PatchList->pe_Interface,
 242                                 PatchList->pe_IFaceOffset, PatchList->pe_NewFunction);
 243
 244                         d1(KPrintF("%s/%s/%ld: Library=%08lx  Offset=%ld  NewFunc=%08lx  OldFunc=%08lx\n", \
 245                                 __FILE__, __FUNC__, __LINE__, *PatchList->pe_Interface, \
 246                                 PatchList->pe_IFaceOffset, PatchList->pe_NewFunction, \
 247                                 *PatchList->pe_OldFunction));
 248                         }
 249
 250                 PatchList++;
 251                 }
 252
 253 //      CacheClearU();
 254         Permit();
 255 }
 256
 257
 258 static BOOL RemovePatches(const struct PatchEntry *PatchList)
 259 {
 260         const struct PatchEntry *pl = PatchList;
 261         BOOL Success = TRUE;
 262
 263         d1(KPrintF("%s/%s/%ld: PatchList=%08lx\n", __FILE__, __FUNC__, __LINE__, PatchList));
 264
 265         Forbid();
 266
 267         while (pl->pe_Interface)
 268                 {
 269                 if (pl->pe_OldFunction)
 270                         {
 271                         APTR patchedFunction;
 272
 273                         patchedFunction = SetMethod(*pl->pe_Interface,
 274                                 pl->pe_IFaceOffset, *pl->pe_OldFunction);
 275                         *pl->pe_OldFunction = NULL;
 276
 277                         d1(KPrintF("%s/%s/%ld: Library=%08lx  Offset=%ld  NewFunc=%08lx  patchedFunc=%08lx\n", \
 278                                 __FILE__, __FUNC__, __LINE__, *pl->pe_Interface, \
 279                                 pl->pe_IFaceOffset, pl->pe_NewFunction, patchedFunction));
 280 #if !defined(__MORPHOS__)
 281                         if (patchedFunction != pl->pe_NewFunction)
 282                                 {
 283                                 *pl->pe_OldFunction = SetMethod(*pl->pe_Interface,
 284                                         pl->pe_IFaceOffset, patchedFunction);
 285
 286                                 Success = FALSE;
 287                                 }
 288 #endif /* __MORPHOS__ */
 289                         }
 290
 291                 pl++;
 292                 }
 293
 294 //      CacheClearU();
 295         Permit();
 296
 297         d1(KPrintF("%s/%s/%ld: Success=%ld\n", __FILE__, __FUNC__, __LINE__, Success));
 298
 299         if (!Success)
 300                 SetPatches(PatchList);
 301
 302         return Success;
 303 }
 304
 305 static LIBFUNC_P7VA(struct AppIcon *, sca_AddAppIcon,
 306         D0, ULONG, Id,
 307         D1, ULONG, UserData,
 308         A0, CONST_STRPTR, IconText,
 309         A1, struct MsgPort *, iconPort,
 310         A2, BPTR, lock,
 311         A3, struct DiskObject *, DiskObj,
 312         A6, struct Library *, wbBase)
 313 {
 314         struct AppIcon *ret;
 315         struct WorkbenchIFace *IWorkbench = (struct WorkbenchIFace *)self;
 316         va_list args;
 317         va_startlinear(args, DiskObj);
 318
 319         (void)wbBase;
 320
 321         ret = AddAppIconA(Id, UserData, IconText, iconPort, lock, DiskObj,
 322                  va_getlinearva(args, const struct TagItem *));
 323
 324         va_end(args);
 325
 326         return(ret);
 327 }
 328 LIBFUNC_END
 329
 330 static LIBFUNC_P5VA(struct AppWindow *, sca_AddAppWindow,
 331         D0, ULONG, Id,
 332         D1, ULONG, UserData,
 333         A0, struct Window *, window,
 334         A1, struct MsgPort *, iconPort,
 335         A6, struct Library *, wbBase)
 336 {
 337         struct AppWindow *ret;
 338         struct WorkbenchIFace *IWorkbench = (struct WorkbenchIFace *)self;
 339         va_list args;
 340         va_startlinear(args, iconPort);
 341
 342         (void)wbBase;
 343
 344         ret = AddAppWindowA(Id, UserData, window, iconPort, va_getlinearva(args, const struct TagItem *));
 345
 346         va_end(args);
 347
 348         return(ret);
 349 }
 350 LIBFUNC_END
 351
 352 static LIBFUNC_P5VA(struct AppMenuItem *, sca_AddAppMenuItem,
 353         D0, ULONG, Id,
 354         D1, ULONG, UserData,
 355         A0, CONST_STRPTR, text,
 356         A1, struct MsgPort *, iconPort,
 357         A6, struct Library *, wbBase)
 358 {
 359         struct AppMenuItem *ret;
 360         struct WorkbenchIFace *IWorkbench = (struct WorkbenchIFace *)self;
 361         va_list args;
 362         va_startlinear(args, iconPort);
 363
 364         (void)wbBase;
 365
 366         ret = AddAppMenuItemA(Id, UserData, text, iconPort, va_getlinearva(args, const struct TagItem *));
 367
 368         va_end(args);
 369
 370         return(ret);
 371 }
 372 LIBFUNC_END
 373
 374 static LIBFUNC_P2VA(struct Screen *, sca_OpenScreen,
 375         A0, struct NewScreen *, newScr,
 376         A6, struct IntuitionBase *, iBase)
 377 {
 378         struct Screen *ret;
 379         struct IntuitionIFace *IIntuition = (struct IntuitionIFace *)self;
 380         va_list args;
 381         va_startlinear(args, newScr);
 382
 383         (void)iBase;
 384
 385         ret = OpenScreenTagList(newScr, va_getlinearva(args, const struct TagItem *));
 386
 387         va_end(args);
 388
 389         return(ret);
 390 }
 391 LIBFUNC_END
 392