Record Code Signature of Downloaded DMG files

chrome/test/data/safe_browsing/mach_o/Makefile

 # Copyright 2015 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 KEYCHAIN_PASSWORD=g0atMaster
 # This must match the commonName in codesign.cfg.
 KEYCHAIN_IDENTITY=untrusted@goat.local
 
+# Funcitons to add and remove key and cert to users keychain.
+pre-build = security import codesign.key && security import codesign.crt
+post-build = security delete-identity -c untrusted@goat.local
+
 executable32: src.c
         clang -m32 -o $@ $^
 
 executable64: src.c
         clang -m64 -o $@ $^
 
 executablefat: executable32 executable64
         lipo -create -output $@ $^
 
 lib32.dylib: src.c
         clang -m32 -shared -o $@ $^
 
 lib64.dylib: src.c
         clang -m64 -shared -o $@ $^
 
 libfat.dylib: lib64.dylib lib32.dylib
         lipo -create -output $@ $^
 
 codesign.key:
         openssl genrsa -out $@ 2048
 
 codesign.csr: codesign.key codesign.cfg
         openssl req -new -key $< -out $@ -config codesign.cfg
 
 codesign.crt: codesign.csr codesign.key codesign.cfg
         openssl x509 -req -signkey codesign.key -sha256 \
                 -extfile codesign.cfg -extensions req_attrs -in $< -out $@
 
-codesign.keychain: codesign.key codesign.crt
-        security create-keychain -p $(KEYCHAIN_PASSWORD) $(PWD)/$@
-        security unlock-keychain -p $(KEYCHAIN_PASSWORD) $(PWD)/$@
-        certtool i ./codesign.crt k=$(PWD)/$@ r=./codesign.key
+signedexecutable32: executable32 codesign.crt
+        $(call pre-build)
+        cp $< $@
+        codesign -s $(KEYCHAIN_IDENTITY) $@
+        $(call post-build)
 
-signedexecutable32: executable32 codesign.keychain
+libsigned64.dylib: lib64.dylib codesign.crt
+        $(call pre-build)
         cp $< $@
-        security unlock-keychain -p $(KEYCHAIN_PASSWORD) \
-                $(PWD)/codesign.keychain
-        codesign -s $(KEYCHAIN_IDENTITY) --keychain $(PWD)/codesign.keychain $@
+        codesign -s $(KEYCHAIN_IDENTITY) $@
+        $(call post-build)
 
-libsigned64.dylib: lib64.dylib codesign.keychain
+signedexecutablefat: executablefat codesign.crt
+        $(call pre-build)
         cp $< $@
-        security unlock-keychain -p $(KEYCHAIN_PASSWORD) \
-                $(PWD)/codesign.keychain
-        codesign -s $(KEYCHAIN_IDENTITY) --keychain $(PWD)/codesign.keychain $@
+        codesign -s $(KEYCHAIN_IDENTITY) $@ --all-architectures
+        $(call post-build)
 
-signedexecutablefat: executablefat codesign.keychain
-        cp $< $@
-        security unlock-keychain -p $(KEYCHAIN_PASSWORD) \
-                $(PWD)/codesign.keychain
-        codesign -s $(KEYCHAIN_IDENTITY) --keychain $(PWD)/codesign.keychain \
-                $@ --all-architectures
+signed-archive.dmg: test-bundle.app codesign.crt
+        $(call pre-build)
+        hdiutil create -srcfolder test-bundle.app -format UDZO -layout \
+                SPUD -volname "Signed Archive" -ov $@
+        codesign -s $(KEYCHAIN_IDENTITY) $@
+        $(call post-build)
 
 .PHONY: test-bundle.app
 test-bundle.app: signedexecutablefat libsigned64.dylib executable32
+        $(call pre-build)
         ditto base-bundle.app $@
         ditto $< $@/Contents/MacOS/test-bundle
         ditto $(word 2,$^) $@/Contents/Frameworks/$(word 2,$^)
         ditto $(word 3,$^) $@/Contents/Resources/$(word 3,$^)
-        security unlock-keychain -p $(KEYCHAIN_PASSWORD) \
-                $(PWD)/codesign.keychain
-        codesign -f -s $(KEYCHAIN_IDENTITY) --keychain $(PWD)/codesign.keychain \
-                $@ --all-architectures --resource-rules ResourceRules
+        codesign -f -s $(KEYCHAIN_IDENTITY) $@ --all-architectures \
+                --resource-rules ResourceRules
+        $(call post-build)
 
 .PHONY: modified-bundle.app
 modified-bundle.app: test-bundle.app lib32.dylib executable64
+        $(call pre-build)
         ditto $< $@
         echo "<xml/>" > $@/Contents/Resources/Base.lproj/InfoPlist.strings
-        security unlock-keychain -p $(KEYCHAIN_PASSWORD) \
-                $(PWD)/codesign.keychain
-        codesign -f -s $(KEYCHAIN_IDENTITY) --keychain $(PWD)/codesign.keychain \
-                $@ --all-architectures --resource-rules ResourceRules
+        codesign -f -s $(KEYCHAIN_IDENTITY) $@ --all-architectures \
+                --resource-rules ResourceRules
         echo "BAD" > $@/Contents/Resources/Base.lproj/InfoPlist.strings
         touch $@/Contents/Resources/codesign.cfg
         ditto $(word 2,$^) $@/Contents/Frameworks/libsigned64.dylib
         ditto $(word 3,$^) $@/Contents/Resources/executable32
         echo "foo" >> $@/Contents/Resources/Base.lproj/MainMenu.nib
-        security unlock-keychain -p $(KEYCHAIN_PASSWORD) \
-                $(PWD)/codesign.keychain
-        codesign -f -s $(KEYCHAIN_IDENTITY) --keychain $(PWD)/codesign.keychain \
+        codesign -f -s $(KEYCHAIN_IDENTITY) \
                 $@/Contents/Resources/Base.lproj/MainMenu.nib
+        $(call post-build)
 
 .PHONY: modified-bundle-and-exec.app
 modified-bundle-and-exec.app: test-bundle.app lib32.dylib executable64
         ditto $< $@
         touch $@/Contents/Resources/codesign.cfg
         ditto $(word 2,$^) $@/Contents/Frameworks/libsigned64.dylib
         ditto $(word 3,$^) $@/Contents/Resources/executable32
         printf '\x31' | dd bs=1 seek=8097 count=1 conv=notrunc \
         of=$@/Contents/MacOS/test-bundle
 
 .PHONY: modified-main-exec32.app
 modified-main-exec32.app: test-bundle.app
         ditto $< $@
         printf '\x31' | dd bs=1 seek=8097 count=1 conv=notrunc \
                                                                                  of=$@/Contents/MacOS/test-bundle
 
 .PHONY: modified-main-exec64.app
 modified-main-exec64.app: test-bundle.app
         ditto $< $@
         printf '\x31' | dd bs=1 seek=24448 count=1 conv=notrunc \
         of=$@/Contents/MacOS/test-bundle
 
 .PHONY: modified-localization.app
 modified-localization.app: test-bundle.app
+        $(call pre-build)
         ditto $< $@
         echo "<xml/>" > $@/Contents/Resources/Base.lproj/InfoPlist.strings
-        security unlock-keychain -p $(KEYCHAIN_PASSWORD) \
-                $(PWD)/codesign.keychain
-        codesign -f -s $(KEYCHAIN_IDENTITY) --keychain $(PWD)/codesign.keychain \
-                $@ --all-architectures --resource-rules ResourceRules
+        codesign -f -s $(KEYCHAIN_IDENTITY) $@ --all-architectures \
+                --resource-rules ResourceRules
         echo "CORRUPT" > $@/Contents/Resources/Base.lproj/InfoPlist.strings
+        $(call post-build)