Changeset 219 for trunk/Cocoa/F-Script Anywhere/Source/FSAAppList.m

Timestamp:

05/09/06 09:13:54 (18 years ago)

Author:

rchin

Message:

Fixed commit of nib file.

Added support for automatic injection into apps.

File:

• trunk/Cocoa/F-Script Anywhere/Source/FSAAppList.m (modified) (10 diffs)

trunk/Cocoa/F-Script Anywhere/Source/FSAAppList.m

@@ -25 +25 @@
 */
 
+#include <sys/types.h>
+#include <sys/sysctl.h>
 #import "FSAAppList.h"
 #import "FSAnywhere.h"
@@ -39 +41 @@
 NSString * const FSATableColumnIdentifier_appNameAndIcon = @"appNameAndIcon";
 NSString * const FSATableColumnIdentifier_checkMark = @"checkMark";
+NSString * const FSATableColumnIdentifier_always = @"always";
 
 NSString *FSACheckMarkCharacter;
@@ -52 +55 @@
     NULL
 };
+
+static int sysctlbyname_with_pid (const char *name, pid_t pid, 
+                                                                  void *oldp, size_t *oldlenp, 
+                                                                  void *newp, size_t newlen);
+int is_pid_native (pid_t pid);
 
 @implementation FSAAppList
@@ -84 +92 @@
     patchingApps = [[NSMutableSet alloc] init];
     appsByPID = [[NSMutableDictionary alloc] init];
+        if(!([[NSUserDefaults standardUserDefaults] objectForKey:@"AlwaysApps"]))
+                alwaysApps = [[NSMutableArray array] retain];
+        else
+                alwaysApps = [[NSMutableArray arrayWithArray:[[NSUserDefaults standardUserDefaults] objectForKey:@"AlwaysApps"]] retain];
 
     [[tableView tableColumnWithIdentifier: FSATableColumnIdentifier_appNameAndIcon]
@@ -91 +103 @@
             setDataCell: [[[NSImageCell alloc] init] autorelease]];
     [window setResizeIncrements: NSMakeSize(1, [tableView cellHeight])];
+        [tableView setTarget:self];
+        [tableView setAction:@selector(clickInTable:)];
 
     [self update];
@@ -169 +183 @@
         return NO;
     return appContainsLibMatching([bundleExecutableLoc fileSystemRepresentation], FSACocoaFrameworks);
+}
+
+-(BOOL)appIsNative:(DeVercruesseProcess *)app
+{
+        return is_pid_native([app pid]);
 }
 
@@ -212 +231 @@
          ( ( [app isCocoa] && ![self appIsPEF: app]) ||
            ( [app isCarbon] && [self appIsCocoa: app]))) {
-        [cocoaApps addObject: app];
-    }
-    [appsByPID setObject: app forKey: [NSNumber numberWithInt: [app pid]]];
+                if([self appIsNative:app]){
+                        [cocoaApps addObject: app];
+                        if(finishedLaunch){
+                                if([alwaysApps containsObject:[app name]] && ![patchedApps containsObject:app])
+                                        [NSApp installBundleInAppWithPID:[app pid]];
+                        }
+                }
+    }
+    [appsByPID setObject: app forKey: [NSNumber numberWithInt: [app pid]]];     
 }
 
@@ -333 +358 @@
                 return FSAEllipsisImage;
         }
-    }
+    } else if([columnIdentifier isEqualToString:FSATableColumnIdentifier_always]){
+                if([alwaysApps containsObject:[[cocoaApps objectAtIndex: rowIndex] name]])
+                        return [NSNumber numberWithBool:YES];
+                else
+                        return [NSNumber numberWithBool:NO];
+        }
     return nil;
 }
@@ -418 +448 @@
 }
 
+- (void)applicationDidFinishLaunching:(NSNotification *)aNotification
+{
+        id anApp;
+        id e = [cocoaApps objectEnumerator];
+        while(anApp = [e nextObject]){
+                if([alwaysApps containsObject:[anApp name]] && ![patchedApps containsObject:anApp]){
+                        [NSApp installBundleInAppWithPID:[anApp pid]];
+                }
+        }
+        finishedLaunch = YES;
+}
+
 - (BOOL)applicationShouldTerminateAfterLastWindowClosed:(NSApplication *)sender;
 {
@@ -423 +465 @@
 }
 
+-(void)clickInTable:(id)sender
+{
+        if([sender clickedColumn] == 2){
+                if([sender clickedRow] >= 0){
+                        NSString *appName = [[cocoaApps objectAtIndex:[sender clickedRow]] name];
+                        if(![alwaysApps containsObject:appName]){
+                                [NSApp installBundleInAppWithPID:[appName pid]];
+                                [alwaysApps addObject:appName];
+                        } else
+                                [alwaysApps removeObject:appName];
+                        [[NSUserDefaults standardUserDefaults] setObject:alwaysApps forKey:@"AlwaysApps"];
+                }
+        }
+}
+
 @end
+
+static int sysctlbyname_with_pid (const char *name, pid_t pid, 
+                                                                  void *oldp, size_t *oldlenp, 
+                                                                  void *newp, size_t newlen)
+{
+    if (pid == 0) {
+        if (sysctlbyname(name, oldp, oldlenp, newp, newlen) == -1)  {
+            fprintf(stderr, "sysctlbyname_with_pid(0): sysctlbyname  failed:"
+                                        "%s\n", strerror(errno));
+            return -1;
+        }
+    } else {
+        int mib[CTL_MAXNAME];
+        size_t len = CTL_MAXNAME;
+        if (sysctlnametomib(name, mib, &len) == -1) {
+            fprintf(stderr, "sysctlbyname_with_pid: sysctlnametomib  failed:"
+                                        "%s\n", strerror(errno));
+            return -1;
+        }
+        mib[len] = pid;
+        len++;
+        if (sysctl(mib, len, oldp, oldlenp, newp, newlen) == -1)  {
+            fprintf(stderr, "sysctlbyname_with_pid: sysctl  failed:"
+                    "%s\n", strerror(errno));
+            return -1;
+        }
+    }
+    return 0;
+}
+
+int is_pid_native (pid_t pid)
+{
+    int ret = 0;
+    size_t sz = sizeof(ret);
+        
+    if (sysctlbyname_with_pid("sysctl.proc_native", pid, 
+                                                          &ret, &sz, NULL, 0) == -1) {
+                if (errno == ENOENT) {
+            // sysctl doesn't exist, which means that this version of Mac OS 
+            // pre-dates Rosetta, so the application must be native.
+            return 1;
+        }
+        fprintf(stderr, "is_pid_native: sysctlbyname_with_pid  failed:" 
+                "%s\n", strerror(errno));
+        return -1;
+    }
+    return ret;
+}